migdal/models.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 import re
   6 from datetime import datetime
   7 from django.conf import settings
   8 from django.contrib.comments.signals import comment_will_be_posted
   9 from django.contrib.sites.models import Site
  10 from django.core.exceptions import ValidationError
  11 from django.core.mail import mail_managers, send_mail
  12 from django.db import models
  13 from django.template import loader, Context
  14 from django.utils.translation import get_language, ugettext_lazy as _, ugettext
  15 from django_comments_xtd.models import XtdComment
  16 from markupfield.fields import MarkupField
  17 from fnpdjango.utils.models.translation import add_translatable, tQ
  18 from migdal import app_settings
  19 from migdal.fields import SlugNullField
  20
  21 class Category(models.Model):
  22     taxonomy = models.CharField(_('taxonomy'), max_length=32,
  23                     choices=app_settings.TAXONOMIES)
  24
  25     class Meta:
  26         verbose_name = _('category')
  27         verbose_name_plural = _('categories')
  28
  29     def __unicode__(self):
  30         return self.title
  31
  32     @models.permalink
  33     def get_absolute_url(self):
  34         return ('migdal_category', [self.slug])
  35
  36
  37 add_translatable(Category, {
  38     'title': models.CharField(max_length=64, unique=True, db_index=True),
  39     'slug': models.SlugField(unique=True, db_index=True),
  40 })
  41
  42
  43 class PublishedEntryManager(models.Manager):
  44     def get_query_set(self):
  45         return super(PublishedEntryManager, self).get_query_set().filter(
  46                 tQ(published=True)
  47             )
  48
  49 class Entry(models.Model):
  50     type = models.CharField(max_length=16,
  51             choices=((t.db, t.slug) for t in app_settings.TYPES),
  52             db_index=True)
  53     date = models.DateTimeField(_('created at'), auto_now_add=True, db_index=True)
  54     changed_at = models.DateTimeField(_('changed at'), auto_now=True, db_index=True)
  55     author = models.CharField(_('author'), max_length=128)
  56     author_email = models.EmailField(_('author email'), max_length=128, null=True, blank=True,
  57             help_text=_('Used only to display gravatar and send notifications.'))
  58     image = models.ImageField(_('image'), upload_to='entry/image/', null=True, blank=True)
  59     promo = models.BooleanField(_('promoted'), default=False)
  60     in_stream = models.BooleanField(_('in stream'), default=True)
  61     categories = models.ManyToManyField(Category, null=True, blank=True, verbose_name=_('categories'))
  62     first_published_at = models.DateTimeField(_('published at'), null=True, blank=True)
  63     canonical_url = models.URLField(_('canonical link'), null = True, blank = True)
  64
  65     objects = models.Manager()
  66     published_objects = PublishedEntryManager()
  67
  68     class Meta:
  69         verbose_name = _('entry')
  70         verbose_name_plural = _('entries')
  71         ordering = ['-date']
  72
  73     def __unicode__(self):
  74         return self.title
  75
  76     def save(self, *args, **kwargs):
  77         published_now = False
  78         for lc, ln in settings.LANGUAGES:
  79             if (getattr(self, "published_%s" % lc)
  80                     and getattr(self, "published_at_%s" % lc) is None):
  81                 now = datetime.now()
  82                 setattr(self, "published_at_%s" % lc, now)
  83                 if self.first_published_at is None:
  84                     self.first_published_at = now
  85                     published_now = True
  86         super(Entry, self).save(*args, **kwargs)
  87         if published_now and self.pk is not None:
  88             self.notify_author_published()
  89
  90     def clean(self):
  91         for lc, ln in settings.LANGUAGES:
  92             if (getattr(self, "published_%s" % lc) and
  93                     not getattr(self, "slug_%s" % lc)):
  94                 raise ValidationError(
  95                     ugettext("Published entry should have a slug in relevant language (%s).") % lc)
  96
  97     @models.permalink
  98     def get_absolute_url(self):
  99         return ('migdal_entry_%s' % self.type, [self.slug])
 100
 101     def get_type(self):
 102         return dict(app_settings.TYPES_DICT)[self.type]
 103
 104     def notify_author_published(self):
 105         if not self.author_email:
 106             return
 107         site = Site.objects.get_current()
 108         mail_text = loader.get_template('migdal/mail/published.txt').render(
 109             Context({
 110                 'entry': self,
 111                 'site': site,
 112             }))
 113         send_mail(
 114             ugettext(u'Your story has been published at %s.') % site.domain,
 115             mail_text, settings.SERVER_EMAIL, [self.author_email]
 116         )
 117
 118     def inline_html(self):
 119         for att in self.attachment_set.all():
 120             if att.file.name.endswith(".html"):
 121                 with open(att.file.path) as f:
 122                     yield f.read()
 123
 124
 125 add_translatable(Entry, languages=app_settings.OPTIONAL_LANGUAGES, fields={
 126     'needed': models.CharField(_('needed'), max_length=1, db_index=True, choices=(
 127                 ('n', _('Unneeded')), ('w', _('Needed')), ('y', _('Done'))),
 128                 default='n'),
 129 })
 130
 131 add_translatable(Entry, {
 132     'slug': SlugNullField(unique=True, db_index=True, null=True, blank=True),
 133     'title': models.CharField(_('title'), max_length=255, null=True, blank=True),
 134     'lead': MarkupField(_('lead'), markup_type='textile_pl', null=True, blank=True,
 135                 help_text=_('Use <a href="http://textile.thresholdstate.com/">Textile</a> syntax.')),
 136     'body': MarkupField(_('body'), markup_type='textile_pl', null=True, blank=True,
 137                 help_text=_('Use <a href="http://textile.thresholdstate.com/">Textile</a> syntax.')),
 138     'published': models.BooleanField(_('published'), default=False),
 139     'published_at': models.DateTimeField(_('published at'), null=True, blank=True),
 140 })
 141
 142
 143 class Attachment(models.Model):
 144     file = models.FileField(_('file'), upload_to='entry/attach/')
 145     entry = models.ForeignKey(Entry)
 146
 147     def url(self):
 148         return self.file.url if self.file else ''
 149
 150
 151
 152 def notify_new_comment(sender, instance, created, **kwargs):
 153     if (created and isinstance(instance.content_object, Entry) and
 154                 instance.content_object.author_email):
 155         site = Site.objects.get_current()
 156         mail_text = loader.get_template('migdal/mail/new_comment.txt').render(
 157             Context({
 158                 'comment': instance,
 159                 'site': site,
 160             }))
 161         send_mail(
 162             ugettext(u'New comment under your story at %s.') % site.domain,
 163             mail_text, settings.SERVER_EMAIL,
 164             [instance.content_object.author_email]
 165         )
 166 models.signals.post_save.connect(notify_new_comment, sender=XtdComment)
 167
 168
 169 def spamfilter(sender, comment, **kwargs):
 170     """Very simple spam filter. Just don't let any HTML links go through."""
 171     if re.search(r"<a\s+href=", comment.comment):
 172         fields = (comment.user, comment.user_name, comment.user_email,
 173             comment.user_url, comment.submit_date, comment.ip_address,
 174             comment.followup, comment.comment)
 175         mail_managers(u"Spam filter report",
 176             (u"""This comment was turned down as SPAM: \n""" +
 177             """\n%s""" * len(fields) +
 178             """\n\nYou don't have to do anything.""") % fields)
 179         return False
 180     return True
 181 comment_will_be_posted.connect(spamfilter)