diff options
Diffstat (limited to 'tests/custom_managers_regress/models.py')
-rw-r--r-- | tests/custom_managers_regress/models.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/tests/custom_managers_regress/models.py b/tests/custom_managers_regress/models.py new file mode 100644 index 0000000000..71073f0fe7 --- /dev/null +++ b/tests/custom_managers_regress/models.py @@ -0,0 +1,45 @@ +""" +Regression tests for custom manager classes. +""" + +from django.db import models +from django.utils.encoding import python_2_unicode_compatible + + +class RestrictedManager(models.Manager): + """ + A manager that filters out non-public instances. + """ + def get_query_set(self): + return super(RestrictedManager, self).get_query_set().filter(is_public=True) + +@python_2_unicode_compatible +class RelatedModel(models.Model): + name = models.CharField(max_length=50) + + def __str__(self): + return self.name + +@python_2_unicode_compatible +class RestrictedModel(models.Model): + name = models.CharField(max_length=50) + is_public = models.BooleanField(default=False) + related = models.ForeignKey(RelatedModel) + + objects = RestrictedManager() + plain_manager = models.Manager() + + def __str__(self): + return self.name + +@python_2_unicode_compatible +class OneToOneRestrictedModel(models.Model): + name = models.CharField(max_length=50) + is_public = models.BooleanField(default=False) + related = models.OneToOneField(RelatedModel) + + objects = RestrictedManager() + plain_manager = models.Manager() + + def __str__(self): + return self.name |