Clean up code a bit

master
Sven Slootweg 12 years ago
parent bfc42dc783
commit f4e3b856cb

@ -84,14 +84,13 @@ for email_file in file_list:
textbody = ""
htmlbody = ""
attachment_list = []
sha1_hash = hashlib.sha1("%s/%s/%s/%s" % (message['from'], message['to'], message['message-id'], subject)).hexdigest()
message_parts = [find_submessages(message)]
message_parts = flatten(message_parts)
attachment_list = []
for part in message_parts:
if part.get_filename() is None:
# Part of the message
@ -112,14 +111,12 @@ for email_file in file_list:
attachment_extension = os.path.splitext(attachment_filename)[1][1:]
attachment_size = len(attachment_data)
attachment_destination = "%s/%s.%s" % (options['attachment_dir'], attachment_sha1, attachment_extension)
#print "Attachment found of type %s: %s (%s)" % (attachment_type, attachment_filename, attachment_destination)
attachment_list.append((attachment_filename, attachment_type, attachment_sha1, attachment_size))
attachment_file = open(attachment_destination, "w")
attachment_file.write(attachment_data)
attachment_file.close()
attachment_list.append((attachment_filename, attachment_type, attachment_sha1, attachment_size))
timestamp = 0
new_row = (getheader(message['message-id']), getheader(message['from']), getheader(message['to']), getheader(subject), timestamp, textbody, htmlbody, sha1_hash)
@ -130,6 +127,7 @@ for email_file in file_list:
for attachment in attachment_list:
new_row = (sha1_hash, attachment[0], attachment[1], attachment[2], attachment[3])
cursor.execute("INSERT INTO attachments VALUES (?, ?, ?, ?, ?)", new_row)
print "Successfully inserted %d attachment(s) for %s." % (len(attachment_list), sha1_hash)
database.commit()

Loading…
Cancel
Save