welco/welco/sources/mail/models.py

# welco - multichannel request processing
# Copyright (C) 2015  Entr'ouvert
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import subprocess

from django.contrib.contenttypes import generic
from django.core.urlresolvers import reverse
from django.db import models
from django.db.models.signals import post_save
from django.dispatch import receiver
from django.utils.translation import ugettext_lazy as _

from welco.qualif.models import Association

class Mail(models.Model):

    class Meta:
        verbose_name = _('Mail')

    content = models.FileField(_('Content'))
    post_date = models.DateField(_('Post Date'), null=True)
    registered_mail_number = models.CharField(_('Registered Mail Number'),
            null=True, max_length=50)
    note = models.TextField(_('Note'), null=True)

    # common to all source types:
    status = models.CharField(_('Status'), blank=True, max_length=50)
    contact_id = models.CharField(max_length=50, null=True)
    associations = generic.GenericRelation(Association,
            content_type_field='source_type', object_id_field='source_pk')

    creation_timestamp = models.DateTimeField(auto_now_add=True)
    last_update_timestamp = models.DateTimeField(auto_now=True)

    @classmethod
    def get_qualification_form_class(cls):
        from .forms import MailQualificationForm
        return MailQualificationForm

    def get_qualification_form(self):
        return self.get_qualification_form_class()({
            'post_date': self.post_date,
            'registered_mail_number': self.registered_mail_number})

    @classmethod
    def get_qualification_form_submit_url(cls):
        return reverse('qualif-mail-save')


@receiver(post_save, sender=Mail)
def create_thumbnail(sender, instance, created, **kwargs):
    if not created:
        return
    subprocess.call(['gm', 'convert', '-geometry', '200x',
        instance.content.file.name, instance.content.file.name + '.png'])