from django.utils.translation import get_language from django.conf import settings from django.db import models import os class Element(models.Model): parent = models.ForeignKey( "Element", on_delete=models.CASCADE, null=True, blank=True, limit_choices_to={"parent": None}, ) link = models.CharField(max_length=512, blank=True) new_window = models.BooleanField(default=False) add_locale = models.BooleanField(default=False) icon = models.FilePathField( path="navbar/static/navbar/icons", match=".png", null=True, blank=True ) weight = models.PositiveSmallIntegerField() def icon_path(self): return os.path.join("navbar/icons", os.path.basename(self.icon)) def children(self): return sorted( list(Element.objects.filter(parent=self.id).order_by("weight")) + list(ElementSeparator.objects.filter(parent=self.id).order_by("weight")), key=lambda c: c.weight, ) def description(self): lang = get_language() if lang is None: lang = settings.LANGUAGE_CODE return ElementDescription.objects.filter(element=self.id, language=lang).first() def localized_link(self): link = self.link desc = self.description() if desc and desc.link_override: link = desc.link_override return link def __str__(self): desc = self.description() return desc.short_name if desc is not None else self.link class ElementDescription(models.Model): element = models.ForeignKey(Element, on_delete=models.CASCADE) language = models.CharField(max_length=10, choices=settings.LANGUAGES) short_name = models.CharField(max_length=32) full_name = models.CharField(max_length=64, blank=True) description = models.CharField(max_length=512, blank=True) link_override = models.CharField(max_length=512, blank=True) def is_separator(self): return False class ElementSeparator(models.Model): parent = models.ForeignKey( Element, on_delete=models.CASCADE, limit_choices_to={"parent": None} ) weight = models.PositiveSmallIntegerField() def is_separator(self): return True def __str__(self): return "{} ({})".format(self.parent, self.weight)