fixes
[wolnelektury.git] / 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 gettext_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