jableader
7b6f4e2a3b
commit 21a17b771743b23ee41d11a90ed8fdc3433468ce
Author: jableader <jacobdunk@gmail.com>
Date: Mon Oct 20 00:12:02 2025 +1100
Completed tooling improvements, fixed remaining errors
commit 7db48e222e3aa1065c326197c33ba6439720f65a
Author: jableader <jacobdunk@gmail.com>
Date: Sun Oct 19 22:05:37 2025 +1100
autoformat
commit 5705ce24b64c2aa6f0b9426730a479165fa97e2a
Author: jableader <jacobdunk@gmail.com>
Date: Sun Oct 19 22:05:29 2025 +1100
tooling changes
commit f0a6b2fd147bb86b484927afd57b9ba0ac07bf47
Author: jableader <jacobdunk@gmail.com>
Date: Sun Oct 19 21:25:49 2025 +1100
Plan
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import aiosqlite
|
|
from fastapi import APIRouter, Depends, Request, Response
|
|
|
|
import persons
|
|
from api.deps import cookie_person, error_response, get_db
|
|
from common import ApiModel, ProblemDetails
|
|
|
|
router = APIRouter(prefix="/auth", tags=["auth"])
|
|
|
|
|
|
class LoginBody(ApiModel):
|
|
username: str
|
|
|
|
|
|
@router.post(
|
|
"/login",
|
|
response_model=persons.Person,
|
|
operation_id="login",
|
|
summary="Login and set user_id cookie",
|
|
responses={
|
|
200: {"model": persons.Person, "description": "Successful Response"},
|
|
404: {
|
|
"model": ProblemDetails,
|
|
"description": "Person not found",
|
|
"content": {"application/problem+json": {}},
|
|
},
|
|
},
|
|
)
|
|
async def login(
|
|
request: Request,
|
|
data: LoginBody,
|
|
response: Response,
|
|
conn: aiosqlite.Connection = Depends(get_db),
|
|
) -> persons.Person | Response:
|
|
person = await persons.get_by_name(conn, data.username)
|
|
if not person:
|
|
return error_response(request, 404, "Person not found")
|
|
|
|
# When using response_model, return the Pydantic model and set the cookie on the Response
|
|
response.set_cookie(key="user_id", value=str(person.id))
|
|
return person
|
|
|
|
|
|
@router.post(
|
|
"/refresh",
|
|
response_model=persons.Person,
|
|
operation_id="refresh",
|
|
summary="Refresh current user from cookie",
|
|
)
|
|
async def current_user(user: persons.Person = Depends(cookie_person)) -> persons.Person:
|
|
return user
|