github/pelican
1
0
Fork 1
mirror of https://github.com/getpelican/pelican.git synced 2025-10-15 20:28:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added a colored logging system

Browse source
This commit is contained in:
Skami18 2011-04-18 20:14:07 +02:00
commit c5578e338d
1 changed files with 67 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

96
pelican/log.py
View file

@ -1,47 +1,83 @@
import logging import logging
import sys
import os
gray = lambda(text) : '\033[1;30m' + str(text) + '\033[1;m' global ANSI
red = lambda(text) : '\033[1;31m' + str(text) + '\033[1;m' ANSI = {
green = lambda(text) : '\033[1;32m' + str(text) + '\033[1;m' 'gray' : lambda(text) : '\033[1;30m' + str(text) + '\033[1;m',
yellow = lambda(text) : '\033[1;33m' + str(text) + '\033[1;m' 'red' : lambda(text) : '\033[1;31m' + str(text) + '\033[1;m',
blue = lambda(text) : '\033[1;34m' + str(text) + '\033[1;m' 'green' : lambda(text) : '\033[1;32m' + str(text) + '\033[1;m',
magenta = lambda(text) : '\033[1;35m' + str(text) + '\033[1;m' 'yellow' : lambda(text) : '\033[1;33m' + str(text) + '\033[1;m',
cyan = lambda(text) : '\033[1;36m' + str(text) + '\033[1;m' 'blue' : lambda(text) : '\033[1;34m' + str(text) + '\033[1;m',
white = lambda(text) : '\033[1;37m' + str(text) + '\033[1;m' 'magenta' : lambda(text) : '\033[1;35m' + str(text) + '\033[1;m',
crimson = lambda(text) : '\033[1;38m' + str(text) + '\033[1;m' 'cyan' : lambda(text) : '\033[1;36m' + str(text) + '\033[1;m',
bgred = lambda(text) : '\033[1;41m' + str(text) + '\033[1;m' 'white' : lambda(text) : '\033[1;37m' + str(text) + '\033[1;m',
bggreen = lambda(text) : '\033[1;42m' + str(text) + '\033[1;m' 'crimson' : lambda(text) : '\033[1;38m' + str(text) + '\033[1;m',
bgbrown = lambda(text) : '\033[1;43m' + str(text) + '\033[1;m' 'bgred' : lambda(text) : '\033[1;41m' + str(text) + '\033[1;m',
bgblue = lambda(text) : '\033[1;44m' + str(text) + '\033[1;m' 'bggreen' : lambda(text) : '\033[1;42m' + str(text) + '\033[1;m',
bgmagenta = lambda(text) : '\033[1;45m' + str(text) + '\033[1;m' 'bgbrown' : lambda(text) : '\033[1;43m' + str(text) + '\033[1;m',
bgcyan = lambda(text) : '\033[1;46m' + str(text) + '\033[1;m' 'bgblue' : lambda(text) : '\033[1;44m' + str(text) + '\033[1;m',
bggray = lambda(text) : '\033[1;47m' + str(text) + '\033[1;m' 'bgmagenta' : lambda(text) : '\033[1;45m' + str(text) + '\033[1;m',
bgcrimson = lambda(text) : '\033[1;48m' + str(text) + '\033[1;m' 'bgcyan' : lambda(text) : '\033[1;46m' + str(text) + '\033[1;m',
'bggray' : lambda(text) : '\033[1;47m' + str(text) + '\033[1;m',
'bgcrimson' : lambda(text) : '\033[1;48m' + str(text) + '\033[1;m'
}
class ANSIFormatter(logging.Formatter):
class Formatter(logging.Formatter): """
Convert a `logging.LogReport' object into colored text, using ANSI escape sequences.
"""
## colors:
def format(self, record): def format(self, record):
if record.levelname is 'INFO': if not record.levelname or record.levelname is 'INFO':
return record.msg return ANSI['cyan'](record.msg)
elif record.levelname is 'WARNING': elif record.levelname is 'WARNING':
return yellow(record.levelname) + ': ' + record.msg return ANSI['yellow'](record.levelname) + ': ' + record.msg
elif record.levelname is 'ERROR': elif record.levelname is 'ERROR':
return red(record.levelname) + ': ' + record.msg return ANSI['red'](record.levelname) + ': ' + record.msg
elif record.levelname is 'CRITICAL': elif record.levelname is 'CRITICAL':
return bgred(record.levelname) + ': ' + record.msg return ANSI['bgred'](record.levelname) + ': ' + record.msg
class TextFormatter(logging.Formatter):
"""
Convert a `logging.LogReport' object into text.
"""
def format(self, record):
if not record.levelname or record.levelname is 'INFO':
return record.msg
else:
return record.levelname + ': ' + record.msg
class Formatter(object):
"""
A dummy class.
Return an instance of the appropriate formatter (ANSIFormatter if sys.stdout.isatty() is True, else TextFormatter)
"""
def __new__(cls, *args, **kwargs):
if os.isatty(sys.stdout.fileno()): # thanks to http://stackoverflow.com/questions/2086961/how-can-i-determine-if-a-python-script-is-executed-from-crontab/2087031#2087031
return ANSIFormatter(*args, **kwargs)
else:
return TextFormatter( *args, **kwargs)
def init(logger=logging.getLogger(), handler=logging.StreamHandler()): def init(logger=logging.getLogger(), handler=logging.StreamHandler()):
fmt = Formatter() fmt = Formatter()
handler.setFormatter(fmt) handler.setFormatter(fmt)
logger.addHandler(handler) logger.addHandler(handler)
logger.setLevel(logging.INFO)
init() init()
if __name__ == '__main__':
logging.info('info') logging.basicConfig(filename='example.log',level=logging.DEBUG)
logging.warning('warning') logging.debug('Logging test')
logging.error('error') logging.info('info')
logging.critical('critical') logging.warning('warning')
logging.error('error')
logging.critical('critical')