-
Notifications
You must be signed in to change notification settings - Fork 2
/
__init__.py
32 lines (26 loc) · 957 Bytes
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# -*- coding: utf-8 -*-
import logging
import logging.handlers
from logutils.colorize import ColorizingStreamHandler
__author__ = 'TwinDB Development Team'
__email__ = 'dev@twindb.com'
__version__ = '1.1.4'
log = logging.getLogger(__name__)
def setup_logging(logger, debug=False):
fmt_str = "%(asctime)s: %(levelname)s:" \
" %(module)s.%(funcName)s():%(lineno)d: %(message)s"
ColorizingStreamHandler.level_map = {
logging.DEBUG: (None, 'blue', False),
logging.INFO: (None, 'green', False),
logging.WARNING: (None, 'yellow', False),
logging.ERROR: (None, 'red', False),
logging.CRITICAL: ('red', 'white', True),
}
console_handler = logging.StreamHandler()
console_handler.setFormatter(logging.Formatter(fmt_str))
logger.handlers = []
logger.addHandler(console_handler)
if debug:
logger.setLevel(logging.DEBUG)
else:
logger.setLevel(logging.INFO)