mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2025-12-19 14:54:21 +01:00
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
import logging, sys
|
|
|
|
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)
|
|
|
|
COLOR_SEQ, ITALIC_COLOR_SEQ = "\033[1;%dm", "\033[3;%dm"
|
|
|
|
COLORS = {
|
|
"DEBUG": CYAN,
|
|
"INFO": BLUE,
|
|
"WARNING": YELLOW,
|
|
"ERROR": RED
|
|
}
|
|
|
|
RESET_SEQ = "\033[0m"
|
|
|
|
class _ColorFormatter(logging.Formatter):
|
|
def __init__(self, msg, use_color = True):
|
|
logging.Formatter.__init__(self, msg, "%d-%m-%Y %H:%M:%S")
|
|
|
|
self.use_color = use_color
|
|
|
|
def format(self, record):
|
|
levelname = record.levelname
|
|
if self.use_color and levelname in COLORS:
|
|
record.name = ITALIC_COLOR_SEQ % (30 + BLACK) + record.name + RESET_SEQ
|
|
record.levelname = COLOR_SEQ % (30 + COLORS[levelname]) + levelname + RESET_SEQ
|
|
return logging.Formatter.format(self, record)
|
|
|
|
class ColorLogger(logging.Logger):
|
|
FORMAT = "[%(name)s] [%(levelname)s] [%(asctime)s] %(message)s"
|
|
|
|
def __init__(self, name, level):
|
|
logging.Logger.__init__(self, name, level)
|
|
|
|
colored_formatter = _ColorFormatter(self.FORMAT, use_color=True)
|
|
handler = logging.StreamHandler(stream=sys.stderr)
|
|
handler.setFormatter(fmt=colored_formatter)
|
|
|
|
self.addHandler(hdlr=handler)
|
|
|
|
class FileLogger(logging.Logger):
|
|
FORMAT = "[%(name)s] [%(levelname)s] [%(asctime)s] %(message)s"
|
|
|
|
def __init__(self, name, level, filename):
|
|
logging.Logger.__init__(self, name, level)
|
|
|
|
formatter = logging.Formatter(self.FORMAT)
|
|
handler = logging.FileHandler(filename=filename)
|
|
handler.setFormatter(fmt=formatter)
|
|
|
|
self.addHandler(hdlr=handler)
|