python 带颜色样式打印到终端

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""
Created on Tue Aug  8 17:01:54 2017

@author: http://www.pythoner.com/357.html

也许你希望在终端上输出一些带有颜色或者粗体、下划线等样式的信息，就像man中的那样，那么这篇文章将会起到些许作用。

在Linux终端中，使用转义序列来进行如上所述的显示，转义序列以ESC开头，即ASCII码下的\033，其格式为：
\033[显示方式;前景色;背景色m
显示方式、前景色、背景色至少一个存在即可。

其中：
显示方式：0（关闭所有效果），1（高亮），4（下划线），5（闪烁），7（反色），8（不可见）。
前景色以3开头，背景色以4开头，具体颜色值有：0（黑色），1（红色），2（绿色），3（黄色），4（蓝色），5（紫色），6（青色），7（白色）。即前景绿色为32，背景蓝色为44。

"""

STYLE = {
        'fore': {
                'black': 30, 'red': 31, 'green': 32, 'yellow': 33,
                'blue': 34, 'purple': 35, 'cyan': 36, 'white': 37,
        },
        'back': {
                'black': 40, 'red': 41, 'green': 42, 'yellow': 43,
                'blue': 44, 'purple': 45, 'cyan': 46, 'white': 47,
        },
        'mode': {
                'bold': 1, 'underline': 4, 'blink': 5, 'invert': 7,
        },
        'default': {
                'end': 0,
        }
}

def define_style(mode='', fore='', back=''):
    mode = '%s' % STYLE['mode'][mode] if STYLE['mode'].has_key(mode) else ''
    fore = '%s' % STYLE['fore'][fore] if STYLE['fore'].has_key(fore) else ''
    back = '%s' % STYLE['back'][back] if STYLE['back'].has_key(back) else ''
    style = ';'.join([s for s in [mode, fore, back] if s])
    style = '\033[%sm' % style if style else ''
    end = '\033[%sm' % STYLE['default']['end'] if style else ''

    def use_style(string):
        return '%s%s%s' % (style, string, end)
    return use_style

def test():
    print use_style('Normal')
    print use_style('Bold', mode='bold')
    print use_style('Underline & red text', mode='underline', fore='red')
    print use_style('Invert & green back', mode='reverse', back='green')
    print use_style('Black text & White back', fore='black', back='white')

error = define_style('bold', 'red')
warn = define_style('bold', 'yellow', 'black')
critical = define_style('reverse', back='green')

if __name__ == '__main__':
#    test()
    print critical('Exception occurs !\n oh my!')