summaryrefslogtreecommitdiff
path: root/tests/proxy_models/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/proxy_models/models.py')
-rw-r--r--tests/proxy_models/models.py169
1 files changed, 169 insertions, 0 deletions
diff --git a/tests/proxy_models/models.py b/tests/proxy_models/models.py
new file mode 100644
index 0000000000..6c962aadc8
--- /dev/null
+++ b/tests/proxy_models/models.py
@@ -0,0 +1,169 @@
+"""
+By specifying the 'proxy' Meta attribute, model subclasses can specify that
+they will take data directly from the table of their base class table rather
+than using a new table of their own. This allows them to act as simple proxies,
+providing a modified interface to the data from the base class.
+"""
+from django.db import models
+from django.utils.encoding import python_2_unicode_compatible
+
+# A couple of managers for testing managing overriding in proxy model cases.
+
+class PersonManager(models.Manager):
+ def get_query_set(self):
+ return super(PersonManager, self).get_query_set().exclude(name="fred")
+
+class SubManager(models.Manager):
+ def get_query_set(self):
+ return super(SubManager, self).get_query_set().exclude(name="wilma")
+
+@python_2_unicode_compatible
+class Person(models.Model):
+ """
+ A simple concrete base class.
+ """
+ name = models.CharField(max_length=50)
+
+ objects = PersonManager()
+
+ def __str__(self):
+ return self.name
+
+class Abstract(models.Model):
+ """
+ A simple abstract base class, to be used for error checking.
+ """
+ data = models.CharField(max_length=10)
+
+ class Meta:
+ abstract = True
+
+class MyPerson(Person):
+ """
+ A proxy subclass, this should not get a new table. Overrides the default
+ manager.
+ """
+ class Meta:
+ proxy = True
+ ordering = ["name"]
+ permissions = (
+ ("display_users", "May display users information"),
+ )
+
+ objects = SubManager()
+ other = PersonManager()
+
+ def has_special_name(self):
+ return self.name.lower() == "special"
+
+class ManagerMixin(models.Model):
+ excluder = SubManager()
+
+ class Meta:
+ abstract = True
+
+class OtherPerson(Person, ManagerMixin):
+ """
+ A class with the default manager from Person, plus an secondary manager.
+ """
+ class Meta:
+ proxy = True
+ ordering = ["name"]
+
+class StatusPerson(MyPerson):
+ """
+ A non-proxy subclass of a proxy, it should get a new table.
+ """
+ status = models.CharField(max_length=80)
+
+# We can even have proxies of proxies (and subclass of those).
+class MyPersonProxy(MyPerson):
+ class Meta:
+ proxy = True
+
+class LowerStatusPerson(MyPersonProxy):
+ status = models.CharField(max_length=80)
+
+@python_2_unicode_compatible
+class User(models.Model):
+ name = models.CharField(max_length=100)
+
+ def __str__(self):
+ return self.name
+
+class UserProxy(User):
+ class Meta:
+ proxy = True
+
+class UserProxyProxy(UserProxy):
+ class Meta:
+ proxy = True
+
+# We can still use `select_related()` to include related models in our querysets.
+class Country(models.Model):
+ name = models.CharField(max_length=50)
+
+@python_2_unicode_compatible
+class State(models.Model):
+ name = models.CharField(max_length=50)
+ country = models.ForeignKey(Country)
+
+ def __str__(self):
+ return self.name
+
+class StateProxy(State):
+ class Meta:
+ proxy = True
+
+# Proxy models still works with filters (on related fields)
+# and select_related, even when mixed with model inheritance
+class BaseUser(models.Model):
+ name = models.CharField(max_length=255)
+
+class TrackerUser(BaseUser):
+ status = models.CharField(max_length=50)
+
+class ProxyTrackerUser(TrackerUser):
+ class Meta:
+ proxy = True
+
+
+@python_2_unicode_compatible
+class Issue(models.Model):
+ summary = models.CharField(max_length=255)
+ assignee = models.ForeignKey(TrackerUser)
+
+ def __str__(self):
+ return ':'.join((self.__class__.__name__,self.summary,))
+
+class Bug(Issue):
+ version = models.CharField(max_length=50)
+ reporter = models.ForeignKey(BaseUser)
+
+class ProxyBug(Bug):
+ """
+ Proxy of an inherited class
+ """
+ class Meta:
+ proxy = True
+
+
+class ProxyProxyBug(ProxyBug):
+ """
+ A proxy of proxy model with related field
+ """
+ class Meta:
+ proxy = True
+
+class Improvement(Issue):
+ """
+ A model that has relation to a proxy model
+ or to a proxy of proxy model
+ """
+ version = models.CharField(max_length=50)
+ reporter = models.ForeignKey(ProxyTrackerUser)
+ associated_bug = models.ForeignKey(ProxyProxyBug)
+
+class ProxyImprovement(Improvement):
+ class Meta:
+ proxy = True