src/chunks/models.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from django.conf import settings
   5 from django.core.cache import cache
   6 from django.db import models
   7 from django.urls import reverse
   8 from django.utils.safestring import mark_safe
   9 from django.utils.translation import ugettext_lazy as _
  10
  11
  12 class Chunk(models.Model):
  13     """
  14     A Chunk is a piece of content associated with a unique key that can be inserted into
  15     any template with the use of a special template tag.
  16     """
  17     key = models.CharField(_('key'), help_text=_('A unique name for this chunk of content'), primary_key=True,
  18                            max_length=255)
  19     description = models.CharField(_('description'), blank=True, max_length=255)
  20     content = models.TextField(_('content'), blank=True)
  21
  22     class Meta:
  23         ordering = ('key',)
  24         verbose_name = _('chunk')
  25         verbose_name_plural = _('chunks')
  26
  27     def __str__(self):
  28         return self.key
  29
  30     def save(self, *args, **kwargs):
  31         ret = super(Chunk, self).save(*args, **kwargs)
  32         for lc, ln in settings.LANGUAGES:
  33             cache.delete('chunk:%s:%s' % (self.key, lc))
  34         return ret
  35
  36
  37 class Attachment(models.Model):
  38     key = models.CharField(_('key'), help_text=_('A unique name for this attachment'), primary_key=True, max_length=255)
  39     attachment = models.FileField(upload_to='chunks/attachment')
  40
  41     class Meta:
  42         ordering = ('key',)
  43         verbose_name, verbose_name_plural = _('attachment'), _('attachments')
  44
  45     def __str__(self):
  46         return self.key
  47
  48     def get_absolute_url(self):
  49         return reverse('chunks_attachment', args=[self.key, self.attachment.name.rsplit('.', 1)[-1]])
  50
  51
  52 class Menu(models.Model):
  53     identifier = models.CharField(max_length=255, unique=True)
  54
  55     def __str__(self):
  56         return self.identifier
  57
  58
  59 class MenuItem(models.Model):
  60     menu = models.ForeignKey(Menu, models.CASCADE)
  61     order = models.SmallIntegerField()
  62     highlight = models.BooleanField()
  63     infopage = models.ForeignKey(
  64         'infopages.InfoPage', models.PROTECT, null=True, blank=True)
  65     url = models.CharField(max_length=255, blank=True)
  66     name = models.CharField(max_length=255, blank=True)
  67
  68     class Meta:
  69         ordering = ('order',)
  70
  71     @property
  72     def final_name(self):
  73         if self.name == '-':
  74             return mark_safe('<hr>')
  75         if self.name:
  76             return self.name
  77         if self.infopage:
  78             return self.infopage.title
  79         return ''
  80
  81     @property
  82     def final_link(self):
  83         if self.infopage:
  84             return self.infopage.get_absolute_url()
  85         return self.url
  86
  87     @property
  88     def has_link(self):
  89         return self.url or self.infopage