example_related_manager.py

example_related_manager.py

class ConfigValueManager(models.Manager):

    def get(self, key):
        config_value = self.filter(key=key).first()
        if config_value:
            type_caster = locate(config_value.type)
            return type_caster(config_value.value)
        return config_value

    def set(self, key, value):
        self.filter(key=key).update(value=value)

    def set2(self, key, value):
        qs = self.filter(key=key)
        if qs:
            qs.update(value=value, type=type(value).__name__, company=self.instance)
        else:
            self.create(key=key, value=value, type=type(value).__name__, company=self.instance)
            

class FooConfigValue(models.Model):
    key = models.CharField(max_length=255, db_index=True)
    value = models.CharField(max_length=255, db_index=True)
    type = models.CharField(max_length=255, db_index=True)

    objects = ConfigValueManager()

    class Meta:
        abstract = True
        
        
def test_saving_company_config_key_value_on_related_manager(self):
    company = CompanyFactory.create()
    company.config_values.set('foo', 123)
    self.assertEquals(company.config_values.get('foo'), 123)