import uvicorn
from fastapi import FastAPI
from tortoise.contrib.fastapi import register_tortoise

from test_orm.api.student import student_api
from test_orm.settings import TORTOISE_ORM

app = FastAPI()

app.include_router(student_api, prefix="/student", tags=["学生接口"])

# register_tortoise是注册函数，fastapi一旦运行，register_tortoise已经执行，通过传递进去的app对象，监听服务启动和终止事件
register_tortoise(
    app=app,
    config=TORTOISE_ORM,
)

if __name__ == '__main__':
    uvicorn.run('main:app', host='127.0.0.1', port=8001, reload=True, workers=1)

from typing import List

from fastapi import APIRouter
from pydantic import BaseModel, field_validator

from test_orm.models import Student, Course

student_api = APIRouter()





class StudentIn(BaseModel):
    name: str
    pwd: str
    sno: int
    clas_id: int
    courses: List[int] = []

    @field_validator("name")
    def name_must_alpha(cls, value):  # 使用 cls 而不是 self
        assert value.isalpha(), 'name must be alpha'
        return value

    @field_validator("sno")
    @classmethod  # 必须在下面
    def sno_validate(cls, value):
        assert 1000 <= value < 10000, '学号要在[1000-10000)的范围内'
        return value


@student_api.post("/")
async def addStudent(student_in: StudentIn):
    # 方式1
    # student = Student(name=student_in.name, pwd=student_in.pwd, sno=student_in.sno, clas_id=student_in.clas_id)
    # await student.save()  # 插入到数据库student表；必须要加await
    # 方式2
    student = await Student.create(name=student_in.name, pwd=student_in.pwd, sno=student_in.sno,
                                   clas_id=student_in.clas_id)
    # print(student, type(student))

    # 多对多的关系绑定
    choose_courses = await Course.filter(id__in=student_in.courses)
    print(choose_courses, type(choose_courses))
    await student.courses.add(*choose_courses)

    return student