31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
from pkg_resources import (get_distribution,
|
|
DistributionNotFound, RequirementParseError)
|
|
try:
|
|
__version__ = get_distribution(__name__).version
|
|
except (DistributionNotFound, RequirementParseError):
|
|
# not installed as package or is being run from source/git checkout
|
|
from setuptools_scm import get_version
|
|
__version__ = get_version()
|
|
|
|
level = logging.INFO
|
|
if 'WHIPPER_DEBUG' in os.environ:
|
|
level = os.environ['WHIPPER_DEBUG'].upper()
|
|
|
|
log_init_func = logging.basicConfig
|
|
if 'WHIPPER_COLOR_LOG' in os.environ:
|
|
import coloredlogs
|
|
def init_coloredlogs(**kwargs):
|
|
# coloredlogs comes with its own log format, we don't want to use that
|
|
coloredlogs.install(fmt=logging.BASIC_FORMAT, **kwargs)
|
|
log_init_func = init_coloredlogs
|
|
|
|
if 'WHIPPER_LOGFILE' in os.environ:
|
|
log_init_func(filename=os.environ['WHIPPER_LOGFILE'],
|
|
filemode='w', level=level)
|
|
else:
|
|
log_init_func(stream=sys.stderr, level=level)
|