gestioCOF/gestioncof/cms/models.py

from django.core.paginator import EmptyPage, PageNotAnInteger, Paginator
from django.db import models
from wagtail.contrib.wagtailroutablepage.models import RoutablePageMixin, route
from wagtail.wagtailadmin.edit_handlers import FieldPanel, StreamFieldPanel
from wagtail.wagtailcore import blocks
from wagtail.wagtailcore.fields import RichTextField, StreamField
from wagtail.wagtailcore.models import Page
from wagtail.wagtailimages.blocks import ImageChooserBlock
from wagtail.wagtailimages.edit_handlers import ImageChooserPanel


# Page pouvant afficher des actualités
class COFActuIndexMixin:
    @property
    def actus(self):
        actus = COFActuPage.objects.live().order_by("-date_start").descendant_of(self)
        return actus


# Racine du site du COF
class COFRootPage(Page, COFActuIndexMixin):
    introduction = RichTextField("Introduction")

    content_panels = Page.content_panels + [
        FieldPanel("introduction", classname="full")
    ]

    subpage_types = ["COFActuIndexPage", "COFPage", "COFDirectoryPage", "COFUtilPage"]

    class Meta:
        verbose_name = "Racine site du COF"
        verbose_name_plural = "Racines site du COF"


# Block iframe
class IFrameBlock(blocks.StructBlock):
    url = blocks.URLBlock("Adresse de la page")
    height = blocks.CharBlock("Hauteur (en pixels)")

    class Meta:
        verbose_name = "Page incluse (iframe, à utiliser avec précaution)"
        verbose_name_plural = "Pages incluses (iframes, à utiliser avec précaution)"
        template = "cofcms/iframe_block.html"


# Page lambda du site
class COFPage(Page):
    body = StreamField(
        [
            ("heading", blocks.CharBlock(classname="full title")),
            ("paragraph", blocks.RichTextBlock()),
            ("image", ImageChooserBlock()),
            ("iframe", IFrameBlock()),
        ]
    )

    content_panels = Page.content_panels + [StreamFieldPanel("body")]

    subpage_types = ["COFDirectoryPage", "COFPage"]
    parent_page_types = ["COFPage", "COFRootPage"]

    class Meta:
        verbose_name = "Page normale COF"
        verbose_name_plural = "Pages normales COF"


# Actualités
class COFActuIndexPage(Page, COFActuIndexMixin):
    subpage_types = ["COFActuPage"]
    parent_page_types = ["COFRootPage"]

    class Meta:
        verbose_name = "Index des actualités"
        verbose_name_plural = "Indexs des actualités"

    def get_context(self, request):
        context = super().get_context(request)
        actus = COFActuPage.objects.live().descendant_of(self).order_by("-date_end")

        page = request.GET.get("page")
        paginator = Paginator(actus, 5)
        try:
            actus = paginator.page(page)
        except PageNotAnInteger:
            actus = paginator.page(1)
        except EmptyPage:
            actus = paginator.page(paginator.num_pages)

        context["actus"] = actus
        return context


class COFActuPage(RoutablePageMixin, Page):
    chapo = models.TextField("Description rapide", blank=True)
    body = RichTextField("Contenu")
    image = models.ForeignKey(
        "wagtailimages.Image",
        verbose_name="Image à la Une",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="+",
    )
    is_event = models.BooleanField("Évènement", default=True, blank=True)
    date_start = models.DateTimeField("Date et heure de début")
    date_end = models.DateTimeField(
        "Date et heure de fin", blank=True, default=None, null=True
    )
    all_day = models.BooleanField("Toute la journée", default=False, blank=True)

    content_panels = Page.content_panels + [
        ImageChooserPanel("image"),
        FieldPanel("chapo"),
        FieldPanel("body", classname="full"),
        FieldPanel("is_event"),
        FieldPanel("date_start"),
        FieldPanel("date_end"),
        FieldPanel("all_day"),
    ]

    subpage_types = []
    parent_page_types = ["COFActuIndexPage"]

    class Meta:
        verbose_name = "Actualité"
        verbose_name_plural = "Actualités"


# Annuaires (Clubs, partenaires, bonnes adresses)
class COFDirectoryPage(Page):
    introduction = RichTextField("Introduction")
    alphabetique = models.BooleanField(
        "Tri par ordre alphabétique ?", default=True, blank=True
    )

    content_panels = Page.content_panels + [
        FieldPanel("introduction"),
        FieldPanel("alphabetique"),
    ]

    subpage_types = ["COFActuPage", "COFDirectoryEntryPage"]
    parent_page_types = ["COFRootPage", "COFPage"]

    @property
    def entries(self):
        entries = COFDirectoryEntryPage.objects.live().descendant_of(self)
        if self.alphabetique:
            entries = entries.order_by("title")
        return entries

    class Meta:
        verbose_name = "Annuaire (clubs, partenaires, bons plans...)"
        verbose_name_plural = "Annuaires"


class COFDirectoryEntryPage(Page):
    body = RichTextField("Description")
    links = StreamField(
        [
            (
                "lien",
                blocks.StructBlock(
                    [
                        ("url", blocks.URLBlock(required=True)),
                        ("texte", blocks.CharBlock()),
                    ]
                ),
            ),
            (
                "contact",
                blocks.StructBlock(
                    [
                        ("email", blocks.EmailBlock(required=True)),
                        ("texte", blocks.CharBlock()),
                    ]
                ),
            ),
        ]
    )

    image = models.ForeignKey(
        "wagtailimages.Image",
        verbose_name="Image",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="+",
    )

    content_panels = Page.content_panels + [
        ImageChooserPanel("image"),
        FieldPanel("body", classname="full"),
        StreamFieldPanel("links"),
    ]

    subpage_types = []
    parent_page_types = ["COFDirectoryPage"]

    class Meta:
        verbose_name = "Entrée d'annuaire"
        verbose_name_plural = "Entrées d'annuaire"


# Pour le calendrier, ne doit pas être pris par ModelTranslation
class COFUtilPage(RoutablePageMixin, Page):

    # Mini calendrier
    @route(r"^calendar/(\d+)/(\d+)/$")
    def calendar(self, request, year, month):
        from .views import raw_calendar_view

        return raw_calendar_view(request, int(year), int(month))

    """
    ModelTranslation override le système des @route de wagtail, ce qui empêche
    COFUtilPage d'être une page traduite pour pouvoir l'utiliser.
    Ce qui fait planter `get_absolute_url` pour des problèmes d'héritage des
    pages parentes (qui sont, elles, traduites).
    Le seul moyen trouvé pour résoudre ce problème est de faire une autre
    fonction à qui on fournit request en argument (donc pas un override de
    get_absolute_url).

    TODO : vérifier si ces problèmes ont été résolus dans les màj de wagtail
    et modeltranslation
    """

    def debugged_get_url(self, request):
        parent = COFRootPage.objects.parent_of(self).live().first()
        burl = parent.relative_url(request.site)
        return burl + self.slug

    class Meta:
        verbose_name = "Page utilitaire"
        verbose_name_plural = "Pages utilitaires"