← Back to branch summary

~ubuntu-branches/ubuntu/quantal/python-django/quantal

« back to all changes in this revision

Viewing changes to tests/modeltests/many_to_one_null/models.py

  • Committer: Bazaar Package Importer
  • Author(s): Scott James Remnant
  • Date: 2008-11-15 19:15:33 UTC
  • mfrom: (1.1.6 upstream)
  • Revision ID: james.westby@ubuntu.com-20081115191533-xbt1ut2xf4fvwtvc
Tags: 1.0.1-0ubuntu1
* New upstream release:
  - Bug fixes.

* The tests/ sub-directory appaers to have been dropped upstream, so pull
  our patch to workaround the tests and modify the rules.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/modeltests/many_to_one_null/models.py
1
 
"""
2
 
16. Many-to-one relationships that can be null
3
 
 
4
 
To define a many-to-one relationship that can have a null foreign key, use
5
 
``ForeignKey()`` with ``null=True`` .
6
 
"""
7
 
 
8
 
from django.db import models
9
 
 
10
 
class Reporter(models.Model):
11
 
    name = models.CharField(max_length=30)
12
 
 
13
 
    def __unicode__(self):
14
 
        return self.name
15
 
 
16
 
class Article(models.Model):
17
 
    headline = models.CharField(max_length=100)
18
 
    reporter = models.ForeignKey(Reporter, null=True)
19
 
 
20
 
    class Meta:
21
 
        ordering = ('headline',)
22
 
 
23
 
    def __unicode__(self):
24
 
        return self.headline
25
 
 
26
 
__test__ = {'API_TESTS':"""
27
 
# Create a Reporter.
28
 
>>> r = Reporter(name='John Smith')
29
 
>>> r.save()
30
 
 
31
 
# Create an Article.
32
 
>>> a = Article(headline="First", reporter=r)
33
 
>>> a.save()
34
 
 
35
 
>>> a.reporter.id
36
 
1
37
 
 
38
 
>>> a.reporter
39
 
<Reporter: John Smith>
40
 
 
41
 
# Article objects have access to their related Reporter objects.
42
 
>>> r = a.reporter
43
 
 
44
 
# Create an Article via the Reporter object.
45
 
>>> a2 = r.article_set.create(headline="Second")
46
 
>>> a2
47
 
<Article: Second>
48
 
>>> a2.reporter.id
49
 
1
50
 
 
51
 
# Reporter objects have access to their related Article objects.
52
 
>>> r.article_set.all()
53
 
[<Article: First>, <Article: Second>]
54
 
>>> r.article_set.filter(headline__startswith='Fir')
55
 
[<Article: First>]
56
 
>>> r.article_set.count()
57
 
2
58
 
 
59
 
# Create an Article with no Reporter by passing "reporter=None".
60
 
>>> a3 = Article(headline="Third", reporter=None)
61
 
>>> a3.save()
62
 
>>> a3.id
63
 
3
64
 
>>> print a3.reporter
65
 
None
66
 
 
67
 
# Need to reget a3 to refresh the cache
68
 
>>> a3 = Article.objects.get(pk=3)
69
 
>>> print a3.reporter.id
70
 
Traceback (most recent call last):
71
 
    ...
72
 
AttributeError: 'NoneType' object has no attribute 'id'
73
 
 
74
 
# Accessing an article's 'reporter' attribute returns None
75
 
# if the reporter is set to None.
76
 
>>> print a3.reporter
77
 
None
78
 
 
79
 
# To retrieve the articles with no reporters set, use "reporter__isnull=True".
80
 
>>> Article.objects.filter(reporter__isnull=True)
81
 
[<Article: Third>]
82
 
 
83
 
# We can achieve the same thing by filtering for the case where the reporter is
84
 
# None.
85
 
>>> Article.objects.filter(reporter=None)
86
 
[<Article: Third>]
87
 
 
88
 
# Set the reporter for the Third article
89
 
>>> r.article_set.add(a3)
90
 
>>> r.article_set.all()
91
 
[<Article: First>, <Article: Second>, <Article: Third>]
92
 
 
93
 
# Remove an article from the set, and check that it was removed.
94
 
>>> r.article_set.remove(a3)
95
 
>>> r.article_set.all()
96
 
[<Article: First>, <Article: Second>]
97
 
>>> Article.objects.filter(reporter__isnull=True)
98
 
[<Article: Third>]
99
 
 
100
 
# Create another article and reporter
101
 
>>> r2 = Reporter(name='Paul Jones')
102
 
>>> r2.save()
103
 
>>> a4 = r2.article_set.create(headline='Fourth')
104
 
>>> r2.article_set.all()
105
 
[<Article: Fourth>]
106
 
 
107
 
# Try to remove a4 from a set it does not belong to
108
 
>>> r.article_set.remove(a4)
109
 
Traceback (most recent call last):
110
 
...
111
 
DoesNotExist: <Article: Fourth> is not related to <Reporter: John Smith>.
112
 
 
113
 
>>> r2.article_set.all()
114
 
[<Article: Fourth>]
115
 
 
116
 
# Use descriptor assignment to allocate ForeignKey. Null is legal, so
117
 
# existing members of set that are not in the assignment set are set null
118
 
>>> r2.article_set = [a2, a3]
119
 
>>> r2.article_set.all()
120
 
[<Article: Second>, <Article: Third>]
121
 
 
122
 
# Clear the rest of the set
123
 
>>> r.article_set.clear()
124
 
>>> r.article_set.all()
125
 
[]
126
 
>>> Article.objects.filter(reporter__isnull=True)
127
 
[<Article: First>, <Article: Fourth>]
128
 
 
129
 
"""}