Skip to content

seeder

django_spire.contrib.seeding.model.django.seeder

TypeModel = TypeVar('TypeModel', bound=Model) module-attribute

DjangoModelSeeder

Bases: BaseModelSeeder

field_config_class = DjangoModelFieldsConfig class-attribute instance-attribute

model_class instance-attribute

prompt = None class-attribute instance-attribute

__init_subclass__ classmethod

Source code in django_spire/contrib/seeding/model/django/seeder.py 
34
35
36
37
38
39
40
41
42
@classmethod
def __init_subclass__(cls, **kwargs):
    super().__init_subclass__(**kwargs)

    if cls.model_class is None:
        raise ValueError("Seeds must have a model class")

    if cls.fields is None:
        raise ValueError("Seeds must have fields")

field_names classmethod

Source code in django_spire/contrib/seeding/model/django/seeder.py 
44
45
46
47
48
49
50
@classmethod
def field_names(cls) -> list[str]:
    # All foreign keys must be _id
    return [
        f.attname if isinstance(f, ForeignKey) else f.name
        for f in cls.model_class._meta.fields
    ]

seed_database classmethod

Source code in django_spire/contrib/seeding/model/django/seeder.py 
52
53
54
55
56
57
58
59
@classmethod
def seed_database(
        cls,
        count=1,
        fields: dict | None = None
) -> list[TypeModel]:
    model_objects = cls.seed(count, fields)
    return cls.model_class.objects.bulk_create(model_objects)