django-simplecms/cms/models.py

import swapper
from django.db import models
from django.urls import reverse
from django.conf import settings
from django.forms import TextInput, Select
from django.utils.translation import gettext_lazy as _
from ckeditor.fields import RichTextField
from embed_video.fields import EmbedVideoField
from polymorphic.models import PolymorphicModel

from numberedmodel.models import NumberedModel

def register(verbose_name):
    def wrapper(model):
        model.__bases__[-1].TYPES.append((model.__name__.lower(), verbose_name))
        return model
    return wrapper

class VarCharField(models.TextField):
    def formfield(self, **kwargs):
        kwargs.update({'widget': TextInput})
        return super().formfield(**kwargs)

class VarCharChoiceField(models.TextField):
    def formfield(self, **kwargs):
        kwargs.update({'widget': Select})
        return super().formfield(**kwargs)

class BasePage(NumberedModel):
    position = models.PositiveIntegerField(_('position'), blank=True)
    title = VarCharField(_('title'))
    slug = models.SlugField(_('slug'), help_text=_('A short identifier to use in URLs'), blank=True, unique=True)
    menu = models.BooleanField(_('visible in menu'), default=True)

    def __str__(self):
        if not self.pk:
            return str(_('New page'))
        else:
            return self.title

    def get_absolute_url(self):
        if self.slug:
            return reverse('cms:page', args=[self.slug])
        else:
            return reverse('cms:page')

    class Meta:
        abstract = True
        verbose_name = _('Page')
        verbose_name_plural = _('Pages')
        ordering = ['position']

class BaseSection(NumberedModel, PolymorphicModel):
    TYPES = []
    page = models.ForeignKey(swapper.get_model_name('cms', 'Page'), verbose_name=_('page'), related_name='sections', on_delete=models.PROTECT)
    type = VarCharChoiceField(_('section type'), default='', choices=TYPES)
    position = models.PositiveIntegerField(_('position'), blank=True)
    title = VarCharField(_('title'), blank=True)
    color = models.PositiveIntegerField(_('color'), default=1, choices=settings.SECTION_COLORS)
    content = RichTextField(_('content'), blank=True)
    image = models.ImageField(_('image'), blank=True)
    video = EmbedVideoField(_('video'), blank=True, help_text=_('Paste a YouTube, Vimeo, or SoundCloud link'))
    button_text = VarCharField(_('button text'), blank=True)
    button_link = VarCharField(_('button link'), blank=True)

    def number_with_respect_to(self):
        return self.page.sections.all()

    def __str__(self):
        if not self.pk:
            return str(_('New section'))
        elif not self.title:
            return str(_('Untitled'))
        else:
            return self.title

    class Meta:
        abstract = True
        verbose_name = _('section')
        verbose_name_plural = _('sections')
        ordering = ['position']

class Page(BasePage):
    class Meta(BasePage.Meta):
        swappable = swapper.swappable_setting('cms', 'Page')

class Section(BaseSection):
    class Meta(BaseSection.Meta):
        swappable = swapper.swappable_setting('cms', 'Section')

class Config(models.Model):
    TYPES = [
        (10, _('Footer')),
    ]

    parameter = models.PositiveIntegerField(choices=TYPES, unique=True)
    content = RichTextField(_('content'), blank=True)

    def __str__(self):
        return "{}. {}".format(self.parameter, self.get_parameter_display())

    class Meta:
        verbose_name = _('configuration parameter')
        verbose_name_plural = _('configuration parameters')
        ordering = ['parameter']