FastAPI+Tortoise ORM+FastAPI用户(Python)-关系-多对多



我正在使用FastAPI、Tortoise ORM和FastAPI用户制作一个API来学习。基本上,我从OpenWeather API获取城市天气数据,并使用Tortoise ORM存储在SQLite数据库中。我有FastAPI用户的身份验证。

我用的是外国钥匙,关系很好。但现在我想要一些改变(改善(。我希望每个用户都有自己的项目,并且在登录并访问端点(@router.get("/me/onlyitems/", response_model=List[schemas.Card_Pydantic])(后,它只接收他们的项目。

它运行得很好,我只接收用户项目(使用owner_id(,但我的项目表使用了一个外部唯一id(项目是一个城市,每个城市都有一个唯一id(,它是主键。但是,当另一个用户尝试添加同一个城市时,它会出现错误,因为即使与另一个所有者在一起,它也不能在表中具有同一城市。好吧,这是应该的……但我想让其他用户访问相同的城市(他们有相同的信息,他们不是定制的(。然后,我希望许多用户能够访问同一项目(并且每个用户都可以访问多个项目(。

我不认为为每个用户重复项目信息(使用另一个ID(是理想的,最好是不同的用户访问相同的项目。

我认为这将是一个多对多的关系,但我不知道如何建立这种关系并使用FastAPI获得它。

我想过创建一个中间表,只使用UserID和CityID(以及一个自动递增ID(,并使用它来获取相关数据,但我不知道这是最好的方法还是可能的。

我试图从Tortoise文档中学习,但我找不到我需要的部分,也无法理解文档中一些复杂的示例和部分。

我使用的是更新版本的软件包和Python 3.9.4

好吧,让我展示一些代码:

型号.py

class UserModel(TortoiseBaseUserModel):  # From FastAPI Users
cards: fields.ReverseRelation["Card"]

class OAuthAccountModel(TortoiseBaseOAuthAccountModel): # From FastAPI Users
user = fields.ForeignKeyField("models.UserModel", related_name="oauth_accounts")

class User(models.BaseUser, models.BaseOAuthAccountMixin): # From FastAPI Users
pass

class UserCreate(models.BaseUserCreate): # From FastAPI Users
pass

class UserUpdate(User, models.BaseUserUpdate): # From FastAPI Users
pass

class UserDB(User, models.BaseUserDB, PydanticModel): # From FastAPI Users
class Config:
orm_mode = True
orig_model = UserModel

user_db = TortoiseUserDatabase(UserDB, UserModel, OAuthAccountModel) # From FastAPI Users

class Card(tmodels.Model):
"""
The Card model, related to the user.
"""
id = fields.IntField(pk=True)
city_name = fields.CharField(max_length=30)
temperature = fields.FloatField()
state = fields.CharField(max_length=30, null=True)
icon = fields.CharField(max_length=3, null=True)
period = fields.CharField(max_length=20, null=True)
created_at = fields.DatetimeField(auto_now_add=True)
updated_at = fields.DatetimeField(auto_now=True)
owner: fields.ForeignKeyRelation[UserModel] = fields.ForeignKeyField(
"models.UserModel", related_name="cards")
class PydanticMeta:
exclude = ["owner"]

我试图更改我的所有者字段以及它们和UserModel:的关系

class Card(tmodels.Model):
owner = fields.ManyToManyField("models.UserModel", related_name="cards")
class UserModel(TortoiseBaseUserModel):
cards: fields.ManyToManyRelation["Card"]

我不知道这是否正确,我无法让它发挥作用,但问题可能在其他地方。

我的用户端点创建一个项目:

@router.post("/", response_model=Card_Pydantic)
async def create_item_for_current_user(
card: CardCreate, user: UserDB = Depends(fastapi_users.get_current_active_user)):
card_obj = await crud.create_user_card(card=card, user=user)
if card_obj is None:
raise HTTPException(status_code=404, detail=f"City '{card.city_name}' not found")
return await card_obj

此处使用的CRUD函数(用于创建项目(:

async def create_user_card(
card: schemas.CardCreate, user: UserDB = Depends(fastapi_users.get_current_active_user)):
card_json = await weather_api.get_api_info_by_name(card.city_name)
if card_json is None:
return None
card_obj = await Card.create(**card_json,
owner_id=user.id)  # ** is used to pass the keys values as kwargs.
return await card_obj

我的端点以获取用户拥有的项目:

@router.get("/me/onlyitems/", response_model=List[schemas.Card_Pydantic])
async def read_current_user_items(user: UserDB = Depends(fastapi_users.get_current_active_user)):
user_cards = await schemas.Card_Pydantic.from_queryset(Card.filter(owner_id=user.id).all())
return await crud.update_only_card_for(user_cards, user.id)

我的crud函数是:

async def update_only_card_for(user_cards, owner_id):
for city in user_cards:
await update_card(city.id, owner_id)
return await schemas.Card_Pydantic.from_queryset(Card.filter(owner_id=owner_id).all())
# The update function (it will check if it need to update or not)
async def update_card(city_id: int, owner_id: UUID = UserDB):
card_obj = await Card.get(id=city_id)
card_time = card_obj.updated_at
timezone = card_time.tzinfo
now = datetime.now(timezone)
time_to_update = now - timedelta(minutes=15)
if card_time < time_to_update:
card_json = await weather_api.get_api_info_by_id(city_id)
await card_obj.update_from_dict(card_json).save()  # with save will update the update_at field
card_updated = await schemas.Card_Pydantic.from_queryset_single(Card.get(id=city_id))
return card_updated

我如何更改它以获得用户访问所需的卡?我想我不能使用owner_id的过滤器(好吧,如果我能创建我的中间表,我可以,但不知道如何做到这一点并获取数据(,但我不知道如何建立关系,将用户链接到他添加的项目,同时其他用户访问相同的项目。

这段代码运行良好(我使用的是nuxt.js前端(,直到出现重复的城市。。。

处理它的最佳方法是什么?如何处理?

谢谢,迭戈。

我解决了这个问题。我认为Tortoise文档有点混乱,但我找到了如何创建关系和检索数据。

型号:(简化版(

class Card(tmodels.Model):
id = fields.IntField(pk=True)
city_name = fields.CharField(max_length=30)
owners: fields.ManyToManyRelation["UserModel"] = fields.ManyToManyField(
"models.UserModel", related_name="cards", through="card_usermodel")
class UserModel(TortoiseBaseUserModel):
cards: fields.ManyToManyRelation[Card]

through="card_usermodel"将创建一个具有卡id和所有者id的中间表,它将定义关系。

进行数据操作:

async def create_user_card(
card: schemas.CardCreate, user: UserDB = Depends(fastapi_users.get_current_active_user)):
card_json = await weather_api.get_api_info_by_name(card.city_name)
if card_json is None:
return None
owner = await UserModel.get_or_none(id=user.id)
card_obj = await Card.update_or_create(**card_json)
await card_obj[0].owners.add(owner)
return await schemas.Card_Response_Pydantic.from_queryset_single(Card.get(id=card_json.get("id"), owners=user.id))

这是基本的。我也更改了一些Pydantic模型,以验证响应。

最新更新