Produits Entr'ouvert » Corbo

from datetime import datetime

import logging

from html2text import HTML2Text

from emails.django import Message

from lxml.etree import HTML as HTMLTree

from django.utils import timezone

from django.conf import settings

from django.db import models

from django.core.files.storage import DefaultStorage

from django.utils.translation import ugettext_lazy as _

from ckeditor.fields import RichTextField

channel_choices = (

    ('mailto', _('Email')),

    ('homepage', _('Homepage'))

)

logger = logging.getLogger(__name__)

class Category(models.Model):

    name = models.CharField(max_length=64, blank=False, null=False)

    ctime = models.DateTimeField(auto_now_add=True)

    def __unicode__(self):

        return self.name

class Announce(models.Model):

    category  = models.ForeignKey('Category', verbose_name=_('category'))

    title = models.CharField(_('title'), max_length=256,

                             help_text=_('maximum 256 characters'))

    text = RichTextField(_('Content'))

    publication_time = models.DateTimeField(_('publication time'), blank=True,

                                            null=True)

    expiration_time = models.DateTimeField(_('Expires on'), blank=True,

                                           null=True)

    ctime = models.DateTimeField(_('creation time'), auto_now_add=True)

    mtime = models.DateTimeField(_('modification time'), auto_now=True)

    def __unicode__(self):

        return u'{title} ({id}) at {mtime}'.format(title=self.title,

                                    id=self.id, mtime=self.mtime)

    def is_expired(self):

        if self.expiration_time:

            return self.expiration_time < timezone.now()

        return False

    def is_published(self):

        if self.is_expired():

            return False

        if self.publication_time:

            return self.publication_time <= timezone.now()

        return False

    class Meta:

        verbose_name = _('announce')

        ordering = ('-mtime',)

class Broadcast(models.Model):

    announce = models.ForeignKey(Announce, verbose_name=_('announce'))

    deliver_time = models.DateTimeField(_('Deliver time'), null=True)

    result = models.TextField(_('result'), blank=True)

    def __unicode__(self):

        if self.deliver_time:

            return u'announce {id} delivered via at {time}'.format(

                id=self.announce.id, time=self.deliver_time)

        return u'announce {id} to deliver'.format(id=self.announce.id)

    def send(self):

        subscriptions = self.announce.category.subscription_set.all()

        total_sent = 0

        handler = HTML2Text()

        m = Message(html=self.announce.text, subject=self.announce.title,

                    text=handler.handle(self.announce.text),

                    mail_from=settings.CORBO_DEFAULT_FROM_EMAIL)

        html_tree = HTMLTree(self.announce.text)

        storage = DefaultStorage()

        for img in html_tree.xpath('//img/@src'):

            img_path = img.lstrip(storage.base_url)

            m.attach(filename=img, data=storage.open(img_path))

            m.attachments[img].is_inline = True

        m.transformer.synchronize_inline_images()

        m.transformer.load_and_transform()

        m.transformer.save()

        for s in subscriptions:

            if not s.identifier:

                continue

            sent = m.send(to=s.identifier)

            if sent:

                total_sent += 1

                logger.info('Announce "%s" sent to %s', self.announce.title, s.identifier)

            else:

                logger.warning('Error occured while sending announce "%s" to %s.',

                               self.announce.title, s.identifier)

        self.result = total_sent

        self.deliver_time = timezone.now()

        self.save()

    class Meta:

        verbose_name = _('sent')

        ordering = ('-deliver_time',)

class Subscription(models.Model):

    category = models.ForeignKey('Category', verbose_name=_('Category'))

    uuid = models.CharField(_('User identifier'), max_length=128, blank=True)

    identifier = models.CharField(_('identifier'), max_length=128, blank=True,

            help_text=_('ex.: mailto, homepage, ...'))

    class Meta:

        unique_together = ('category', 'identifier', 'uuid')