diff --git a/tests/testapp/models/__init__.py b/tests/testapp/models/__init__.py index a8aeada597..a5a41035d6 100644 --- a/tests/testapp/models/__init__.py +++ b/tests/testapp/models/__init__.py @@ -2,4 +2,4 @@ __all__ = ['basic', 'repr', 'custom_methods', 'many_to_one', 'many_to_many', 'ordering', 'lookup', 'get_latest', 'm2m_intermediary', 'one_to_one', 'm2o_recursive', 'm2o_recursive2', 'save_delete_hooks', 'custom_pk', 'subclassing', 'many_to_one_null', 'custom_columns', 'reserved_names', - 'or_lookups'] + 'or_lookups', 'm2m_multiple'] diff --git a/tests/testapp/models/m2m_multiple.py b/tests/testapp/models/m2m_multiple.py new file mode 100644 index 0000000000..d8793acb73 --- /dev/null +++ b/tests/testapp/models/m2m_multiple.py @@ -0,0 +1,99 @@ +""" +20. Multiple many-to-many relationships between the same two tables + +In this example, an Article can have many Categories (as "primary") and many +Categories (as "secondary"). + +Set ``related_name`` to designate what the reverse relationship is called. + +Set ``singular`` to designate what the category object is called. This is +required if a model has multiple ``ManyToManyFields`` to the same object. +""" + +from django.core import meta + +class Category(meta.Model): + name = meta.CharField(maxlength=20) + class META: + module_name = 'categories' + ordering = ('name',) + + def __repr__(self): + return self.name + +class Article(meta.Model): + headline = meta.CharField(maxlength=50) + pub_date = meta.DateTimeField() + primary_categories = meta.ManyToManyField(Category, + singular='primary_category', related_name='primary_article') + secondary_categories = meta.ManyToManyField(Category, + singular='secondary_category', related_name='secondary_article') + class META: + ordering = ('pub_date',) + + def __repr__(self): + return self.headline + +API_TESTS = """ +>>> from datetime import datetime + +>>> c1 = categories.Category(name='Sports') +>>> c1.save() +>>> c2 = categories.Category(name='News') +>>> c2.save() +>>> c3 = categories.Category(name='Crime') +>>> c3.save() +>>> c4 = categories.Category(name='Life') +>>> c4.save() + +>>> a1 = articles.Article(headline='Area man steals', pub_date=datetime(2005, 11, 27)) +>>> a1.save() +>>> a1.set_primary_categories([c2.id, c3.id]) +True +>>> a1.set_secondary_categories([c4.id]) +True + +>>> a2 = articles.Article(headline='Area man runs', pub_date=datetime(2005, 11, 28)) +>>> a2.save() +>>> a2.set_primary_categories([c1.id, c2.id]) +True +>>> a2.set_secondary_categories([c4.id]) +True + +# The "primary_category" here comes from the "singular" parameter. If we hadn't +# specified the "singular" parameter, Django would just use "category", which +# would cause a conflict because the "primary_categories" and +# "secondary_categories" fields both relate to Category. +>>> a1.get_primary_category_list() +[Crime, News] + +# Ditto for the "primary_category" here. +>>> a2.get_primary_category_list() +[News, Sports] + +# Ditto for the "secondary_category" here. +>>> a1.get_secondary_category_list() +[Life] + +# Ditto for the "secondary_category" here. +>>> a2.get_secondary_category_list() +[Life] + + +>>> c1.get_primary_article_list() +[Area man runs] +>>> c1.get_secondary_article_list() +[] +>>> c2.get_primary_article_list() +[Area man steals, Area man runs] +>>> c2.get_secondary_article_list() +[] +>>> c3.get_primary_article_list() +[Area man steals] +>>> c3.get_secondary_article_list() +[] +>>> c4.get_primary_article_list() +[] +>>> c4.get_secondary_article_list() +[Area man steals, Area man runs] +"""