diff options
author | 2021-03-14 10:45:36 +0300 | |
---|---|---|
committer | 2021-03-15 02:16:24 +0300 | |
commit | a423419eca285d0b23ad3f31956297851a608d22 (patch) | |
tree | da008db807af16fc58c1910fad6367d4e91aef66 /backend/logs.py | |
parent | Use .gitattributes to normalise line endings (diff) |
Adds Logging Support
Adds formatters and handlers for logging.
Signed-off-by: Hassan Abouelela <[email protected]>
Diffstat (limited to 'backend/logs.py')
-rw-r--r-- | backend/logs.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/backend/logs.py b/backend/logs.py new file mode 100644 index 0000000..6b4a300 --- /dev/null +++ b/backend/logs.py @@ -0,0 +1,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) |