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 from datetime import datetime
   6 from django.conf import settings
   7 from django.contrib.sites.models import Site
   8 from django.core.exceptions import ValidationError
   9 from django.core.mail import send_mail
  10 from django.db import models
  11 from django.template import loader, Context
  12 from django.utils.translation import get_language, ugettext_lazy as _, ugettext
  13 from django_comments_xtd.models import XtdComment
  14 from markupfield.fields import MarkupField
  15 from migdal import app_settings
  16 from migdal.helpers import add_translatable
  17 from migdal.fields import SlugNullField
  18
  19 class Category(models.Model):
  20     taxonomy = models.CharField(_('taxonomy'), max_length=32,
  21                     choices=app_settings.TAXONOMIES)
  22
  23     class Meta:
  24         verbose_name = _('category')
  25         verbose_name_plural = _('categories')
  26
  27     def __unicode__(self):
  28         return self.title
  29
  30     @models.permalink
  31     def get_absolute_url(self):
  32         return ('migdal_category', [self.slug])
  33
  34
  35 add_translatable(Category, {
  36     'title': models.CharField(max_length=64, unique=True, db_index=True),
  37     'slug': models.SlugField(unique=True, db_index=True),
  38 })
  39
  40
  41 class Entry(models.Model):
  42     type = models.CharField(max_length=16,
  43             choices=((t.db, t.slug) for t in app_settings.TYPES),
  44             db_index=True)
  45     date = models.DateTimeField(_('created at'), auto_now_add=True, db_index=True)
  46     changed_at = models.DateTimeField(_('changed at'), auto_now=True, db_index=True)
  47     author = models.CharField(_('author'), max_length=128)
  48     author_email = models.EmailField(_('author email'), max_length=128, null=True, blank=True,
  49             help_text=_('Used only to display gravatar and send notifications.'))
  50     image = models.ImageField(_('image'), upload_to='entry/image/', null=True, blank=True)
  51     promo = models.BooleanField(_('promoted'), default=False)
  52     categories = models.ManyToManyField(Category, null=True, blank=True, verbose_name=_('categories'))
  53
  54     class Meta:
  55         verbose_name = _('entry')
  56         verbose_name_plural = _('entries')
  57         ordering = ['-date']
  58
  59     def __unicode__(self):
  60         return self.title
  61
  62     def save(self, *args, **kwargs):
  63         published_now = False
  64         for lc, ln in settings.LANGUAGES:
  65             if (getattr(self, "published_%s" % lc)
  66                     and getattr(self, "published_at_%s" % lc) is None):
  67                 setattr(self, "published_at_%s" % lc, datetime.now())
  68                 published_now = True
  69         super(Entry, self).save(*args, **kwargs)
  70         if published_now:
  71             self.notify_author_published()
  72
  73     def clean(self):
  74         for lc, ln in settings.LANGUAGES:
  75             if (getattr(self, "published_%s" % lc) and
  76                     not getattr(self, "slug_%s" % lc)):
  77                 raise ValidationError(
  78                     ugettext("Published entry should have a slug in relevant language (%s).") % lc)
  79
  80     @models.permalink
  81     def get_absolute_url(self):
  82         return ('migdal_entry_%s' % self.type, [self.slug])
  83
  84     def get_type(self):
  85         return dict(app_settings.TYPES_DICT)[self.type]
  86
  87     def notify_author_published(self):
  88         if not self.author_email:
  89             return
  90         site = Site.objects.get_current()
  91         mail_text = loader.get_template('migdal/mail/published.txt').render(
  92             Context({
  93                 'entry': self,
  94                 'site': site,
  95             }))
  96         send_mail(
  97             ugettext(u'Your story has been published at %s.') % site.domain,
  98             mail_text, settings.SERVER_EMAIL, [self.author_email]
  99         )
 100
 101
 102 add_translatable(Entry, languages=app_settings.OPTIONAL_LANGUAGES, fields={
 103     'needed': models.CharField(_('needed'), max_length=1, db_index=True, choices=(
 104                 ('n', _('Unneeded')), ('w', _('Needed')), ('y', _('Done'))),
 105                 default='n'),
 106 })
 107
 108 add_translatable(Entry, {
 109     'slug': SlugNullField(unique=True, db_index=True, null=True, blank=True),
 110     'title': models.CharField(_('title'), max_length=255, null=True, blank=True),
 111     'lead': MarkupField(_('lead'), markup_type='textile_pl', null=True, blank=True,
 112                 help_text=_('Use <a href="http://textile.thresholdstate.com/">Textile</a> syntax.')),
 113     'body': MarkupField(_('body'), markup_type='textile_pl', null=True, blank=True,
 114                 help_text=_('Use <a href="http://textile.thresholdstate.com/">Textile</a> syntax.')),
 115     'published': models.BooleanField(_('published'), default=False),
 116     'published_at': models.DateTimeField(_('published at'), null=True, blank=True),
 117 })
 118
 119
 120 class Attachment(models.Model):
 121     file = models.FileField(_('file'), upload_to='entry/attach/')
 122     entry = models.ForeignKey(Entry)
 123
 124     def url(self):
 125         return self.file.url if self.file else ''
 126
 127
 128
 129 def notify_new_comment(sender, instance, created, **kwargs):
 130     if (created and isinstance(instance.content_object, Entry) and
 131                 instance.content_object.author_email):
 132         site = Site.objects.get_current()
 133         mail_text = loader.get_template('migdal/mail/new_comment.txt').render(
 134             Context({
 135                 'comment': instance,
 136                 'site': site,
 137             }))
 138         send_mail(
 139             ugettext(u'New comment under your story at %s.') % site.domain,
 140             mail_text, settings.SERVER_EMAIL,
 141             [instance.content_object.author_email]
 142         )
 143 models.signals.post_save.connect(notify_new_comment, sender=XtdComment)