chloro/phyll/models.py

   1 # chloro - personal space
   2 # Copyright (C) 2019  Frederic Peters
   3 #
   4 # This program is free software: you can redistribute it and/or modify it
   5 # under the terms of the GNU Affero General Public License as published
   6 # by the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 import html
  18
  19 from django.db import models, transaction
  20 from django.utils.html import strip_tags
  21 from django.utils.translation import ugettext_lazy as _
  22 from taggit.managers import TaggableManager
  23
  24 from .fields import RichTextField
  25
  26
  27 class Note(models.Model):
  28     title = models.CharField(_('Title'), max_length=150)
  29     slug = models.SlugField(_('Slug'), max_length=150)
  30     text = RichTextField(_('Text'), blank=True, null=True)
  31     plain_text = models.TextField(blank=True, null=True)
  32     tags = TaggableManager(_('Tags'), blank=True)
  33     published = models.BooleanField(_('Published'), default=True)
  34     creation_timestamp = models.DateTimeField(auto_now_add=True)
  35     last_update_timestamp = models.DateTimeField(auto_now=True)
  36
  37     class Meta:
  38         ordering = ['-creation_timestamp']
  39
  40     def get_absolute_url(self):
  41         return '/%s/' % self.slug
  42
  43     def lang(self):
  44         if self.tags.filter(name='lang-en').exists():
  45             return 'en'
  46         return 'fr'
  47
  48     def save(self, *args, **kwargs):
  49         if kwargs.get('update_fields') is None or 'plain_text' in kwargs.get('update_fields'):
  50             self.plain_text = html.unescape(strip_tags(self.text))
  51         return super().save(*args, **kwargs)