waylon/learn-sql-model
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add pet

Browse source
This commit is contained in:
Waylon Walker 2023-05-22 08:28:00 -05:00
parent 3772d7b023
commit 99908c6a5a
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
2 changed files with 22 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
learn_sql_model/factories/pet.py Normal file
View file

@ -0,0 +1,16 @@
from polyfactory.factories.pydantic_factory import ModelFactory
from learn_sql_model.models.hero import Hero
from learn_sql_model.models.pet import Pet
__relationship__ = [Hero]
class PetFactory(ModelFactory[Pet]):
__model__ = Pet
__set_as_default_factory_for_type__ = True
id = None
@classmethod
def name(cls) -> str:
return cls.__faker__.first_name()

14
learn_sql_model/models/pet.py
View file

@ -1,16 +1,14 @@
from typing import Optional from typing import Optional, TYPE_CHECKING
from sqlmodel import Field from sqlmodel import Field, Relationship
from learn_sql_model.models.fast_model import FastModel from learn_sql_model.models.fast_model import FastModel
if TYPE_CHECKING:
from learn_sql_model.models.hero import Hero
class Pet(FastModel, table=True): class Pet(FastModel, table=True):
id: Optional[int] = Field(default=None, primary_key=True) id: Optional[int] = Field(default=None, primary_key=True)
name: str = "Jim" name: str = "Jim"
hero: "Hero" = Relationship(back_populates="pet")
# age: Optional[int] = None
# hero_id: int = Field(default=None, foreign_key="hero.id")
# hero: Optional[Hero] = Relationship(back_populates="pets")