python/lib/django_1_4/tests/regressiontests/managers_regress/models.py

   1 """
   2 Various edge-cases for model managers.
   3 """
   4
   5 from django.db import models
   6
   7
   8 class OnlyFred(models.Manager):
   9     def get_query_set(self):
  10         return super(OnlyFred, self).get_query_set().filter(name='fred')
  11
  12 class OnlyBarney(models.Manager):
  13     def get_query_set(self):
  14         return super(OnlyBarney, self).get_query_set().filter(name='barney')
  15
  16 class Value42(models.Manager):
  17     def get_query_set(self):
  18         return super(Value42, self).get_query_set().filter(value=42)
  19
  20 class AbstractBase1(models.Model):
  21     name = models.CharField(max_length=50)
  22
  23     class Meta:
  24         abstract = True
  25
  26     # Custom managers
  27     manager1 = OnlyFred()
  28     manager2 = OnlyBarney()
  29     objects = models.Manager()
  30
  31 class AbstractBase2(models.Model):
  32     value = models.IntegerField()
  33
  34     class Meta:
  35         abstract = True
  36
  37     # Custom manager
  38     restricted = Value42()
  39
  40 # No custom manager on this class to make sure the default case doesn't break.
  41 class AbstractBase3(models.Model):
  42     comment = models.CharField(max_length=50)
  43
  44     class Meta:
  45         abstract = True
  46
  47 class Parent(models.Model):
  48     name = models.CharField(max_length=50)
  49
  50     manager = OnlyFred()
  51
  52     def __unicode__(self):
  53         return self.name
  54
  55 # Managers from base classes are inherited and, if no manager is specified
  56 # *and* the parent has a manager specified, the first one (in the MRO) will
  57 # become the default.
  58 class Child1(AbstractBase1):
  59     data = models.CharField(max_length=25)
  60
  61     def __unicode__(self):
  62         return self.data
  63
  64 class Child2(AbstractBase1, AbstractBase2):
  65     data = models.CharField(max_length=25)
  66
  67     def __unicode__(self):
  68         return self.data
  69
  70 class Child3(AbstractBase1, AbstractBase3):
  71     data = models.CharField(max_length=25)
  72
  73     def __unicode__(self):
  74         return self.data
  75
  76 class Child4(AbstractBase1):
  77     data = models.CharField(max_length=25)
  78
  79     # Should be the default manager, although the parent managers are
  80     # inherited.
  81     default = models.Manager()
  82
  83     def __unicode__(self):
  84         return self.data
  85
  86 class Child5(AbstractBase3):
  87     name = models.CharField(max_length=25)
  88
  89     default = OnlyFred()
  90     objects = models.Manager()
  91
  92     def __unicode__(self):
  93         return self.name
  94
  95 # Will inherit managers from AbstractBase1, but not Child4.
  96 class Child6(Child4):
  97     value = models.IntegerField()
  98
  99 # Will not inherit default manager from parent.
 100 class Child7(Parent):
 101     pass