mirror/django-mailbox
1
0
Fork 1
mirror of https://github.com/coddingtonbear/django-mailbox.git synced 2026-07-09 22:38:19 +02:00
Code Activity

Cache email object in Message

Browse source
This commit is contained in:
Adam Dobrawy 2017-07-15 19:41:55 +02:00
parent 9f8d653ada
commit 67900104a2
1 changed files with 16 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

29
django_mailbox/models.py
View file

@ -345,6 +345,7 @@ class Mailbox(models.Model):
def _process_message(self, message):
msg = Message()
msg._email_object = message
settings = utils.get_settings()
if settings['store_original_message']:
@ -710,20 +711,22 @@ class Message(models.Model):
(https://docs.python.org/2/library/email.message.html)
"""
if self.eml:
if self.eml.name.endswith('.gz'):
body = gzip.GzipFile(fileobj=self.eml).read()
if not hasattr(self, '_email_object'): # Cache fill
if self.eml:
if self.eml.name.endswith('.gz'):
body = gzip.GzipFile(fileobj=self.eml).read()
else:
self.eml.open()
body = self.eml.file.read()
self.eml.close()
else:
self.eml.open()
body = self.eml.file.read()
self.eml.close()
else:
body = self.get_body()
if six.PY3:
flat = email.message_from_bytes(body)
else:
flat = email.message_from_string(body)
return self._rehydrate(flat)
body = self.get_body()
if six.PY3:
flat = email.message_from_bytes(body)
else:
flat = email.message_from_string(body)
self._email_object = self._rehydrate(flat)
return self._email_object
def delete(self, *args, **kwargs):
"""Delete this message and all stored attachments."""