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
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Annotated, AsyncGenerator, Optional
|
|
|
|
import aiosqlite
|
|
from fastapi import Cookie, Depends, Request
|
|
from fastapi.responses import JSONResponse
|
|
|
|
import db
|
|
import persons
|
|
from common import ProblemDetails
|
|
from settings import settings
|
|
|
|
|
|
# Dependency to create SQLite connection with PRAGMAs and per-request transaction
|
|
async def get_db() -> AsyncGenerator[aiosqlite.Connection, None]:
|
|
sql_db = await db.connect(settings.database_path)
|
|
# Connection-level configuration
|
|
try:
|
|
# Enable FK enforcement
|
|
await sql_db.execute("PRAGMA foreign_keys=ON;")
|
|
# Prefer WAL for better concurrency; ignore result
|
|
async with sql_db.execute("PRAGMA journal_mode=WAL;") as _:
|
|
await _.fetchone()
|
|
# Reasonable durability/perf tradeoff
|
|
await sql_db.execute("PRAGMA synchronous=NORMAL;")
|
|
# Begin a transaction for the whole request
|
|
await sql_db.execute("BEGIN;")
|
|
|
|
try:
|
|
yield sql_db
|
|
await sql_db.commit()
|
|
except Exception:
|
|
await sql_db.rollback()
|
|
raise
|
|
finally:
|
|
await sql_db.close()
|
|
|
|
|
|
async def cookie_person(
|
|
user_id: Optional[int] = Cookie(None, alias="user_id"),
|
|
conn: aiosqlite.Connection = Depends(get_db),
|
|
) -> Optional[persons.Person]:
|
|
if user_id is None:
|
|
return None
|
|
return await persons.get_by_id(conn, user_id)
|
|
|
|
|
|
def error_response(request: Optional[Request], status_code: int, message: str) -> JSONResponse:
|
|
body = ProblemDetails(
|
|
title=message,
|
|
status=status_code,
|
|
type=f"https://httpstatuses.com/{status_code}",
|
|
instance=str(request.url) if request else None,
|
|
)
|
|
return JSONResponse(
|
|
content=body.model_dump(by_alias=True),
|
|
status_code=status_code,
|
|
media_type="application/problem+json",
|
|
)
|