99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.contrib.postgres.fields import JSONField
|
|
from utils.models import ProjectBaseMixin
|
|
|
|
|
|
# Mixins
|
|
class CollectionNameMixin(models.Model):
|
|
"""CollectionName mixin"""
|
|
name = models.CharField(_('name'), max_length=250)
|
|
|
|
class Meta:
|
|
"""Meta class"""
|
|
abstract = True
|
|
|
|
|
|
class CollectionDateMixin(models.Model):
|
|
"""CollectionDate mixin"""
|
|
start = models.DateTimeField(_('start'))
|
|
end = models.DateTimeField(_('end'))
|
|
|
|
class Meta:
|
|
"""Meta class"""
|
|
abstract = True
|
|
|
|
|
|
# Models
|
|
class CollectionQuerySet(models.QuerySet):
|
|
"""QuerySet for model Collection"""
|
|
|
|
def by_country(self, country):
|
|
"""Filter collection by country."""
|
|
return self.filter(country=country)
|
|
|
|
def published(self):
|
|
"""Returned only published collection"""
|
|
return self.filter(is_publish=True)
|
|
|
|
|
|
class Collection(ProjectBaseMixin, CollectionNameMixin, CollectionDateMixin):
|
|
"""Collection model."""
|
|
is_publish = models.BooleanField(
|
|
default=False, verbose_name=_('Publish status'))
|
|
filters = JSONField(
|
|
_('filters'), null=True, blank=True,
|
|
default=None, help_text='{"key":"value"}')
|
|
selectors = JSONField(
|
|
_('selectors'), null=True, blank=True,
|
|
default=None, help_text='{"key":"value"}')
|
|
targets = JSONField(
|
|
_('targets'), null=True, blank=True,
|
|
default=None, help_text='{"key":"value"}')
|
|
country = models.ForeignKey(
|
|
'location.Country', verbose_name=_('country'), on_delete=models.CASCADE)
|
|
|
|
objects = CollectionQuerySet.as_manager()
|
|
|
|
|
|
class CollectionItemQuerySet(models.QuerySet):
|
|
"""QuerySet for model CollectionItem."""
|
|
|
|
def by_collection(self, collection_id):
|
|
"""Filter by collection id"""
|
|
return self.filter(collection=collection_id)
|
|
|
|
|
|
class CollectionItem(ProjectBaseMixin):
|
|
"""CollectionItem model."""
|
|
collection = models.ForeignKey(
|
|
Collection, verbose_name=_('collection'), on_delete=models.CASCADE)
|
|
item_type = models.IntegerField(verbose_name=_('item type identifier'))
|
|
item_ids = JSONField(
|
|
_('item_ids'), null=True, blank=True,
|
|
default=None, help_text='{"key":"value"}')
|
|
|
|
objects = CollectionItemQuerySet.as_manager()
|
|
|
|
|
|
class GuideQuerySet(models.QuerySet):
|
|
"""QuerySet for Guide."""
|
|
|
|
def by_collection_id(self, collection_id):
|
|
"""Filter by collection id"""
|
|
return self.filter(collection=collection_id)
|
|
|
|
|
|
class Guide(ProjectBaseMixin, CollectionNameMixin, CollectionDateMixin):
|
|
"""Guide model."""
|
|
parent = models.ForeignKey(
|
|
'self', verbose_name=_('parent'), on_delete=models.CASCADE)
|
|
advertorials = JSONField(
|
|
_('advertorials'), null=True, blank=True,
|
|
default=None, help_text='{"key":"value"}')
|
|
collection = models.ForeignKey(
|
|
Collection, verbose_name=_('collection'), on_delete=models.CASCADE)
|
|
|
|
objects = GuideQuerySet.as_manager()
|
|
|