sanspie/akarpov
1
1
Fork 0
mirror of https://github.com/Alexander-D-Karpov/akarpov synced 2025-04-19 10:31:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Provide session to worker to convert it to user id

Browse Source
This commit is contained in:
Damir Modyarov 2023-08-07 23:06:37 +03:00
parent 44982a1190
commit c15df62c17
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
redirect/app.py
View File

@ -1,7 +1,7 @@
from typing import Annotated
import django
from fastapi import FastAPI, Depends, Header
from fastapi import FastAPI, Depends, Header, Cookie
from fastapi.responses import RedirectResponse, HTMLResponse
from sqlalchemy.orm import Session
from starlette.requests import Request
@ -44,7 +44,13 @@ async def unicorn_exception_handler(request: Request, exc: LinkNotFoundException
@app.get("/{slug}")
def redirect(slug: str, request: Request, db: Session = Depends(get_db), user_agent: Annotated[str | None, Header()] = None) -> RedirectResponse:
def redirect(
slug: str,
request: Request,
db: Session = Depends(get_db),
user_agent: Annotated[str | None, Header()] = None,
sessionid: Annotated[str | None, Cookie()] = None
) -> RedirectResponse:
"""Main route that redirects to a page based on the slug."""
if '+' in slug:
return RedirectResponse(url=f'/tools/shortener/p/{slug.replace("+", "")}')
@ -56,7 +62,7 @@ def redirect(slug: str, request: Request, db: Session = Depends(get_db), user_ag
"pk": link[0],
"ip": request.client.host,
"user_agent": user_agent,
"user_id": 0,
"token": sessionid,
},
)