增删改查

  1. from typing import List, Any, Union
  3. from fastapi import APIRouter, Depends
  5. from core import deps
  6. from models import Movie
  7. from scheams import Movie_Pydantic, MovieIn_Pydantic, Response200, Response400
  9. movie = APIRouter(tags=["电影相关"])
  12. @movie.get("/movie", summary="电影列表", response_model=Union[Response200, Response400])
  13. async def movie_list(limit: int = 10, page: int = 1):
  14.     skip = (page - 1) * limit
  15.     data = {
  16.         "total": await Movie.all().count(),
  17.         "movies": await Movie_Pydantic.from_queryset(Movie.all().offset(skip).limit(limit).order_by('-id'))
  18.     }
  19.     return Response200(data=data)
  22. @movie.post("/movie", summary="新增电影")
  23. async def movie_create(movie_form: MovieIn_Pydantic, token: Any = Depends(deps.get_current_user)):
  24.     return Response200(data=await Movie_Pydantic.from_tortoise_orm(await Movie.create(**movie_form.dict())))
  27. @movie.put("/movie/{pk}", summary="编辑电影")
  28. async def movie_update(pk: int, movie_form: MovieIn_Pydantic, token: Any = Depends(deps.get_current_user)):
  29.     if await Movie.filter(pk=pk).update(**movie_form.dict()):
  30.         return Response200()
  31.     return Response400(msg="更新失败")
  34. @movie.delete("/movie/{pk}", summary="删除电影")
  35. async def movie_delete(pk: int, token: Any = Depends(deps.get_current_user)):
  36.     if await Movie.filter(pk=pk).delete():
  37.         return Response200()
  38.     return Response400(msg="删除失败")
  1. from tortoise import models
  2. from tortoise import fields
  5. class Movie(models.Model):
  6.     name = fields.CharField(max_length=50, null=False, description="电影名称")
  7.     year = fields.CharField(max_length=50, null=False, description="电影年份")
  1. from tortoise.contrib.pydantic import pydantic_model_creator
  3. from models import Movie
  5. Movie_Pydantic = pydantic_model_creator(Movie, name="Movie")
  6. MovieIn_Pydantic = pydantic_model_creator(Movie, name="MovieIn", exclude_readonly=True)