From 82b3eea1fd325ac437db95eecbe994362d662e05 Mon Sep 17 00:00:00 2001 From: Darrel Pol Date: Sun, 3 Nov 2024 19:25:43 -0600 Subject: [PATCH] DO NOT DO THIS AT HOME --- basic_auth_middleware.py | 3 +++ webui.py | 7 +++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/basic_auth_middleware.py b/basic_auth_middleware.py index 55c818b43..d32234d66 100644 --- a/basic_auth_middleware.py +++ b/basic_auth_middleware.py @@ -2,7 +2,9 @@ import base64 from starlette.middleware.base import BaseHTTPMiddleware from starlette.responses import Response from fastapi import FastAPI +import logging +logger = logging.getLogger(__name__) class BasicAuthMiddleware(BaseHTTPMiddleware): def __init__(self, app, username: str, password: str): super().__init__(app) @@ -11,6 +13,7 @@ class BasicAuthMiddleware(BaseHTTPMiddleware): async def dispatch(self, request, call_next): # Extract the Authorization header + logger.warning(f"{request.headers}") auth_header = request.headers.get("Authorization") if not auth_header or not auth_header.startswith("Basic "): return self._unauthorized_response() diff --git a/webui.py b/webui.py index e3de3120a..238ef12d9 100644 --- a/webui.py +++ b/webui.py @@ -7,6 +7,9 @@ import time from modules import timer from modules import initialize_util from modules import initialize +import logging + +logger = logging.getLogger(__name__) startup_timer = timer.startup_timer startup_timer.record("launcher") @@ -37,8 +40,8 @@ def api_only(): from basic_auth_middleware import BasicAuthMiddleware USERNAME = os.getenv('SDAPI_USERNAME') PASSWORD = os.getenv('SDAPI_PASSWORD') - print(f"{USERNAME}: {PASSWORD}") - app.add_middleware(BasicAuthMiddleware, username=USERNAME, password=PASSWORD) + logger.warning(f"{USERNAME}:{PASSWORD}") + app.add_middleware(BasicAuthMiddleware, username="administrator", password="vY0WFuGCpL3lILb74ZzMCbSAL10x5V") initialize_util.setup_middleware(app) api = create_api(app)