Reorganised models

publichealth/home/models/__init__.py

@@ -0,0 +1,2 @@
+from .models import *
+from .snippets import *

publichealth/home/models/models.py

@@ -0,0 +1,200 @@
+# -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+
+from django.db import models
+from modelcluster.fields import ParentalKey
+
+from wagtail.wagtailcore import blocks
+from wagtail.wagtailcore.models import Page, Orderable
+from wagtail.wagtailcore.fields import StreamField, RichTextField
+from wagtail.wagtailadmin.edit_handlers import FieldPanel, StreamFieldPanel, InlinePanel, MultiFieldPanel
+from wagtail.wagtailimages.blocks import ImageChooserBlock
+from wagtail.wagtailimages.edit_handlers import ImageChooserPanel
+from wagtail.wagtailsearch import index
+
+from puput.models import EntryPage
+
+from ..util import TranslatedField
+
+class ArticleIndexPage(Page):
+    title_fr = models.CharField(max_length=255, default="")
+    trans_title = TranslatedField(
+        'title',
+        'title_fr',
+    )
+
+    intro_de = RichTextField(default='', blank=True)
+    intro_fr = RichTextField(default='', blank=True)
+    trans_intro = TranslatedField(
+        'intro_de',
+        'intro_fr',
+    )
+
+    content_panels = Page.content_panels + [
+        FieldPanel('intro_de'),
+        FieldPanel('title_fr'),
+        FieldPanel('intro_fr'),
+    ]
+
+    def get_context(self, request):
+        context = super(ArticleIndexPage, self).get_context(request)
+        articles = ArticlePage.objects.child_of(self).live()
+        context['articles'] = articles
+        return context
+
+    subpage_types = ['home.ArticlePage', 'home.ArticleIndexPage']
+    class Meta:
+        verbose_name = "Rubrik"
+
+class ArticlePage(Page):
+    title_fr = models.CharField(max_length=255, default="")
+    trans_title = TranslatedField(
+        'title',
+        'title_fr',
+    )
+
+    intro_de = RichTextField(default='', blank=True)
+    intro_fr = RichTextField(default='', blank=True)
+    trans_intro = TranslatedField(
+        'intro_de',
+        'intro_fr',
+    )
+
+    body_de = StreamField([
+        ('paragraph', blocks.RichTextBlock()),
+        ('image', ImageChooserBlock()),
+        ('section', blocks.CharBlock(classname="full title")),
+    ], null=True, blank=True)
+    body_fr = StreamField([
+        ('paragraph', blocks.RichTextBlock()),
+        ('image', ImageChooserBlock()),
+        ('section', blocks.CharBlock(classname="full title")),
+    ], null=True, blank=True)
+    trans_body = TranslatedField(
+        'body_de',
+        'body_fr',
+    )
+
+    date = models.DateField("Date", null=True, blank=True)
+    feed_image = models.ForeignKey(
+        'wagtailimages.Image',
+        null=True,
+        blank=True,
+        on_delete=models.SET_NULL,
+        related_name='+'
+    )
+
+    search_fields = Page.search_fields + [
+        index.SearchField('body_de'),
+        index.SearchField('body_fr'),
+        index.SearchField('title'),
+        index.SearchField('title_fr'),
+        index.SearchField('intro_de'),
+        index.SearchField('intro_fr'),
+    ]
+    content_panels = [
+        MultiFieldPanel([
+            FieldPanel('title'),
+            FieldPanel('intro_de'),
+            StreamFieldPanel('body_de'),
+        ], heading="Deutsch"),
+        MultiFieldPanel([
+            FieldPanel('title_fr'),
+            FieldPanel('intro_fr'),
+            StreamFieldPanel('body_fr'),
+        ], heading="Français"),
+        ImageChooserPanel('feed_image'),
+    ]
+    promote_panels = [
+        FieldPanel('date'),
+        InlinePanel('related_links', label="Links"),
+        MultiFieldPanel(Page.promote_panels, "Common page configuration"),
+    ]
+    parent_page_types = ['home.ArticleIndexPage']
+    subpage_types = []
+    class Meta:
+        verbose_name = "Artikel"
+
+class ArticleRelatedLink(Orderable):
+    page = ParentalKey(ArticlePage, related_name='related_links')
+    name = models.CharField(max_length=255)
+    url = models.URLField()
+    panels = [
+        FieldPanel('name'),
+        FieldPanel('url'),
+    ]
+
+class InfoBlock(blocks.StructBlock):
+    title = blocks.CharBlock(required=True)
+    photo = ImageChooserBlock()
+    summary = blocks.RichTextBlock(required=True)
+    action = blocks.CharBlock()
+    url = blocks.URLBlock()
+
+class HomePage(Page):
+    intro_de = RichTextField(default='')
+    intro_fr = RichTextField(default='')
+    trans_intro = TranslatedField(
+        'intro_de',
+        'intro_fr',
+    )
+
+    body_de = RichTextField(default='', blank=True)
+    body_fr = RichTextField(default='', blank=True)
+    trans_body = TranslatedField(
+        'body_de',
+        'body_fr',
+    )
+
+    infos_de = StreamField([
+        ('info', InfoBlock())
+    ], null=True, blank=True)
+    infos_fr = StreamField([
+        ('info', InfoBlock())
+    ], null=True, blank=True)
+    trans_infos = TranslatedField(
+        'infos_de',
+        'infos_fr',
+    )
+
+    content_panels = Page.content_panels + [
+        MultiFieldPanel([
+            FieldPanel('intro_de', classname="full"),
+            FieldPanel('body_de', classname="full"),
+            StreamFieldPanel('infos_de'),
+        ], heading="Deutsch"),
+            MultiFieldPanel([
+            FieldPanel('intro_fr', classname="full"),
+            FieldPanel('body_fr', classname="full"),
+            StreamFieldPanel('infos_fr'),
+        ], heading="Français"),
+    ]
+
+    @property
+    def featured(self):
+        # Get list of live pages that are descendants of this page
+        articles = ArticlePage.objects.live() #.descendant_of(self)
+        # Order by most recent date first
+        #articles = articles.order_by('-date')
+        return articles[:4]
+
+    @property
+    def newsfeed(self):
+        # Get list of latest news
+        # TODO: fetch children of 'News (DE)'
+        entries = EntryPage.objects.live().descendant_of(self)
+        # Order by most recent date first
+        entries = entries.order_by('-date')
+        return entries[:4]
+
+    def get_context(self, request):
+        # Update template context
+        context = super(HomePage, self).get_context(request)
+        context['featured'] = self.featured
+        context['newsfeed'] = self.newsfeed
+        return context
+
+    parent_page_types = []
+    class Meta:
+        verbose_name = "Frontpage"

publichealth/home/models/snippets.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+
+from django.db import models
+
+from wagtail.wagtailsnippets.models import register_snippet
+
+from wagtail.wagtailcore.models import Page
+from wagtail.wagtailadmin.edit_handlers import FieldPanel
+
+from ..util import TranslatedField
+
+@register_snippet
+class Contact(models.Model):
+    title = models.CharField(max_length=255, default="")
+    title_fr = models.CharField(max_length=255, default="")
+    trans_title = TranslatedField(
+        'title',
+        'title_fr',
+    )
+    address = models.TextField(default="", blank=True)
+    phone = models.CharField(max_length=40, default="")
+    email = models.CharField(max_length=40, default="")
+    www = models.URLField(null=True, blank=True)
+
+    panels = Page.content_panels + [
+        FieldPanel('title_fr'),
+        FieldPanel('address'),
+        FieldPanel('phone'),
+        FieldPanel('email'),
+        FieldPanel('www'),
+    ]
+
+    def phone_link(self):
+        return 'tel:%s' % self.phone.replace(' ', '')
+    def email_link(self):
+        return 'mailto:%s' % self.email
+    def www_domain(self):
+        return self.www.replace('http://', '').replace('https://', '')
+    def __str__(self):
+        return self.trans_title