Skip to content

custom

django_spire.contrib.seeding.field.custom

CustomFieldSeeder

Bases: BaseFieldSeeder

Source code in django_spire/contrib/seeding/field/base.py 
11
12
13
14
15
16
17
18
def __init__(
        self,
        fields: dict = None,
        default_to: str = "llm"
):

    self.fields = self._normalize_fields(fields or {})
    self.default_to = default_to

keyword = FieldSeederTypesEnum.CUSTOM class-attribute instance-attribute

in_order

Source code in django_spire/contrib/seeding/field/custom.py 
11
12
13
14
15
16
17
18
def in_order(self, values: list, index: int) -> any:
    if not values:
        raise ValueError(
            "Cannot select from empty values list. "
            "Make sure the related model has existing records before seeding foreign keys."
        )
    index = index % len(values)  # Index loops back on itself
    return values[index]

date_time_between

Source code in django_spire/contrib/seeding/field/custom.py 
20
21
22
23
def date_time_between(self, start_date: str, end_date: str):
    faker = Faker()
    naive_dt = faker.date_time_between(start_date=start_date, end_date=end_date)
    return timezone.make_aware(naive_dt)

fk_random

Source code in django_spire/contrib/seeding/field/custom.py 
25
26
27
def fk_random(self, model_class, ids: list[int]):
    faker = Faker()
    return faker.random_element(elements=ids)

fk_in_order

Takes a queryset, calls it then returns a random id

Source code in django_spire/contrib/seeding/field/custom.py 
29
30
31
def fk_in_order(self, model_class, index: int, ids: list[int]):
    """Takes a queryset, calls it then returns a random id"""
    return self.in_order(values=ids, index=index)

seed

Source code in django_spire/contrib/seeding/field/custom.py 
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
def seed(self, manager, count) -> list[dict]:
    data = []

    foreign_keys = {}

    for i in range(count):
        row = {}

        for field_name, config in self.seeder_fields.items():
            method_name = config[1] if len(config) > 1 else field_name
            kwargs = config[2] if len(config) > 2 else {}
            method = getattr(self, method_name, None)

            if not callable(method):
                message = f"Custom method '{method_name}' not found for field '{field_name}'"
                raise TypeError(message)

            if method_name in ['in_order', 'fk_in_order']:
                kwargs["index"] = i

            if method_name in ['fk_random', 'fk_in_order']:
                model_class = kwargs["model_class"]

                if model_class not in foreign_keys:
                    foreign_keys[model_class] = list(model_class.objects.all().values_list('id', flat=True))

                kwargs['ids'] = foreign_keys[model_class]

            row[field_name] = method(**kwargs)

        data.append(row)

    return data