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 import re
  19 import urllib.parse
  20
  21 from django.db import models, transaction
  22 from django.utils.html import strip_tags
  23 from django.utils.translation import gettext_lazy as _
  24 from taggit.managers import TaggableManager
  25
  26 from .fields import RichTextField
  27
  28
  29 class Note(models.Model):
  30     title = models.CharField(_('Title'), max_length=150)
  31     slug = models.SlugField(_('Slug'), max_length=150)
  32     text = RichTextField(_('Text'), blank=True, null=True)
  33     plain_text = models.TextField(blank=True, null=True)
  34     tags = TaggableManager(_('Tags'), blank=True)
  35     included_in_feed = models.BooleanField(_('Include in feed'), default=True)
  36     published = models.BooleanField(_('Published'), default=True)
  37     creation_timestamp = models.DateTimeField(auto_now_add=True)
  38     last_update_timestamp = models.DateTimeField(auto_now=True)
  39
  40     class Meta:
  41         ordering = ['-creation_timestamp']
  42
  43     def get_absolute_url(self):
  44         if self.slug == 'index':
  45             return '/'
  46         return '/%s/' % self.slug
  47
  48     def lang(self):
  49         if self.tags.filter(name='lang-en').exists():
  50             return 'en'
  51         return 'fr'
  52
  53     def save(self, *args, **kwargs):
  54         if kwargs.get('update_fields') is None or 'plain_text' in kwargs.get('update_fields'):
  55             self.plain_text = re.sub(
  56                 r"[’'\"«»,;:\. ]", ' ', html.unescape(strip_tags(self.text)).replace('’', ' ')
  57             )
  58         if not kwargs.get('update_fields'):
  59             self.maintain_interlinks()
  60         return super().save(*args, **kwargs)
  61
  62     def maintain_interlinks(self):
  63         links = re.findall(r'href="(.*?)"', self.text or '')
  64         with transaction.atomic():
  65             Interlink.objects.filter(note1=self).delete()
  66             for link in links:
  67                 parsed = urllib.parse.urlparse(link)
  68                 if parsed.netloc != '':
  69                     continue
  70                 path_parts = parsed.path.strip('/').split()
  71                 if len(path_parts) != 1:
  72                     continue
  73                 try:
  74                     note2 = Note.objects.get(slug=path_parts[0])
  75                     Interlink.objects.create(note1=self, note2=note2)
  76                 except Note.DoesNotExist:
  77                     pass
  78
  79
  80 class Interlink(models.Model):
  81     note1 = models.ForeignKey(Note, on_delete=models.SET_NULL, null=True, related_name='+')
  82     note2 = models.ForeignKey(Note, on_delete=models.SET_NULL, null=True, related_name='linking_note')