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

adding more logs and also considering lower auth header

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

3
basic_auth_middleware.py
View file

@ -14,7 +14,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")
auth_header = request.headers.get("Authorization") or request.headers.get("authorization")
if not auth_header or not auth_header.startswith("Basic "):
return self._unauthorized_response()
@ -24,6 +24,7 @@ class BasicAuthMiddleware(BaseHTTPMiddleware):
# We should add a step to santize the input here
decoded_credentials = base64.b64decode(encoded_credentials).decode("utf-8")
provided_username, provided_password = decoded_credentials.split(":")
logger.warning(f"provided_username: {provided_username}: provided_password: {provided_password}")
# Check credentials
if provided_username == self.username and provided_password == self.password: