在FastApi中简单封装Tortoise-orm的CRUD操作

我们可以看一下官网给的例子

@app.get("/users", response_model=List[User_Pydantic])
async def get_users():
    return await User_Pydantic.from_queryset(Users.all())


@app.post("/users", response_model=User_Pydantic)
async def create_user(user: UserIn_Pydantic):
    user_obj = await Users.create(**user.dict(exclude_unset=True))
    return await User_Pydantic.from_tortoise_orm(user_obj)

为提高代码复用性，我们可以把Tortoise-orm的CRUD操作像下面这样封装起来

async def get_users(user_name: str):
    return await User.filter(user_name__contains=user_name).values()

这就遇到了一个问题，上述方法返回的是一个list对象，response_model无法对list对象做级联查询，此时我们可以这样做

async def get_users(user_name: str):
    return await User.filter(user_name__contains=user_name)

这样就返回了一个tortoise.queryset.QuerySet对象，和sqlalchemy类似，response_model可以自动做级联查询，但又遇到一个问题就是TypeError: 'QuerySet' object is not iterable，QuerySet不可迭代，这又在操作上做了很多限制，失去了封装的意义，在查询资料的过程中发现传递QuerySet不会真实的操作数据库，那么封装的方法就不必使用异步了，所以可以这样

def get_users(user_name: str):
    return User.filter(user_name__contains=user_name)

然后这样调用就可以了

list_object = await get_users(user_name).values()
query_set_object = await get_users(user_name)

具体代码可以参考https://github.com/chdxia/lrtest-api

标签：users,get,FastApi,await,CRUD,name,user,User,Tortoise
来源： https://www.cnblogs.com/chdxia/p/16525291.html