sanspie/akarpov
1
1
Fork 0
mirror of https://github.com/Alexander-D-Karpov/akarpov synced 2025-07-15 02:32:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrong task argument name

Browse Source
This commit is contained in:
Damir Modyarov 2023-08-08 13:16:12 +03:00
parent 3835fc5f7f
commit e1959f7232
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
redirect/app.py
View File

@ -2,12 +2,12 @@
from typing import Annotated from typing import Annotated
import django import django
from fastapi import FastAPI, Depends, Header, Cookie from fastapi import Cookie, Depends, FastAPI, Header
from fastapi.responses import RedirectResponse, HTMLResponse from fastapi.responses import HTMLResponse, RedirectResponse
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from starlette.requests import Request from starlette.requests import Request
from redirect.db.curd import get_link_by_slug, LinkNotFoundException from redirect.db.curd import LinkNotFoundException, get_link_by_slug
from redirect.db.dependency import get_db from redirect.db.dependency import get_db
from redirect.settings import settings from redirect.settings import settings
@ -16,7 +16,9 @@
app = FastAPI() app = FastAPI()
from akarpov.tools.shortener.tasks import save_view_meta # noqa: This has to be imported strictly AFTER django setup from akarpov.tools.shortener.tasks import ( # noqa: This has to be imported strictly AFTER django setup
save_view_meta,
)
@app.exception_handler(LinkNotFoundException) @app.exception_handler(LinkNotFoundException)
@ -51,10 +53,10 @@ def redirect(
request: Request, request: Request,
db: Session = Depends(get_db), db: Session = Depends(get_db),
user_agent: Annotated[str | None, Header()] = None, user_agent: Annotated[str | None, Header()] = None,
sessionid: Annotated[str | None, Cookie()] = None sessionid: Annotated[str | None, Cookie()] = None,
) -> RedirectResponse: ) -> RedirectResponse:
"""Main route that redirects to a page based on the slug.""" """Main route that redirects to a page based on the slug."""
if '+' in slug: if "+" in slug:
return RedirectResponse(url=f'/tools/shortener/p/{slug.replace("+", "")}') return RedirectResponse(url=f'/tools/shortener/p/{slug.replace("+", "")}')
link_id, link_target = get_link_by_slug(db, slug) link_id, link_target = get_link_by_slug(db, slug)
@ -64,8 +66,12 @@ def redirect(
"pk": link_id, "pk": link_id,
"ip": request.client.host, "ip": request.client.host,
"user_agent": user_agent, "user_agent": user_agent,
"token": sessionid, "user_id": sessionid,
}, },
) )
return RedirectResponse(url=(settings.relative_base + link_target) if link_target.startswith('/') else link_target) return RedirectResponse(
url=(settings.relative_base + link_target)
if link_target.startswith("/")
else link_target
)