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 coloredlogs
|
||||
import logging
|
||||
import sys
|
||||
|
||||
|
||||
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
|
||||
logging.basicConfig(
|
||||
coloredlogs.install(
|
||||
stream=sys.stdout,
|
||||
format='%(asctime)s %(levelname)s: %(message)s',
|
||||
datefmt='%m/%d/%Y %H:%M:%S',
|
||||
fmt='%(asctime)s %(levelname)s: %(message)s',
|
||||
datefmt='%Y-%m-%d %H:%M:%S',
|
||||
level=level,
|
||||
level_styles=level_colors,
|
||||
field_styles=field_colors,
|
||||
)
|
||||
logging.debug('Logging set up.')
|
||||
|
||||
|
|
|
@ -3,3 +3,4 @@ appdirs==1.4.4
|
|||
joblib==1.0.1
|
||||
toml
|
||||
typing_extensions
|
||||
coloredlogs
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue