colorful_format_line.py

#!/usr/bin/env python2.6
# coding: utf-8


import types


# DOUBLE_LINE_MAP = [ u'\u2800\u2880\u28a0\u28b0\u28b8',
#                     u'\u2840\u28C0\u28E0\u28F0\u28F8',
#                     u'\u2844\u28C4\u28E4\u28F4\u28FC',
#                     u'\u2846\u28C6\u28E6\u28F6\u28FE',
#                     u'\u2847\u28C7\u28E7\u28F7\u28FF', ]
DOUBLE_LINE_MAP = [
        [ "⠀","⢀","⢠","⢰","⢸", ],
        [ "⡀","⣀","⣠","⣰","⣸", ],
        [ "⡄","⣄","⣤","⣴","⣼", ],
        [ "⡆","⣆","⣦","⣶","⣾", ],
        [ "⡇","⣇","⣧","⣷","⣿", ],
]

x = DOUBLE_LINE_MAP
DOT_COUNTER_BOTTOM = [
        x[0][0],
        x[1][0], x[1][1],
        x[2][1], x[2][2],
        x[3][2], x[3][3],
        x[4][3], x[4][4],
]
del x

DOT_COUNTER = [ x[0] for x in DOUBLE_LINE_MAP ] \
        + [ x for x in DOUBLE_LINE_MAP[ 4 ][ 1: ] ]

DOT_COUNTER_2 = sum( DOUBLE_LINE_MAP, [] )

def tokenize( line ):
    # double quoted segment is preseverd

    tokens = line.split( ' ' )

    stck = [ [] ]

    for t in tokens:

        sp = t.split( '"' )
        n = len( sp )

        if n % 2 == 0:
            if len( stck ) == 1:
                stck.append( [ t ] )
            else:
                stck[ -1 ].append( t )
                sss = stck.pop()
                stck[ -1 ].append( ' '.join( sss ) )
        else:
            stck[ -1 ].append( t )

    return stck[ 0 ]


def line_pad( linestr, padding = '' ):

    lines = linestr.split( "\n" )

    if type( padding ) in types.StringTypes:
        lines = [ padding + x for x in lines ]

    elif callable( padding ):
        lines = [ padding( x ) + x for x in lines ]

    lines = "\n".join( lines )

    return lines


def format_line( items, sep=' ', aligns='' ):

    listtype = ( type( [] ), type( () ) )

    aligns = [ x for x in aligns ] + [ '' ] * len( items )
    aligns = aligns[ :len( items ) ]
    aligns = [ 'r' if x == 'r' else x for x in aligns ]


    items = [ ( x if type( x ) in listtype else [ x ] )
              for x in items ]

    items = [ [ to_str( y )
                for y in x ]
              for x in items ]


    maxHeight = max( [ len( x ) for x in items ] + [ 0 ] )

    max_width = lambda x: max( [ y.__len__()
                                 for y in x ] + [ 0 ] )

    widths = [ max_width( x ) for x in items ]

    items = [ ( x + [ '' ] * maxHeight )[ :maxHeight ]
              for x in items ]

    lines = []
    for i in range( maxHeight ):
        line = []
        for j in range( len(items) ):
            width = widths[ j ]
            elt = items[ j ][ i ]

            actualWidth = elt.__len__()
            elt = to_output_format( elt )

            if actualWidth < width:
                padding = ' ' * ( width - actualWidth )
                if aligns[ j ] == 'l':
                    elt = elt + padding
                else:
                    elt = padding + elt

            line.append( elt )

        line = sep.join( line )

        lines.append( line )

    return "\n".join( lines )

def to_str( y ):
    if isinstance( y, ColoredString ):
        pass
    elif type(y) in ( type(0), type(0L) ):
        y = str(y)
    elif type(y) in ( type([]), type( () ), type( {} ) ):
        y = str(y)

    return y

def colorize( v, total, ptn='{0}' ):
    if total > 0:
        color = fading_color( v, total )
    else:
        color = fading_color( -total - v, -total )
    return ColoredString( ptn.format( v ), color )


class ColoredString( object ):

    def __init__( self, v, color=None ):
        if type( color ) == type( '' ):
            color = _named_colors[ color ]

        if isinstance( v, ColoredString ):
            vs = ''.join( [ x[0] for x in v.elts ] )
            self.elts = [ ( vs, color ) ]
        else:
            self.elts = [ ( str(v), color ) ]

    def __str__( self ):
        rst = []
        for e in self.elts:
            if e[1] is None:
                val = e[0]
            else:
                val = '\033[38;5;' + str(e[1]) + 'm' + str(e[0]) + '\033[0m'
            rst.append( val )
        return ''.join( rst )

    def __len__( self ):
        return sum( [ len(x[0])
                      for x in self.elts ] )

    def __add__( self, other ):
        c = ColoredString( 0 )
        if isinstance( other, ColoredString ):
            c.elts = self.elts + other.elts
        else:
            c.elts = self.elts[:] + [ (str(other), None) ]
        return c

    def __mul__( self, num ):
        c = ColoredString( 0 )
        c.elts = self.elts * num
        return c


def blue( v ): return ColoredString( v, 'blue' )
def cyan( v ): return ColoredString( v, 'cyan' )
def green( v ): return ColoredString( v, 'green' )
def yellow( v ): return ColoredString( v, 'yellow' )
def red( v ): return ColoredString( v, 'red' )
def purple( v ): return ColoredString( v, 'purple' )
def white( v ): return ColoredString( v, 'white' )

def fading_color( v, total ):
    return _clrs[ fading_idx( v, total ) ]


def fading_idx( v, total = 100 ):
    l = len( _clrs )
    pos = int( v * l / total + 0.5 )
    pos = min( pos, l - 1 )
    pos = max( pos, 0 )
    return pos


_clrs = [63,67,37,36,41,46,82,118,154,190,226,220,214,208,202,196]
_named_colors = {
        # by emergence levels
        'danger': _clrs[ fading_idx( 100 ) ],
        'warn': _clrs[ fading_idx( 60 ) ],
        'loaded': _clrs[ fading_idx( 30 ) ],
        'optimal': _clrs[ fading_idx( 0 ) ],

        'dark': _clrs[ 1 ],

        # for human
        'blue': 67,
        'cyan': 37,
        'green': 46,
        'yellow': 226,
        'red': 196,
        'purple': 128,
        'white': 255,
}

def to_output_format( s ):

    if type( s ) == type( u'' ):
        s = s.encode( 'utf-8' )
    else:
        s = str( s )

    return s

if __name__ == "__main__":

    cc = ColoredString

    line = [
        [cc('hello', 'red'), cc('world', 'blue')],
        [cc('hi', 'red'), cc('colorstring', 'green')],
    ]

    line = zip(*line)

    print format_line(line)