urec/chat_back
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Изменения мидлваре

Browse source
This commit is contained in:
urec56 2024-06-07 13:52:42 +05:00
parent 4efdd1d091
commit 4dc2fc58b1
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
app/main.py
View file

@ -46,9 +46,9 @@ app.mount("/static", StaticFiles(directory="app/static"), name="static")
class AddHeaderMiddleware(BaseHTTPMiddleware): class AddHeaderMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request: Request, call_next): async def dispatch(self, request: Request, call_next):
logging.critical(f"Запрос \n{request.headers["Sec-Websocket-Protocol"]}") logging.critical(f"Запрос \n{request.headers.get("Sec-Websocket-Protocol", "НИХУЯ")}")
response = await call_next(request) response = await call_next(request)
logging.critical(f"Ответ \n {response.headers["Sec-Websocket-Protocol"]}") logging.critical(f"Ответ \n {response.headers.get("Sec-Websocket-Protocol", "НИХУЯ")}")
if "Sec-Websocket-Protocol" in request.headers: if "Sec-Websocket-Protocol" in request.headers:
raise ValueError(request.headers["Sec-Websocket-Protocol"]) raise ValueError(request.headers["Sec-Websocket-Protocol"])
response.headers["Sec-Websocket-Protocol"] = request.headers["Sec-Websocket-Protocol"] response.headers["Sec-Websocket-Protocol"] = request.headers["Sec-Websocket-Protocol"]