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 ugettext_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     published = models.BooleanField(_('Published'), default=True)
  36     creation_timestamp = models.DateTimeField(auto_now_add=True)
  37     last_update_timestamp = models.DateTimeField(auto_now=True)
  38
  39     class Meta:
  40         ordering = ['-creation_timestamp']
  41
  42     def get_absolute_url(self):
  43         if self.slug == 'index':
  44             return '/'
  45         return '/%s/' % self.slug
  46
  47     def lang(self):
  48         if self.tags.filter(name='lang-en').exists():
  49             return 'en'
  50         return 'fr'
  51
  52     def save(self, *args, **kwargs):
  53         if kwargs.get('update_fields') is None or 'plain_text' in kwargs.get('update_fields'):
  54             self.plain_text = re.sub(
  55                 r"[’'\"«»,;:\. ]", ' ', html.unescape(strip_tags(self.text)).replace('’', ' ')
  56             )
  57         if not kwargs.get('update_fields'):
  58             self.maintain_interlinks()
  59         return super().save(*args, **kwargs)
  60
  61     def maintain_interlinks(self):
  62         links = re.findall(r'href="(.*?)"', self.text or '')
  63         with transaction.atomic():
  64             Interlink.objects.filter(note1=self).delete()
  65             for link in links:
  66                 parsed = urllib.parse.urlparse(link)
  67                 if parsed.netloc != '':
  68                     continue
  69                 path_parts = parsed.path.strip('/').split()
  70                 if len(path_parts) != 1:
  71                     continue
  72                 try:
  73                     note2 = Note.objects.get(slug=path_parts[0])
  74                     Interlink.objects.create(note1=self, note2=note2)
  75                 except Note.DoesNotExist:
  76                     pass
  77
  78
  79 class Interlink(models.Model):
  80     note1 = models.ForeignKey(Note, on_delete=models.SET_NULL, null=True, related_name='+')
  81     note2 = models.ForeignKey(Note, on_delete=models.SET_NULL, null=True, related_name='linking_note')