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