log_util.py

import logging
import logging.handlers
import logging.config
import os
import time
import traceback
from typing import Set

import yaml


logging_initialized: Set[str] = set()


def init_logging(module: str):
    if module in logging_initialized:
        return

    logging_initialized.add(module)

    if not os.path.exists('logs'):
        os.mkdir('logs/')

    with open('logging.cfg.yml', 'r', encoding='UTF-8') as fstream:
        # noinspection PyBroadException
        try:
            logging.config.dictConfig(yaml.safe_load(fstream))
            logger = logging.getLogger(module)

            # Т.к. suffix нельзя установить через конфиг, приходится делать так...
            for handler in logger.handlers:
                if type(handler) == logging.handlers.TimedRotatingFileHandler:
                    handler.suffix = '%Y.%m.%d.log'

            return logger
        except Exception:
            print('\n\n\nНе удалось настроить журналирование для модуля \"%s\":' % module)
            traceback.print_exc()
            print('\n\n\nВыход через 30 секунд...')
            time.sleep(30)
            print('\n\n\nВыход')
            exit(1)