mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-31 04:42:09 -08:00
Code Issues Projects Releases Packages Wiki Activity

DO NOT DO THIS AT HOME

Browse source
This commit is contained in:
Darrel Pol 2024-11-03 19:25:43 -06:00
parent de497acd69
commit 82b3eea1fd
2 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
basic_auth_middleware.py
View file

@ -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()

7
webui.py
View file

@ -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)