75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
"""Tag app models."""
|
|
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
from configuration.models import TranslationSettings
|
|
from utils.models import TJSONField, TranslatedFieldsMixin
|
|
|
|
|
|
class Tag(TranslatedFieldsMixin, models.Model):
|
|
"""Tag model."""
|
|
|
|
label = TJSONField(blank=True, null=True, default=None,
|
|
verbose_name=_('label'),
|
|
help_text='{"en-GB":"some text"}')
|
|
category = models.ForeignKey('TagCategory', on_delete=models.PROTECT,
|
|
null=True, related_name='tags',
|
|
verbose_name=_('Category'))
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
verbose_name = _('Tag')
|
|
verbose_name_plural = _('Tags')
|
|
|
|
def __str__(self):
|
|
label = 'None'
|
|
lang = TranslationSettings.get_solo().default_language
|
|
if self.label and lang in self.label:
|
|
label = self.label[lang]
|
|
return f'id:{self.id}-{label}'
|
|
|
|
|
|
class TagCategoryQuerySet(models.QuerySet):
|
|
"""Extended queryset for TagCategory model."""
|
|
|
|
def with_base_related(self):
|
|
return self.prefetch_related('tags')
|
|
|
|
def for_news(self):
|
|
return self.filter(news_types__isnull=True)
|
|
|
|
def for_establishments(self):
|
|
return self.filter(models.Q(est_subtype_tag_categories__isnull=True) |
|
|
models.Q(est_type_tag_categories__isnull=True)).\
|
|
distinct()
|
|
|
|
def with_related(self):
|
|
return self.select_related('country').prefetch_related('tags')
|
|
|
|
|
|
class TagCategory(TranslatedFieldsMixin, models.Model):
|
|
"""Tag base category model."""
|
|
|
|
label = TJSONField(blank=True, null=True, default=None,
|
|
verbose_name=_('label'),
|
|
help_text='{"en-GB":"some text"}')
|
|
country = models.ForeignKey('location.Country',
|
|
on_delete=models.SET_NULL, null=True,
|
|
default=None)
|
|
public = models.BooleanField(default=False)
|
|
|
|
objects = TagCategoryQuerySet.as_manager()
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
verbose_name = _('Tag category')
|
|
verbose_name_plural = _('Tag categories')
|
|
|
|
def __str__(self):
|
|
label = 'None'
|
|
lang = TranslationSettings.get_solo().default_language
|
|
if self.label and lang in self.label:
|
|
label = self.label[lang]
|
|
return f'id:{self.id}-{label}'
|