logger: add colors
This commit is contained in:
parent
42a82a10e8
commit
933ffd833c
2 changed files with 9 additions and 3 deletions
11
logger.py
11
logger.py
|
@ -1,15 +1,20 @@
|
||||||
import click
|
import click
|
||||||
|
import coloredlogs
|
||||||
import logging
|
import logging
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
|
||||||
def setup_logging(verbose: bool):
|
def setup_logging(verbose: bool):
|
||||||
|
level_colors = coloredlogs.DEFAULT_LEVEL_STYLES | {'info': {'color': 'magenta', 'bright': True}, 'debug': {'color': 'blue', 'bright': True}}
|
||||||
|
field_colors = coloredlogs.DEFAULT_FIELD_STYLES | {'asctime': {'color': 'white', 'faint': True}}
|
||||||
level = logging.DEBUG if verbose else logging.INFO
|
level = logging.DEBUG if verbose else logging.INFO
|
||||||
logging.basicConfig(
|
coloredlogs.install(
|
||||||
stream=sys.stdout,
|
stream=sys.stdout,
|
||||||
format='%(asctime)s %(levelname)s: %(message)s',
|
fmt='%(asctime)s %(levelname)s: %(message)s',
|
||||||
datefmt='%m/%d/%Y %H:%M:%S',
|
datefmt='%Y-%m-%d %H:%M:%S',
|
||||||
level=level,
|
level=level,
|
||||||
|
level_styles=level_colors,
|
||||||
|
field_styles=field_colors,
|
||||||
)
|
)
|
||||||
logging.debug('Logging set up.')
|
logging.debug('Logging set up.')
|
||||||
|
|
||||||
|
|
|
@ -3,3 +3,4 @@ appdirs==1.4.4
|
||||||
joblib==1.0.1
|
joblib==1.0.1
|
||||||
toml
|
toml
|
||||||
typing_extensions
|
typing_extensions
|
||||||
|
coloredlogs
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue