blob: 6b4a30015cef562f0124df632a8b90477d6c5a8f (
plain) (
blame)
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
|
"""
Patch the uvicorn, watchgod, and global loggers.
"""
import logging
import sys
from logging import handlers
from pathlib import Path
from backend import constants
# Setup constants
LOGGING_LEVEL = logging.INFO if constants.PRODUCTION else logging.DEBUG
FORMATTER = logging.Formatter("%(asctime)s | %(name)s | %(levelname)s | %(message)s")
# Setup the project logger
handler = logging.StreamHandler(sys.stdout)
handler.setFormatter(FORMATTER)
logger = logging.getLogger("backend")
logger.setLevel(LOGGING_LEVEL)
logger.addHandler(handler)
# Format uvicorn logging
try:
logging.getLogger("uvicorn").handlers[0].setFormatter(FORMATTER)
except KeyError:
logger.warning("Could not patch uvicorn logger, continuing.")
# Add file handlers for local development
if constants.LOG_FILES:
# Setup uvicorn handler
uvicorn_log = Path("logs", ".uvicorn.log")
uvicorn_log.parent.mkdir(exist_ok=True)
# Store two copies of 100KB Files
uvicorn_handler = handlers.RotatingFileHandler(uvicorn_log, maxBytes=100000, backupCount=1)
uvicorn_handler.setFormatter(FORMATTER)
# Setup app handler
backend_log = Path("logs", ".backend.log")
file_handler = logging.FileHandler(backend_log)
file_handler.setFormatter(FORMATTER)
# Add all handlers
logging.getLogger("uvicorn").addHandler(uvicorn_handler)
logger.addHandler(file_handler)
|
