add pet
This commit is contained in:
parent
3772d7b023
commit
99908c6a5a
2 changed files with 22 additions and 8 deletions
16
learn_sql_model/factories/pet.py
Normal file
16
learn_sql_model/factories/pet.py
Normal 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()
|
||||
Loading…
Add table
Add a link
Reference in a new issue