contrib/penfield/models.py

   1 from django.db import models
   2 from philo.models import Entity, MultiView, 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 Tag(models.Model):
  11         name = models.CharField(max_length=250)
  12         slug = models.SlugField()
  13
  14         def __unicode__(self):
  15                 return self.name
  16
  17
  18 class Titled(models.Model):
  19         title = models.CharField(max_length=255)
  20         slug = models.SlugField()
  21
  22         def __unicode__(self):
  23                 return self.title
  24
  25         class Meta:
  26                 abstract = True
  27
  28
  29 class Blog(Entity, Titled):
  30         pass
  31
  32
  33 class BlogEntry(Entity, Titled):
  34         blog = models.ForeignKey(Blog, related_name='entries')
  35         author = models.ForeignKey(User, related_name='blogentries')
  36         date = models.DateTimeField(default=datetime.now)
  37         content = models.TextField()
  38         excerpt = models.TextField()
  39         tags = models.ManyToManyField(Tag)
  40
  41         class Meta:
  42                 ordering = ['-date']
  43                 verbose_name_plural = "Blog Entries"
  44
  45
  46 register_value_model(BlogEntry)
  47
  48
  49 class BlogView(MultiView):
  50         PERMALINK_STYLE_CHOICES = (
  51                 ('D', 'Year, month, and day'),
  52                 ('M', 'Year and month'),
  53                 ('Y', 'Year'),
  54                 ('B', 'Custom base'),
  55                 ('N', 'No base')
  56         )
  57
  58         blog = models.ForeignKey(Blog, related_name='nodes')
  59
  60         index_template = models.ForeignKey(Template, related_name='blog_index_related')
  61         archive_template = models.ForeignKey(Template, related_name='blog_archive_related')
  62         tag_template = models.ForeignKey(Template, related_name='blog_tag_related')
  63         entry_template = models.ForeignKey(Template, related_name='blog_entry_related')
  64
  65         entry_permalink_style = models.CharField(max_length=1, choices=PERMALINK_STYLE_CHOICES)
  66         entry_permalink_base = models.CharField(max_length=255, blank=False, default='entries')
  67         tag_permalink_base = models.CharField(max_length=255, blank=False, default='tags')
  68
  69         @property
  70         def urlpatterns(self):
  71                 base_patterns = patterns('',
  72                         url(r'^$', self.index_view),
  73                         url((r'^(?:%s)/?' % self.tag_permalink_base), self.tag_view),
  74                         url((r'^(?:%s)/(?P<tag>>[-\w]+)/?' % self.tag_permalink_base), self.tag_view)
  75                 )
  76                 if self.entry_permalink_style == 'D':
  77                         entry_patterns = patterns('',
  78                                 url(r'^(?P<year>\d{4})/?$', self.archive_view),
  79                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/?$', self.archive_view),
  80                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d+)/?$', self.archive_view),
  81                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<day>\d+)/(?P<slug>[-\w]+)/?', self.entry_view)
  82                         )
  83                 elif self.entry_permalink_style == 'M':
  84                         entry_patterns = patterns('',
  85                                 url(r'^(?P<year>\d{4})/?$', self.archive_view),
  86                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/?$', self.archive_view),
  87                                 url(r'^(?P<year>\d{4})/(?P<month>\d{2})/(?P<slug>>[-\w]+)/?', self.entry_view)
  88                         )
  89                 elif self.entry_permalink_style == 'Y':
  90                         entry_patterns = patterns('',
  91                                 url(r'^(?P<year>\d{4})/?$', self.archive_view),
  92                                 url(r'^(?P<year>\d{4})/(?P<slug>>[-\w]+)/?', self.entry_view)
  93                         )
  94                 elif self.entry_permalink_style == 'B':
  95                         entry_patterns = patterns('',
  96                                 url((r'^(?:%s)/?' % self.entry_permalink_base), self.archive_view),
  97                                 url((r'^(?:%s)/(?P<slug>>[-\w]+)/?' % self.entry_permalink_base), self.entry_view)
  98                         )
  99                 else:
 100                         entry_patterns = patterns('',
 101                                 url(r'^(?P<slug>>[-\w]+)/?', self.entry_view)
 102                         )
 103                 return base_patterns + entry_patterns
 104
 105         def index_view(self, request):
 106                 entries = self.blog.entries.order_by('-date')
 107                 return HttpResponse(self.index_template.django_template.render(RequestContext(request, {'blog': self.blog, 'entries': entries})), mimetype=self.index_template.mimetype)
 108
 109         def archive_view(self, request, year=None, month=None, day=None):
 110                 entries = self.blog.entries.all()
 111                 if year:
 112                         entries = entries.filter(date__year=year)
 113                 if month:
 114                         entries = entries.filter(date__month=month)
 115                 if day:
 116                         entries = entries.filter(date__day=day)
 117                 return HttpResponse(self.archive_template.django_template.render(RequestContext(request, {'blog': self.blog, 'year': year, 'month': month, 'day': day, 'entries': entries})), mimetype=self.archive_template.mimetype)
 118
 119         def tag_view(self, request, tag=None):
 120                 # return HttpResponse(self.tag_template.django_template.render(RequestContext(request, {'blog': self.blog, 'tag': tag, 'entries': None})), mimetype=self.tag_template.mimetype)
 121                 raise Http404
 122
 123         def entry_view(self, request, slug, year=None, month=None, day=None):
 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                 try:
 132                         entry = entries.get(slug=slug)
 133                 except:
 134                         raise Http404
 135                 return HttpResponse(self.entry_template.django_template.render(RequestContext(request, {'blog': self.blog, 'entry': entry})), mimetype=self.entry_template.mimetype)
 136
 137
 138 class Newsletter(Entity, Titled):
 139         pass
 140
 141
 142 class NewsStory(Entity, Titled):
 143         newsletter = models.ForeignKey(Newsletter, related_name='stories')
 144         authors = models.ManyToManyField(User, related_name='newsstories')
 145         date = models.DateTimeField(default=datetime.now)
 146         lede = models.TextField(null=True, blank=True)
 147         full_text = models.TextField()
 148
 149
 150 register_value_model(NewsStory)