fastapi-start/log_config.py at main · agorinenko/fastapi-start · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
from pathlib import Path
import logging
import logging.config

from web_app.core.settings import APP_SETTINGS, parse_str_to_list

LOG_LEVEL = APP_SETTINGS.log_level
LOGGING_DEFAULT_HANDLER = parse_str_to_list(APP_SETTINGS.log_handler, default=['console'])
ENV_TYPE = APP_SETTINGS.env

Path(APP_SETTINGS.log_dir).mkdir(parents=True, exist_ok=True)

log_sql = APP_SETTINGS.log_sql
if log_sql:
    db_log_cfg = {
        'handlers': ['sql_console'],
        'level': LOG_LEVEL,
        'propagate': False,
    }
else:
    db_log_cfg = {
        'handlers': ['null'],
        'level': LOG_LEVEL,
        'propagate': False,
    }

LOGGING = {
    'version': 1,
    'disable_existing_loggers': False,
    'filters': {
    },
    'formatters': {
        'sql': {
            'format': '[SQL] %(levelname)s %(name)s %(asctime)s %(module)s %(message)s',
        },
        'default': {
            'format': '%(levelname)s %(name)s %(asctime)s %(module)s %(message)s',
        }
    },
    'handlers': {
        'file': {
            'class': 'logging.handlers.RotatingFileHandler',
            'maxBytes': 1024 * 1024 * 5,  # 5 MB
            'backupCount': 7,
            'filename': f'{APP_SETTINGS.log_dir}/{ENV_TYPE}.log',
            'formatter': 'default',
        },
        'console': {
            'class': 'logging.StreamHandler',
            'formatter': 'default'
        },
        'sql_console': {
            'class': 'logging.StreamHandler',
            'formatter': 'sql'
        },
        'null': {
            'class': 'logging.NullHandler',
        }
    },
    'loggers': {
        '': {
            'handlers': LOGGING_DEFAULT_HANDLER,
            'level': LOG_LEVEL,
            'propagate': False,
        },
        'py.warnings': {
            'handlers': LOGGING_DEFAULT_HANDLER,
            'level': 'WARNING',
            'propagate': False,
        },
        'sqlalchemy.engine': db_log_cfg,
        'sqlalchemy.pool': {
            'handlers': LOGGING_DEFAULT_HANDLER,
            'level': 'WARNING',
            'propagate': False,
        },
        'sqlalchemy.dialects': {
            'handlers': LOGGING_DEFAULT_HANDLER,
            'level': 'WARNING',
            'propagate': False,
        }
    }
}


def init_log():
    logging.config.dictConfig(LOGGING)