contrib/penfield/models.py

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