/tests/regressiontests/managers_regress/models.py
Python | 100 lines | 58 code | 29 blank | 13 comment | 0 complexity | 4058d229faaaabbe391dd1800c8f580a MD5 | raw file
Possible License(s): BSD-3-Clause
- """
- Various edge-cases for model managers.
- """
- from django.db import models
- class OnlyFred(models.Manager):
- def get_query_set(self):
- return super(OnlyFred, self).get_query_set().filter(name='fred')
- class OnlyBarney(models.Manager):
- def get_query_set(self):
- return super(OnlyBarney, self).get_query_set().filter(name='barney')
- class Value42(models.Manager):
- def get_query_set(self):
- return super(Value42, self).get_query_set().filter(value=42)
- class AbstractBase1(models.Model):
- name = models.CharField(max_length=50)
- class Meta:
- abstract = True
- # Custom managers
- manager1 = OnlyFred()
- manager2 = OnlyBarney()
- objects = models.Manager()
- class AbstractBase2(models.Model):
- value = models.IntegerField()
- class Meta:
- abstract = True
- # Custom manager
- restricted = Value42()
- # No custom manager on this class to make sure the default case doesn't break.
- class AbstractBase3(models.Model):
- comment = models.CharField(max_length=50)
- class Meta:
- abstract = True
- class Parent(models.Model):
- name = models.CharField(max_length=50)
- manager = OnlyFred()
- def __unicode__(self):
- return self.name
- # Managers from base classes are inherited and, if no manager is specified
- # *and* the parent has a manager specified, the first one (in the MRO) will
- # become the default.
- class Child1(AbstractBase1):
- data = models.CharField(max_length=25)
- def __unicode__(self):
- return self.data
- class Child2(AbstractBase1, AbstractBase2):
- data = models.CharField(max_length=25)
- def __unicode__(self):
- return self.data
- class Child3(AbstractBase1, AbstractBase3):
- data = models.CharField(max_length=25)
- def __unicode__(self):
- return self.data
- class Child4(AbstractBase1):
- data = models.CharField(max_length=25)
- # Should be the default manager, although the parent managers are
- # inherited.
- default = models.Manager()
- def __unicode__(self):
- return self.data
- class Child5(AbstractBase3):
- name = models.CharField(max_length=25)
- default = OnlyFred()
- objects = models.Manager()
- def __unicode__(self):
- return self.name
- # Will inherit managers from AbstractBase1, but not Child4.
- class Child6(Child4):
- value = models.IntegerField()
- # Will not inherit default manager from parent.
- class Child7(Parent):
- pass