contrib/penfield/models.py

   1 from django.db import models
   2 from philo.models import Entity, MultiNode, Template, register_value_model
   3 from django.contrib.auth.models import User
   4 from django.conf.urls.defaults import url, patterns
   5 from django.http import Http404, HttpResponse
   6 from django.template import RequestContext
   7 from datetime import datetime
   8
   9
  10 class TitledContent(models.Model):
  11         title = models.CharField(max_length=255)
  12         slug = models.SlugField()
  13         content = models.TextField()
  14         excerpt = models.TextField()
  15
  16         class Meta:
  17                 abstract = True
  18
  19
  20 class Blog(MultiNode):
  21         PERMALINK_STYLE_CHOICES = (
  22                 ('D', 'Year, month, and day'),
  23                 ('M', 'Year and month'),
  24                 ('Y', 'Year'),
  25                 ('B', 'Custom base'),
  26                 ('N', 'No base')
  27         )
  28
  29         title = models.CharField(max_length=255)
  30
  31         index_template = models.ForeignKey(Template, related_name='blog_index_related')
  32         archive_template = models.ForeignKey(Template, related_name='blog_archive_related')
  33         tag_template = models.ForeignKey(Template, related_name='blog_tag_related')
  34         entry_template = models.ForeignKey(Template, related_name='blog_entry_related')
  35
  36         entry_permalink_style = models.CharField(max_length=1, choices=PERMALINK_STYLE_CHOICES)
  37         entry_permalink_base = models.CharField(max_length=255, blank=False, default='entries')
  38         tag_permalink_base = models.CharField(max_length=255, blank=False, default='tags')
  39
  40         @property
  41         def urlpatterns(self):
  42                 base_patterns = patterns('',
  43                         url(r'^$', self.index_view),
  44                         url((r'^(?:%s)/?' % self.tag_permalink_base), self.tag_view),
  45                         url((r'^(?:%s)/(?P<tag>>[-\w]+)/?' % self.tag_permalink_base), self.tag_view)
  46                 )
  47                 if self.entry_permalink_style == 'D':
  48                         entry_patterns = patterns('',
  49                                 url(r'^(?P<year>\d{4})/?$', self.archive_view),
  50                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/?$', self.archive_view),
  51                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d+)/?$', self.archive_view),
  52                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d+)/(?P<slug>[-\w]+)/?', self.entry_view)
  53                         )
  54                 elif self.entry_permalink_style == 'M':
  55                         entry_patterns = patterns('',
  56                                 url(r'^(?P<year>\d{4})/?$', self.archive_view),
  57                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/?$', self.archive_view),
  58                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<slug>>[-\w]+)/?', self.entry_view)
  59                         )
  60                 elif self.entry_permalink_style == 'Y':
  61                         entry_patterns = patterns('',
  62                                 url(r'^(?P<year>\d{4})/?$', self.archive_view),
  63                                 url(r'^(?P<year>\d{4})/(?P<slug>>[-\w]+)/?', self.entry_view)
  64                         )
  65                 elif self.entry_permalink_style == 'B':
  66                         entry_patterns = patterns('',
  67                                 url((r'^(?:%s)/?' % self.entry_permalink_base), self.archive_view),
  68                                 url((r'^(?:%s)/(?P<slug>>[-\w]+)/?' % self.entry_permalink_base), self.entry_view)
  69                         )
  70                 else:
  71                         entry_patterns = patterns('',
  72                                 url(r'^(?P<slug>>[-\w]+)/?', self.entry_view)
  73                         )
  74                 return base_patterns + entry_patterns
  75
  76         def index_view(self, request):
  77                 return HttpResponse(self.index_template.django_template.render(RequestContext(request, {'blog': self})), mimetype=self.index_template.mimetype)
  78
  79         def archive_view(self, request, year=None, month=None, day=None):
  80                 entries = self.entries.all()
  81                 if year:
  82                         entries = entries.filter(date__year=year)
  83                 if month:
  84                         entries = entries.filter(date__month=month)
  85                 if day:
  86                         entries = entries.filter(date__day=day)
  87                 return HttpResponse(self.archive_template.django_template.render(RequestContext(request, {'blog': self, 'year': year, 'month': month, 'day': day, 'entries': entries})), mimetype=self.archive_template.mimetype)
  88
  89         def tag_view(self, request, tag=None):
  90                 return HttpResponse(self.tag_template.django_template.render(RequestContext(request, {'blog': self, 'tag': tag, 'entries': None})), mimetype=self.tag_template.mimetype)
  91
  92         def entry_view(self, request, slug, year=None, month=None, day=None):
  93                 entries = self.entries.all()
  94                 if year:
  95                         entries = entries.filter(date__year=year)
  96                 if month:
  97                         entries = entries.filter(date__month=month)
  98                 if day:
  99                         entries = entries.filter(date__day=day)
 100                 try:
 101                         entry = entries.get(slug=slug)
 102                 except:
 103                         raise Http404
 104                 return HttpResponse(self.entry_template.django_template.render(RequestContext(request, {'blog': self, 'entry': entry})), mimetype=self.entry_template.mimetype)
 105
 106
 107 class BlogEntry(Entity, TitledContent):
 108         blog = models.ForeignKey(Blog, related_name='entries')
 109         author = models.ForeignKey(User, related_name='blogentries')
 110         date = models.DateTimeField(default=datetime.now)
 111
 112
 113 register_value_model(BlogEntry)