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