mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2025-12-19 14:54:21 +01:00
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
"""
|
|
Module used to describe all of the different data types
|
|
"""
|
|
|
|
import logging
|
|
|
|
RESET_SEQ = "\033[0m"
|
|
COLOR_SEQ = "\033[1;%dm"
|
|
BOLD_SEQ = "\033[1m"
|
|
UNDERLINE_SEQ = "\033[04m"
|
|
|
|
YELLOW = '\033[93m'
|
|
WHITE = '\33[37m'
|
|
BLUE = '\033[34m'
|
|
LIGHT_BLUE = '\033[94m'
|
|
RED = '\033[91m'
|
|
GREY = '\33[90m'
|
|
|
|
KEYWORD_COLORS = {
|
|
'WARNING': YELLOW,
|
|
'INFO': LIGHT_BLUE,
|
|
'DEBUG': WHITE,
|
|
'CRITICAL': YELLOW,
|
|
'ERROR': RED,
|
|
'TRADE': '\33[102m\33[30m'
|
|
}
|
|
|
|
def formatter_message(message, use_color = True):
|
|
"""
|
|
Syntax highlight certain keywords
|
|
"""
|
|
if use_color:
|
|
message = message.replace("$RESET", RESET_SEQ).replace("$BOLD", BOLD_SEQ)
|
|
else:
|
|
message = message.replace("$RESET", "").replace("$BOLD", "")
|
|
return message
|
|
|
|
def format_word(message, word, color_seq, bold=False, underline=False):
|
|
"""
|
|
Surround the given word with a sequence
|
|
"""
|
|
replacer = color_seq + word + RESET_SEQ
|
|
if underline:
|
|
replacer = UNDERLINE_SEQ + replacer
|
|
if bold:
|
|
replacer = BOLD_SEQ + replacer
|
|
return message.replace(word, replacer)
|
|
|
|
class Formatter(logging.Formatter):
|
|
"""
|
|
This Formatted simply colors in the levelname i.e 'INFO', 'DEBUG'
|
|
"""
|
|
def __init__(self, msg, use_color = True):
|
|
logging.Formatter.__init__(self, msg)
|
|
self.use_color = use_color
|
|
|
|
def format(self, record):
|
|
"""
|
|
Format and highlight certain keywords
|
|
"""
|
|
levelname = record.levelname
|
|
if self.use_color and levelname in KEYWORD_COLORS:
|
|
levelname_color = KEYWORD_COLORS[levelname] + levelname + RESET_SEQ
|
|
record.levelname = levelname_color
|
|
record.name = GREY + record.name + RESET_SEQ
|
|
return logging.Formatter.format(self, record)
|
|
|
|
class CustomLogger(logging.Logger):
|
|
"""
|
|
This adds extra logging functions such as logger.trade and also
|
|
sets the logger to use the custom formatter
|
|
"""
|
|
FORMAT = "[$BOLD%(name)s$RESET] [%(levelname)s] %(message)s"
|
|
COLOR_FORMAT = formatter_message(FORMAT, True)
|
|
TRADE = 50
|
|
|
|
def __init__(self, name, logLevel='DEBUG'):
|
|
logging.Logger.__init__(self, name, logLevel)
|
|
color_formatter = Formatter(self.COLOR_FORMAT)
|
|
console = logging.StreamHandler()
|
|
console.setFormatter(color_formatter)
|
|
self.addHandler(console)
|
|
logging.addLevelName(self.TRADE, "TRADE")
|
|
return
|
|
|
|
def set_level(self, level):
|
|
logging.Logger.setLevel(self, level)
|
|
|
|
def trade(self, message, *args, **kws):
|
|
"""
|
|
Print a syntax highlighted trade signal
|
|
"""
|
|
if self.isEnabledFor(self.TRADE):
|
|
message = format_word(message, 'CLOSED ', YELLOW, bold=True)
|
|
message = format_word(message, 'OPENED ', LIGHT_BLUE, bold=True)
|
|
message = format_word(message, 'UPDATED ', BLUE, bold=True)
|
|
message = format_word(message, 'CLOSED_ALL ', RED, bold=True)
|
|
# Yes, logger takes its '*args' as 'args'.
|
|
self._log(self.TRADE, message, args, **kws) |