126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
from django.db import models
|
|
from django.contrib.auth import models as auth
|
|
|
|
# No related name is needed here, since symmetrical relations are not
|
|
# explicitly reversible.
|
|
class SelfRefer(models.Model):
|
|
name = models.CharField(max_length=10)
|
|
references = models.ManyToManyField('self')
|
|
related = models.ManyToManyField('self')
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
class Tag(models.Model):
|
|
name = models.CharField(max_length=10)
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
# Regression for #11956 -- a many to many to the base class
|
|
class TagCollection(Tag):
|
|
tags = models.ManyToManyField(Tag, related_name='tag_collections')
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
# A related_name is required on one of the ManyToManyField entries here because
|
|
# they are both addressable as reverse relations from Tag.
|
|
class Entry(models.Model):
|
|
name = models.CharField(max_length=10)
|
|
topics = models.ManyToManyField(Tag)
|
|
related = models.ManyToManyField(Tag, related_name="similar")
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
# Two models both inheriting from a base model with a self-referential m2m field
|
|
class SelfReferChild(SelfRefer):
|
|
pass
|
|
|
|
class SelfReferChildSibling(SelfRefer):
|
|
pass
|
|
|
|
# Many-to-Many relation between models, where one of the PK's isn't an Autofield
|
|
class Line(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
|
|
class Worksheet(models.Model):
|
|
id = models.CharField(primary_key=True, max_length=100)
|
|
lines = models.ManyToManyField(Line, blank=True, null=True)
|
|
|
|
# Regression for #11226 -- A model with the same name that another one to
|
|
# which it has a m2m relation. This shouldn't cause a name clash between
|
|
# the automatically created m2m intermediary table FK field names when
|
|
# running syncdb
|
|
class User(models.Model):
|
|
name = models.CharField(max_length=30)
|
|
friends = models.ManyToManyField(auth.User)
|
|
|
|
__test__ = {"regressions": """
|
|
# Multiple m2m references to the same model or a different model must be
|
|
# distinguished when accessing the relations through an instance attribute.
|
|
|
|
>>> s1 = SelfRefer.objects.create(name='s1')
|
|
>>> s2 = SelfRefer.objects.create(name='s2')
|
|
>>> s3 = SelfRefer.objects.create(name='s3')
|
|
>>> s1.references.add(s2)
|
|
>>> s1.related.add(s3)
|
|
|
|
>>> e1 = Entry.objects.create(name='e1')
|
|
>>> t1 = Tag.objects.create(name='t1')
|
|
>>> t2 = Tag.objects.create(name='t2')
|
|
>>> e1.topics.add(t1)
|
|
>>> e1.related.add(t2)
|
|
|
|
>>> s1.references.all()
|
|
[<SelfRefer: s2>]
|
|
>>> s1.related.all()
|
|
[<SelfRefer: s3>]
|
|
|
|
>>> e1.topics.all()
|
|
[<Tag: t1>]
|
|
>>> e1.related.all()
|
|
[<Tag: t2>]
|
|
|
|
# The secret internal related names for self-referential many-to-many fields
|
|
# shouldn't appear in the list when an error is made.
|
|
>>> SelfRefer.objects.filter(porcupine='fred')
|
|
Traceback (most recent call last):
|
|
...
|
|
FieldError: Cannot resolve keyword 'porcupine' into field. Choices are: id, name, references, related, selfreferchild, selfreferchildsibling
|
|
|
|
# Test to ensure that the relationship between two inherited models
|
|
# with a self-referential m2m field maintains symmetry
|
|
>>> sr_child = SelfReferChild(name="Hanna")
|
|
>>> sr_child.save()
|
|
|
|
>>> sr_sibling = SelfReferChildSibling(name="Beth")
|
|
>>> sr_sibling.save()
|
|
>>> sr_child.related.add(sr_sibling)
|
|
>>> sr_child.related.all()
|
|
[<SelfRefer: Beth>]
|
|
>>> sr_sibling.related.all()
|
|
[<SelfRefer: Hanna>]
|
|
|
|
# Regression for #11311 - The primary key for models in a m2m relation
|
|
# doesn't have to be an AutoField
|
|
>>> w = Worksheet(id='abc')
|
|
>>> w.save()
|
|
>>> w.delete()
|
|
|
|
# Regression for #11956 -- You can add an object to a m2m with the
|
|
# base class without causing integrity errors
|
|
>>> c1 = TagCollection.objects.create(name='c1')
|
|
>>> c1.tags = [t1,t2]
|
|
|
|
>>> c1 = TagCollection.objects.get(name='c1')
|
|
>>> c1.tags.all()
|
|
[<Tag: t1>, <Tag: t2>]
|
|
|
|
>>> t1.tag_collections.all()
|
|
[<TagCollection: c1>]
|
|
|
|
"""
|
|
}
|