# Python library for ansi colorization
import re,sys

class code:
    K="\033[1;30m"
    R="\033[1;31m"
    G="\033[1;32m"
    B="\033[1;34m"
    Y="\033[1;33m"
    M="\033[1;35m"
    C="\033[1;36m"
    W="\033[1;37m"

    k="\033[0;30m"
    r="\033[0;31m"
    g="\033[0;32m"
    b="\033[0;34m"
    y="\033[0;33m"
    m="\033[0;35m"
    c="\033[0;36m"
    w="\033[0;37m"

    bk="\033[40m"
    br="\033[41m"
    bg="\033[42m"
    by="\033[43m"
    bb="\033[44m"
    bm="\033[45m"
    bc="\033[46m"
    bw="\033[47m"

    S = '\033[1m'#strong
    s = '\033[2m'#strong off
    U = '\033[4m'#underline
    u = '\033[24m'#underline off
    Z = '\033[0m'#zero colors
    ic = '\033[7m'#inverse colors
    io = '\033[27m'#inverse off
    st = '\033[9m'#strike on
    so = '\033[29m'#strike off
    CLR = '\033[2J'
    CLREND = '\033[K'
    CLRBEG = '\033[1K'
    CLRSCR = CLR+"\033[0;0H"

    color_keys="K,R,G,B,Y,M,C,W,k,r,g,b,y,m,c,w,S,s,U,u,Z,ic,io,st,so,bk,br,bg,by,bb,bm,bc,bw,CLR,CLREND,CLRBEG,CLRSCR".split(",")
    color_list=[K,R,G,B,Y,M,C,W,k,r,g,b,y,m,c,w,S,s,U,u,Z,ic,io,st,so,bk,br,bg,by,bb,bm,bc,bw,CLR,CLREND,CLRBEG,CLRSCR]
    custom_match=re.compile(r'(\${)([0-9;]*[ABCDEFGHJKSTfminsu]+)(})')

    def pos(self,y,x):
        """ Go to absolute position """
        return "\033["+str(y)+";"+str(x)+"H"
    def column(self,x):
        """ Go to absolute column """
        return "\033["+str(x)+"G"

    def posprint(self, y,x,s):
        """ Print string at a location """
        sys.stdout.write( self.pos(y,x) + str(s) )
        self.flush()

    def clear(self):
        sys.stdout.write( self.CLRSCR+self.pos(0,0) )
    def clear_to_end(self):
        sys.stdout.write( self.CLREND )
    def clear_to_beginning(self):
        sys.stdout.write( self.CLRBEG )

    def up(self,n=1):
        sys.stdout.write( "\033["+str(n)+"A" )
    def down(self,n=1):
        sys.stdout.write( "\033["+str(n)+"B" )
    def right(self,n=1):
        sys.stdout.write( "\033["+str(n)+"C" )
    def left(self,n=1):
        sys.stdout.write( "\033["+str(n)+"D" )
    def up_line(self,n=1):
        sys.stdout.write( "\033["+str(n)+"F" )
    def down_line(self,n=1):
        sys.stdout.write( "\033["+str(n)+"E" )

    def save(self):
        """ Save cursor position """
        sys.stdout.write( "\033[s" )
    def restore(self):
        """ Restore cursor position """
        sys.stdout.write( "\033[u" )

    def color_string(self,s):
        for i,c in enumerate(self.color_keys):
            s=s.replace("${"+c+"}",self.color_list[i])
        return self.custom_color(s)
    def nocolor_string(self,s):
        for i,c in enumerate(self.color_keys):
            s=s.replace("${"+c+"}","")
        return self.custom_nocolor(s)

    def custom_color(self,s):
        return self.custom_match.sub('\033[\\2',s)
    def custom_nocolor(self,s):
        return self.custom_match.sub('',s)

    def get_keys(self):
        return self.color_keys

    def flush(self):
        sys.stdout.flush()


def demo():
    """ Print all control sequences """
    c=code()
    unformatted="""${S}ANSI CODES
==========${Z}
${S}Fo${U}rm${st}at${u}ti${ic}ng${Z}
${S}==========${Z}
   0  Z Clear format
   1  S ${S}Strong   ${Z}     2  s ${s}Off${Z}
   4  U ${U}Underline${Z}    24  u Off
   7 ic ${ic}Inverse${Z}      27 io Off
   9 st ${st}Strike${Z}       29 so Off
${R}Co${G}lo${B}rs${Z}
${S}======${Z}
  30 k ${k}Black   ${Z}1 K ${K}Strong   ${Z}40 bk ${bk}Background${Z}
  31 r ${r}Red     ${Z}1 R ${R}Strong   ${Z}41 br ${br}Background${Z}
  32 g ${g}Green   ${Z}1 G ${G}Strong   ${Z}42 bg ${bg}Background${Z}
  33 y ${y}Yellow  ${Z}1 Y ${Y}Strong   ${Z}43 by ${by}Background${Z}
  34 b ${b}Blue    ${Z}1 B ${B}Strong   ${Z}44 bb ${bb}Background${Z}
  35 m ${m}Magenta ${Z}1 M ${M}Strong   ${Z}45 bm ${bm}Background${Z}
  36 c ${c}Cyan    ${Z}1 C ${C}Strong   ${Z}46 bc ${bc}Background${Z}
  37 w ${w}White   ${Z}1 W ${W}Strong   ${Z}47 bw ${bw}Background${Z}
${S}Clearing and movement
${S}=====================${Z}
   2J     CLR     Clear screen
   2J ;H  CLRSCR .clear()  Clear screen and cursor to upper left
   K      CLREND  Clear to end of line
   1K     CLRBEG  Clear to beginning of line

   s .save()  Save location   u .restore()   Restore location
   A .up()    Up              E .up_line()   Up line
   B .down()  Down            F .down_line() Down line
   C .left()  Left         y;xH .pos()       Absolute Position
   D .right() Right """

    return(c.color_string(unformatted))