public-health-ch/publichealth/home/models/snippets.py

# -*- 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.EmailField(max_length=100, 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 trans_title_styled(self):
        v = self.trans_title.split(' ')
        if len(v) != 3: return v
        return "<strong>%s %s</strong> %s" % tuple(v)
    def __str__(self):
        return self.trans_title