Log uvicorn output to selfie.log

vana-com · Mar 20, 2024 · 9683cef · 9683cef
1 parent a1d288d
commit 9683cef
Showing 1 changed file with 42 additions and 1 deletion.
diff --git a/selfie/__main__.py b/selfie/__main__.py
@@ -10,6 +10,7 @@
 from dotenv import load_dotenv
 
 from selfie.config import create_app_config, default_port, get_app_config
+from selfie.logging import get_log_path
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
@@ -91,7 +92,47 @@ def start_fastapi_server():
         uvicorn.run("selfie.__main__:get_configured_app", host="0.0.0.0", port=args.port, reload=True, factory=True)
     else:
         fastapi_app = get_configured_app(shareable=True)
-        uvicorn.run(fastapi_app, host="0.0.0.0", port=args.port)  # TODO: write to selfie.log
+
+        uvicorn_log_config = uvicorn.config.LOGGING_CONFIG
+
+        uvicorn_log_config["handlers"]["selfie"] = {
+            "class": "logging.handlers.RotatingFileHandler",
+            "filename": get_log_path(),
+            "maxBytes": 1024*1024,
+            "backupCount": 5,
+            "formatter": "default",
+        }
+
+        uvicorn_log_config["formatters"]["default"] = {
+            "class": "uvicorn.logging.DefaultFormatter",
+            "fmt": "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+            "use_colors": False,
+        }
+
+        uvicorn_log_config["formatters"]["access"] = {
+            "fmt": '%(asctime)s - %(levelname)s - %(client_addr)s - "%(request_line)s" %(status_code)s',
+            "datefmt": "%Y-%m-%d %H:%M:%S",
+            "use_colors": False,
+        }
+
+        uvicorn_log_config["handlers"]["console"] = {
+            "class": "logging.StreamHandler",
+            "formatter": "default",
+        }
+
+        uvicorn_log_config["loggers"]["uvicorn"] = {
+            "handlers": ["selfie", "console"],
+            "level": "INFO",
+            "propagate": False,
+        }
+
+        uvicorn_log_config["loggers"]["uvicorn.*"] = {
+            "handlers": ["selfie", "console"],
+            "level": "INFO",
+            "propagate": False,
+        }
+        # Start Uvicorn with the modified log config
+        uvicorn.run(fastapi_app, host="0.0.0.0", port=args.port, log_config=uvicorn_log_config)
 
 
 def main():