Skip to content

override

django_spire.contrib.seeding.field.override

FieldOverride

Source code in django_spire/contrib/seeding/field/override.py 
2
3
4
def __init__(self, seeder_class):
    self.seeder_class = seeder_class
    self.overrides = {}

seeder_class = seeder_class instance-attribute

overrides = {} instance-attribute

__getattr__

Delegate attribute lookup to self.seeder_class if the attribute isn't found in this FieldOverride instance.

Source code in django_spire/contrib/seeding/field/override.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
def __getattr__(self, name):
    """
    Delegate attribute lookup to self.seeder_class if the attribute isn't found
    in this FieldOverride instance.
    """

    attr = getattr(self.seeder_class, name)
    if callable(attr):
        def wrapper(*args, **kwargs):
            # TODO: Error here if fields is passed as an arg.
            # Pass the overrides field to the seed method
            if 'fields' in kwargs and isinstance(kwargs['fields'], dict):
                kwargs['fields'] = {**kwargs['fields'], **self.overrides}
            else:
                kwargs['fields'] = self.overrides

            return attr(*args, **kwargs)

        return wrapper
    return attr

filter

Source code in django_spire/contrib/seeding/field/override.py 
27
28
29
def filter(self, **kwargs):
    self.overrides.update(kwargs)
    return self

seed

Source code in django_spire/contrib/seeding/field/override.py 
31
32
def seed(self, count=1):
    return self.seeder_class.seed(count=count, fields=self.overrides)