federico/django-mailbox
1
0
Fork 0
forked from mirror/django-mailbox
Code Activity

Close attachments

Browse source
This commit is contained in:
Martin Manzo 2023-08-11 10:33:45 -03:00
parent 462fdd3e49
commit 7e250b2504
2 changed files with 18 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -14,3 +14,4 @@ dummy_project/*
messages
*.sqlite3
.idea/
venv/

12
django_mailbox/models.py
View file

@ -677,7 +677,6 @@ class Message(models.Model):
def _rehydrate(self, msg):
new = EmailMessage()
settings = utils.get_settings()
if msg.is_multipart():
for header, value in msg.items():
new[header] = value
@ -696,10 +695,11 @@ class Message(models.Model):
if encoding and encoding.lower() == 'quoted-printable':
# Cannot use `email.encoders.encode_quopri due to
# bug 14360: http://bugs.python.org/issue14360
with open(attachment.document.path, 'rb') as f:
output = BytesIO()
encode_quopri(
BytesIO(
attachment.document.read()
f.read()
),
output,
quotetabs=True,
@ -711,9 +711,8 @@ class Message(models.Model):
del new['Content-Transfer-Encoding']
new['Content-Transfer-Encoding'] = 'quoted-printable'
else:
new.set_payload(
attachment.document.read()
)
with open(attachment.document.path, 'rb') as f:
new.set_payload(f.read())
del new['Content-Transfer-Encoding']
encode_base64(new)
except MessageAttachment.DoesNotExist:
@ -777,9 +776,8 @@ class Message(models.Model):
if self.eml.name.endswith('.gz'):
body = gzip.GzipFile(fileobj=self.eml).read()
else:
self.eml.open()
with self.eml.open():
body = self.eml.file.read()
self.eml.close()
else:
body = self.get_body()
flat = email.message_from_bytes(body)