summaryrefslogtreecommitdiff
path: root/tests/generic_relations/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/generic_relations/models.py')
-rw-r--r--tests/generic_relations/models.py24
1 files changed, 14 insertions, 10 deletions
diff --git a/tests/generic_relations/models.py b/tests/generic_relations/models.py
index eec40e87a4..e99d2c7e5e 100644
--- a/tests/generic_relations/models.py
+++ b/tests/generic_relations/models.py
@@ -9,15 +9,14 @@ The canonical example is tags (although this example implementation is *far*
from complete).
"""
-from django.contrib.contenttypes.fields import (
- GenericForeignKey, GenericRelation,
-)
+from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
from django.contrib.contenttypes.models import ContentType
from django.db import models
class TaggedItem(models.Model):
"""A tag on an item."""
+
tag = models.SlugField()
content_type = models.ForeignKey(ContentType, models.CASCADE)
object_id = models.PositiveIntegerField()
@@ -38,7 +37,9 @@ class ValuableTaggedItem(TaggedItem):
class AbstractComparison(models.Model):
comparative = models.CharField(max_length=50)
- content_type1 = models.ForeignKey(ContentType, models.CASCADE, related_name="comparative1_set")
+ content_type1 = models.ForeignKey(
+ ContentType, models.CASCADE, related_name="comparative1_set"
+ )
object_id1 = models.PositiveIntegerField()
first_obj = GenericForeignKey(ct_field="content_type1", fk_field="object_id1")
@@ -49,7 +50,10 @@ class Comparison(AbstractComparison):
A model that tests having multiple GenericForeignKeys. One is defined
through an inherited abstract model and one defined directly on this class.
"""
- content_type2 = models.ForeignKey(ContentType, models.CASCADE, related_name="comparative2_set")
+
+ content_type2 = models.ForeignKey(
+ ContentType, models.CASCADE, related_name="comparative2_set"
+ )
object_id2 = models.PositiveIntegerField()
other_obj = GenericForeignKey(ct_field="content_type2", fk_field="object_id2")
@@ -62,10 +66,10 @@ class Animal(models.Model):
common_name = models.CharField(max_length=150)
latin_name = models.CharField(max_length=150)
- tags = GenericRelation(TaggedItem, related_query_name='animal')
- comparisons = GenericRelation(Comparison,
- object_id_field="object_id1",
- content_type_field="content_type1")
+ tags = GenericRelation(TaggedItem, related_query_name="animal")
+ comparisons = GenericRelation(
+ Comparison, object_id_field="object_id1", content_type_field="content_type1"
+ )
def __str__(self):
return self.common_name
@@ -116,7 +120,7 @@ class ValuableRock(Mineral):
class ManualPK(models.Model):
id = models.IntegerField(primary_key=True)
- tags = GenericRelation(TaggedItem, related_query_name='manualpk')
+ tags = GenericRelation(TaggedItem, related_query_name="manualpk")
class ForProxyModelModel(models.Model):