restic-compose-backup/restic_compose_backup/log.py

import logging
import os
import sys

logger = logging.getLogger('restic_compose_backup')
HOSTNAME = os.environ['HOSTNAME']

DEFAULT_LOG_LEVEL = logging.INFO
LOG_LEVELS = {
    'debug': logging.DEBUG,
    'info': logging.INFO,
    'warning': logging.WARNING,
    'error': logging.ERROR,
}

def setup(level: str = 'warning'):
    """Set up logging"""
    # Get log level from env if not set directly
    if level is None:
        level = os.environ.get('LOG_LEVEL')

    level = level or ""
    level = LOG_LEVELS.get(level.lower(), DEFAULT_LOG_LEVEL)
    logger.setLevel(level)

    ch = logging.StreamHandler(stream=sys.stdout)
    ch.setLevel(level)
    ch.setFormatter(logging.Formatter(f'%(asctime)s - {HOSTNAME} - %(name)s - %(levelname)s - %(message)s'))
    logger.addHandler(ch)