summaryrefslogtreecommitdiff
path: root/postgresqleu/mailqueue/backendforms.py
blob: 7e1e43f2b5e33934123b4fa687a5d8d1732a3986 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from django import forms
from django.utils import timezone

from collections import OrderedDict

from postgresqleu.util.widgets import StaticTextWidget
from postgresqleu.util.backendforms import BackendForm
from postgresqleu.mailqueue.models import QueuedMail
from postgresqleu.mailqueue.util import parse_mail_content, recursive_parse_attachments_from_message


class BackendMailqueueAttachmentManager:
    title = 'Attachments'
    singular = 'attachment'
    can_add = False

    def get_list(self, instance):
        for id, name, ctype, content in recursive_parse_attachments_from_message(instance.parsed_msg):
            yield id, name, ctype


class BackendMailqueueForm(BackendForm):
    decoded = forms.CharField(label="Decoded message", widget=StaticTextWidget(monospace=True))

    list_fields = ['sendtime', 'regtime', 'sendtime', 'sender', 'receiver', 'subject', ]
    helplink = 'mail'
    readonly_fields = ['sender', 'receiver', 'sendtime', 'subject', 'fullmsg', ]
    linked_objects = OrderedDict({
        'attachments': BackendMailqueueAttachmentManager(),
    })

    class Meta:
        model = QueuedMail
        fields = ['sender', 'receiver', 'sendtime', 'subject', 'fullmsg', ]

    def fix_fields(self):
        self.initial['decoded'] = self.parsed_content().decode('utf8', errors='ignore').replace("\n", "<br/>")

    def parsed_content(self):
        if not hasattr(self.instance, 'parsed_msg') or not hasattr(self.instance, 'parsed_txt'):
            self.instance.parsed_msg, self.instance.parsed_txt = parse_mail_content(self.instance.fullmsg)

        return self.instance.parsed_txt

    @classmethod
    def get_rowclass_and_title(self, obj, cache):
        if obj.sendtime < timezone.now():
            return "warning", None
        else:
            return "", None