pgweb/mailqueue/admin.py

   1 from django.contrib import admin
   2
   3 from email.parser import Parser
   4
   5 from .models import QueuedMail
   6
   7
   8 class QueuedMailAdmin(admin.ModelAdmin):
   9     model = QueuedMail
  10     readonly_fields = ('parsed_content', )
  11
  12     def parsed_content(self, obj):
  13         # We only try to parse the *first* piece, because we assume
  14         # all our emails are trivial.
  15         try:
  16             parser = Parser()
  17             msg = parser.parsestr(obj.fullmsg)
  18             b = msg.get_payload(decode=True)
  19             if b:
  20                 return b
  21
  22             pl = msg.get_payload()
  23             for p in pl:
  24                 b = p.get_payload(decode=True)
  25                 if b:
  26                     return b
  27             return "Could not find body"
  28         except Exception as e:
  29             return "Failed to get body: %s" % e
  30
  31     parsed_content.short_description = 'Parsed mail'
  32
  33
  34 admin.site.register(QueuedMail, QueuedMailAdmin)