main.py

#!/usr/bin/env python
import subprocess
from contextlib import asynccontextmanager
from typing import TYPE_CHECKING, List, Optional

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from tortoise.contrib.fastapi import RegisterTortoise
from tortoise.contrib.pydantic import PydanticModel

from models.users import User
from settings import ALLOW_ORIGINS, DB_URL

if TYPE_CHECKING:

    class UserIn_Pydantic(User, PydanticModel):  # type:ignore[misc]
        pass

    class User_Pydantic(User, PydanticModel):  # type:ignore[misc]
        pass
else:
    from models.users import User_Pydantic, UserIn_Pydantic


@asynccontextmanager
async def lifespan(app: FastAPI):
    async with RegisterTortoise(
        app,
        config={
            "connections": {"default": DB_URL},
            "apps": {"models": {"models": ["models"]}},
            "use_tz": True,
            "timezone": "Asia/Shanghai",
        },
        generate_schemas=True,
        add_exception_handlers=True,
    ):
        yield


app = FastAPI(lifespan=lifespan)

app.add_middleware(
    CORSMiddleware,
    allow_origins=ALLOW_ORIGINS,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


@app.post("/testpost", response_model=User_Pydantic)
async def world(user: UserIn_Pydantic):
    data = user.model_dump(exclude_unset=True)
    user_obj = await User.create(**data)
    return await User_Pydantic.from_tortoise_orm(user_obj)


@app.get("/users", response_model=List[User_Pydantic])
async def user_list(name: Optional[str] = None, age: Optional[int] = None):
    qs = User.all()
    if name:
        qs = qs.filter(username__icontains=name)
    if age is not None:
        qs = qs.filter(age=age)
    return await User_Pydantic.from_queryset(qs)


def main() -> None:
    """Run server in development mode"""
    subprocess.run(["fastcdn", __file__, "--port=8000"])


if __name__ == "__main__":
    main()