Merge branch 'master' into penfield, cleaned up.
[philo.git] / contrib / penfield / models.py
1 from django.db import models
2 from django.conf import settings
3 from philo.models import Tag, Titled, Entity, MultiView, Page, register_value_model
4 from django.conf.urls.defaults import url, patterns
5 from django.http import Http404, HttpResponse
6 from datetime import datetime
7 from philo.contrib.penfield.utils import paginate
8 from philo.contrib.penfield.validators import validate_pagination_count
9
10
11 class Blog(Entity, Titled):
12         @property
13         def entry_tags(self):
14                 """ Returns a QuerySet of Tags that are used on any entries in this blog. """
15                 return Tag.objects.filter(blogentries__blog=self)
16
17
18 class BlogEntry(Entity, Titled):
19         blog = models.ForeignKey(Blog, related_name='entries')
20         author = models.ForeignKey(getattr(settings, 'PHILO_PERSON_MODULE', 'auth.User'), related_name='blogentries')
21         date = models.DateTimeField(default=datetime.now)
22         content = models.TextField()
23         excerpt = models.TextField()
24         tags = models.ManyToManyField(Tag, related_name='blogentries')
25         
26         class Meta:
27                 ordering = ['-date']
28                 verbose_name_plural = "blog entries"
29
30
31 register_value_model(BlogEntry)
32
33
34 class BlogView(MultiView):
35         ENTRY_PERMALINK_STYLE_CHOICES = (
36                 ('D', 'Year, month, and day'),
37                 ('M', 'Year and month'),
38                 ('Y', 'Year'),
39                 ('B', 'Custom base'),
40                 ('N', 'No base')
41         )
42         
43         blog = models.ForeignKey(Blog, related_name='blogviews')
44         
45         index_page = models.ForeignKey(Page, related_name='blog_index_related')
46         entry_page = models.ForeignKey(Page, related_name='blog_entry_related')
47         entry_archive_page = models.ForeignKey(Page, related_name='blog_entry_archive_related', null=True, blank=True)
48         tag_page = models.ForeignKey(Page, related_name='blog_tag_related')
49         tag_archive_page = models.ForeignKey(Page, related_name='blog_tag_archive_related', null=True, blank=True)
50         entries_per_page = models.IntegerField(blank=True, validators=[validate_pagination_count])
51         
52         entry_permalink_style = models.CharField(max_length=1, choices=ENTRY_PERMALINK_STYLE_CHOICES)
53         entry_permalink_base = models.CharField(max_length=255, blank=False, default='entries')
54         tag_permalink_base = models.CharField(max_length=255, blank=False, default='tags')
55         
56         @property
57         def urlpatterns(self):
58                 base_patterns = patterns('',
59                         url(r'^$', self.index_view),
60                         url((r'^(?:%s)/?$' % self.tag_permalink_base), self.tag_archive_view),
61                         url((r'^(?:%s)/(?P<tag>[-\w]+)/?$' % self.tag_permalink_base), self.tag_view)
62                 )
63                 if self.entry_permalink_style == 'D':
64                         entry_patterns = patterns('',
65                                 url(r'^(?P<year>\d{4})/?$', self.entry_archive_view),
66                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/?$', self.entry_archive_view),
67                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d{2})/?$', self.entry_archive_view),
68                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d{2})/(?P<slug>[-\w]+)/?$', self.entry_view)
69                         )
70                 elif self.entry_permalink_style == 'M':
71                         entry_patterns = patterns('',
72                                 url(r'^(?P<year>\d{4})/?$', self.entry_archive_view),
73                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/?$', self.entry_archive_view),
74                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<slug>[-\w]+)/?$', self.entry_view)
75                         )
76                 elif self.entry_permalink_style == 'Y':
77                         entry_patterns = patterns('',
78                                 url(r'^(?P<year>\d{4})/?$', self.entry_archive_view),
79                                 url(r'^(?P<year>\d{4})/(?P<slug>[-\w]+)/?$', self.entry_view)
80                         )
81                 elif self.entry_permalink_style == 'B':
82                         entry_patterns = patterns('',
83                                 url((r'^(?:%s)/?$' % self.entry_permalink_base), self.entry_archive_view),
84                                 url((r'^(?:%s)/(?P<slug>[-\w]+)/?$' % self.entry_permalink_base), self.entry_view)
85                         )
86                 else:
87                         entry_patterns = patterns('',
88                                 url(r'^(?P<slug>[-\w]+)/?$', self.entry_view)
89                         )
90                 return base_patterns + entry_patterns
91         
92         def index_view(self, request, node=None, extra_context=None):
93                 entries = self.blog.entries.all()
94                 if self.entries_per_page:
95                         paginated_page = paginate(request, entries, self.entries_per_page)
96                         entries = paginated_page.object_list
97                 else:
98                         paginated_page = None
99                 context = {}
100                 context.update(extra_context or {})
101                 context.update({'blog': self.blog, 'entries': entries, 'paginated_page': paginated_page})
102                 return self.index_page.render_to_response(node, request, extra_context=context)
103         
104         def entry_view(self, request, slug, year=None, month=None, day=None, node=None, extra_context=None):
105                 entries = self.blog.entries.all()
106                 if year:
107                         entries = entries.filter(date__year=year)
108                 if month:
109                         entries = entries.filter(date__month=month)
110                 if day:
111                         entries = entries.filter(date__day=day)
112                 try:
113                         entry = entries.get(slug=slug)
114                 except:
115                         raise Http404
116                 context = {}
117                 context.update(extra_context or {})
118                 context.update({'blog': self.blog, 'entry': entry})
119                 return self.entry_page.render_to_response(node, request, extra_context=context)
120         
121         def entry_archive_view(self, request, year=None, month=None, day=None, node=None, extra_context=None):
122                 if not self.entry_archive_page:
123                         raise Http404
124                 entries = self.blog.entries.all()
125                 if year:
126                         entries = entries.filter(date__year=year)
127                 if month:
128                         entries = entries.filter(date__month=month)
129                 if day:
130                         entries = entries.filter(date__day=day)
131                 if self.entries_per_page:
132                         paginated_page = paginate(request, entries, self.entries_per_page)
133                         entries = paginated_page.object_list
134                 else:
135                         paginated_page = None
136                 context = {}
137                 context.update(extra_context or {})
138                 context.update({'blog': self.blog, 'year': year, 'month': month, 'day': day, 'entries': entries, 'paginated_page': paginated_page})
139                 return self.entry_archive_page.render_to_response(node, request, extra_context=context)
140         
141         def tag_view(self, request, tag, node=None, extra_context=None):
142                 try:
143                         tag = self.blog.entry_tags.filter(slug=tag)
144                 except:
145                         raise Http404
146                 entries = self.blog.entries.filter(tags=tag)
147                 if self.entries_per_page:
148                         paginated_page = paginate(request, entries, self.entries_per_page)
149                         entries = paginated_page.object_list
150                 else:
151                         paginated_page = None
152                 context = {}
153                 context.update(extra_context or {})
154                 context.update({'blog': self.blog, 'tag': tag, 'entries': entries, 'paginated_page': paginated_page})
155                 return self.tag_page.render_to_response(node, request, extra_context=context)
156         
157         def tag_archive_view(self, request, node=None, extra_context=None):
158                 if not self.tag_archive_page:
159                         raise Http404
160                 context = {}
161                 context.update(extra_context or {})
162                 context.update({'blog': self.blog})
163                 return self.tag_archive_page.render_to_response(node, request, extra_context=context)
164
165
166 class Newsletter(Entity, Titled):
167         pass
168
169
170 class NewsletterArticle(Entity, Titled):
171         newsletter = models.ForeignKey(Newsletter, related_name='articles')
172         authors = models.ManyToManyField(getattr(settings, 'PHILO_PERSON_MODULE', 'auth.User'), related_name='newsletterarticles')
173         date = models.DateTimeField(default=datetime.now)
174         lede = models.TextField(null=True, blank=True)
175         full_text = models.TextField()
176         
177         class Meta:
178                 ordering = ['-date']
179                 unique_together = (('newsletter', 'slug'),)
180
181
182 register_value_model(NewsletterArticle)
183
184
185 class NewsletterIssue(Entity, Titled):
186         newsletter = models.ForeignKey(Newsletter, related_name='issues')
187         number = models.PositiveIntegerField()
188         articles = models.ManyToManyField(NewsletterArticle)
189         
190         class Meta:
191                 ordering = ['-number']
192                 unique_together = (('newsletter', 'number'),)
193
194
195 class NewsletterView(MultiView):
196         ARTICLE_PERMALINK_STYLE_CHOICES = (
197                 ('D', 'Year, month, and day'),
198                 ('M', 'Year and month'),
199                 ('Y', 'Year'),
200                 ('S', 'Slug only')
201         )
202         
203         newsletter = models.ForeignKey(Newsletter, related_name='newsletterviews')
204         
205         index_page = models.ForeignKey(Page, related_name='newsletter_index_related')
206         article_page = models.ForeignKey(Page, related_name='newsletter_article_related')
207         article_archive_page = models.ForeignKey(Page, related_name='newsletter_article_archive_related', null=True, blank=True)
208         issue_page = models.ForeignKey(Page, related_name='newsletter_issue_related')
209         issue_archive_page = models.ForeignKey(Page, related_name='newsletter_issue_archive_related', null=True, blank=True)
210         
211         article_permalink_style = models.CharField(max_length=1, choices=ARTICLE_PERMALINK_STYLE_CHOICES)
212         article_permalink_base = models.CharField(max_length=255, blank=False, default='articles')
213         issue_permalink_base = models.CharField(max_length=255, blank=False, default='issues')
214         
215         @property
216         def urlpatterns(self):
217                 base_patterns = patterns('',
218                         url(r'^$', self.index_view),
219                         url((r'^(?:%s)/?$' % self.issue_permalink_base), self.issue_archive_view),
220                         url((r'^(?:%s)/(?P<number>\d+)/?$' % self.issue_permalink_base), self.issue_view)
221                 )
222                 article_patterns = patterns('',
223                         url((r'^(?:%s)/?$' % self.article_permalink_base), self.article_archive_view)
224                 )
225                 if self.article_permalink_style in 'DMY':
226                         article_patterns += patterns('',
227                                 url((r'^(?:%s)/(?P<year>\d{4})/?$' % self.article_permalink_base), self.article_archive_view)
228                         )
229                         if self.article_permalink_style in 'DM':
230                                 article_patterns += patterns('',
231                                         url((r'^(?:%s)/(?P<year>\d{4})/(?P<month>\d{2})/?$' % self.article_permalink_base), self.article_archive_view)
232                                 )
233                                 if self.article_permalink_style == 'D':
234                                         article_patterns += patterns('',
235                                                 url((r'^(?:%s)/(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d{2})/?$' % self.article_permalink_base), self.article_archive_view),
236                                                 url((r'^(?:%s)/(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d{2})/(?P<slug>[-\w]+)/?$' % self.article_permalink_base), self.article_view)
237                                         )
238                                 else:
239                                         article_patterns += patterns('',
240                                                 url((r'^(?:%s)/(?P<year>\d{4})/(?P<month>\d{2})/(?P<slug>[-\w]+)/?$' % self.article_permalink_base), self.article_view)
241                                         )
242                         else:
243                                 article_patterns += patterns('',
244                                         url((r'^(?:%s)/(?P<year>\d{4})/(?P<slug>[-\w]+)/?$' % self.article_permalink_base), self.article_view)
245                                 )
246                 else:
247                         article_patterns += patterns('',
248                                 url((r'^(?:%s)/(?P<slug>[-\w]+)/?$' % self.article_permalink_base), self.article_view)
249                         )
250                 return base_patterns + article_patterns
251         
252         def index_view(self, request, node=None, extra_context=None):
253                 context = {}
254                 context.update(extra_context or {})
255                 context.update({'newsletter': self.newsletter})
256                 return self.index_page.render_to_response(node, request, extra_context=context)
257         
258         def article_view(self, request, slug, year=None, month=None, day=None, node=None, extra_context=None):
259                 articles = self.newsletter.articles.all()
260                 if year:
261                         articles = articles.filter(date__year=year)
262                 if month:
263                         articles = articles.filter(date__month=month)
264                 if day:
265                         articles = articles.filter(date__day=day)
266                 try:
267                         article = articles.get(slug=slug)
268                 except:
269                         raise Http404
270                 context = {}
271                 context.update(extra_context or {})
272                 context.update({'newsletter': self.newsletter, 'article': article})
273                 return self.article_page.render_to_response(node, request, extra_context=context)
274         
275         def article_archive_view(self, request, year=None, month=None, day=None, node=None, extra_context=None):
276                 if not self.article_archive_page:
277                         raise Http404
278                 articles = self.newsletter.articles.all()
279                 if year:
280                         articles = articles.filter(date__year=year)
281                 if month:
282                         articles = articles.filter(date__month=month)
283                 if day:
284                         articles = articles.filter(date__day=day)
285                 context = {}
286                 context.update(extra_context or {})
287                 context.update({'newsletter': self.newsletter, 'year': year, 'month': month, 'day': day, 'articles': articles})
288                 return self.article_archive_page.render_to_response(node, request, extra_context=context)
289         
290         def issue_view(self, request, number, node=None, extra_context=None):
291                 try:
292                         issue = self.newsletter.issues.get(number=number)
293                 except:
294                         raise Http404
295                 context = {}
296                 context.update(extra_context or {})
297                 context.update({'newsletter': self.newsletter, 'issue': issue})
298                 return self.issue_page.render_to_response(node, request, extra_context=context)
299         
300         def issue_archive_view(self, request, node=None, extra_context=None):
301                 if not self.issue_archive_page:
302                         raise Http404
303                 context = {}
304                 context.update(extra_context or {})
305                 context.update({'newsletter': self.newsletter})
306                 return self.issue_archive_page.render_to_response(node, request, extra_context=context)