import logging import os import sys from importlib.metadata import version, PackageNotFoundError try: __version__ = version('whipper') except PackageNotFoundError: # 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)