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

Only store text/plain and text/html in body field

Browse source 
When parsing multipart message it was storing the whole message as
string in database which could lead to database errors. Only want to
store the actual message text
This commit is contained in:
Patrick Craston 2013-03-01 16:35:26 +00:00
parent 09fbf4ea56
commit 8fac7c9711
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
django_mailbox/models.py
View file

@ -162,7 +162,14 @@ class Mailbox(models.Model):
msg.message_id = message['message-id'][0:255]
msg.from_header = message['from']
msg.to_header = message['to']
msg.body = message.as_string()
# if message is multipart want to only store the actual message body in database (not attachments)
if message.is_multipart():
for part in message.walk():
if part.get_content_type() == 'text/plain' or part.get_content_type() == 'text/html':
msg.body = part.get_payload()
# else store the whole message in database
else:
msg.body = message.as_string()
if message['in-reply-to']:
try:
msg.in_reply_to = Message.objects.filter(message_id=message['in-reply-to'])[0]