X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/165161a173154072eb7b999386a13562e110bd0a..739f55be14e161e485b2107b628689c4f01ded6b:/src/chunks/models.py?ds=sidebyside diff --git a/src/chunks/models.py b/src/chunks/models.py index f5fdbbf77..e1ed94db4 100644 --- a/src/chunks/models.py +++ b/src/chunks/models.py @@ -5,7 +5,8 @@ from django.conf import settings from django.core.cache import cache from django.db import models from django.urls import reverse -from django.utils.translation import ugettext_lazy as _ +from django.utils.safestring import mark_safe +from django.utils.translation import gettext_lazy as _ class Chunk(models.Model): @@ -46,3 +47,43 @@ class Attachment(models.Model): def get_absolute_url(self): return reverse('chunks_attachment', args=[self.key, self.attachment.name.rsplit('.', 1)[-1]]) + + +class Menu(models.Model): + identifier = models.CharField(max_length=255, unique=True) + + def __str__(self): + return self.identifier + + +class MenuItem(models.Model): + menu = models.ForeignKey(Menu, models.CASCADE) + order = models.SmallIntegerField() + highlight = models.BooleanField() + infopage = models.ForeignKey( + 'infopages.InfoPage', models.PROTECT, null=True, blank=True) + url = models.CharField(max_length=255, blank=True) + name = models.CharField(max_length=255, blank=True) + + class Meta: + ordering = ('order',) + + @property + def final_name(self): + if self.name == '-': + return mark_safe('