List saving and sorting

master
Sven Slootweg 13 years ago
parent 8eb58adf55
commit 36a398549e

@ -1,6 +1,6 @@
#!/usr/bin/python #!/usr/bin/python
import os, argparse, hashlib, sqlite3, time, shutil, cgi import os, argparse, hashlib, sqlite3, time, shutil, cgi, re
from datetime import datetime from datetime import datetime
parser = argparse.ArgumentParser(description='Renders static HTML pages and indexes from an SQLite database of emails and an attachment folder.') parser = argparse.ArgumentParser(description='Renders static HTML pages and indexes from an SQLite database of emails and an attachment folder.')
@ -47,6 +47,8 @@ try:
except OSError: except OSError:
pass pass
email_list = []
for message_id, sender, recipient, subject, timestamp, textbody, htmlbody, sha1_hash in cursor.execute("SELECT * FROM emails"): for message_id, sender, recipient, subject, timestamp, textbody, htmlbody, sha1_hash in cursor.execute("SELECT * FROM emails"):
sender = cgi.escape(sender, True) sender = cgi.escape(sender, True)
@ -122,4 +124,41 @@ for message_id, sender, recipient, subject, timestamp, textbody, htmlbody, sha1_
open('%s/messages/%s_html.html' % (options['output_dir'], sha1_hash), 'w').write(generated.encode('UTF-8')) open('%s/messages/%s_html.html' % (options['output_dir'], sha1_hash), 'w').write(generated.encode('UTF-8'))
print "Successfully generated HTML version of %s." % sha1_hash print "Successfully generated HTML version of %s." % sha1_hash
try:
snippet = "%s..." % re.search("^(.{0,200})\\b", textbody, re.DOTALL).group(1)
except AttributeError:
snippet = ""
email_list.append((sha1_hash, available_text, available_html, sender, recipient, subject, timestamp, len(attachment_list), snippet))
# Sort by timestamp, ascending
sorted_list = sorted(email_list, key=lambda email: email[6].lower())
# Sort by timestamp, descending
sorted_list.reverse()
# Sort by sender, ascending
sorted_list = sorted(email_list, key=lambda email: email[3].lower())
# Sort by sender, descending
sorted_list.reverse()
# Sort by recipient, ascending
sorted_list = sorted(email_list, key=lambda email: email[4].lower())
# Sort by recipient, descending
sorted_list.reverse()
# Sort by hash, ascending
sorted_list = sorted(email_list, key=lambda email: email[0].lower())
# Sort by hash, descending
sorted_list.reverse()
# Sort by subject, ascending
sorted_list = sorted(email_list, key=lambda email: email[5].lower())
# Sort by subject, descending
sorted_list.reverse()
shutil.copy('%s/style.css' % options['template_dir'], '%s/style.css' % options['output_dir']) shutil.copy('%s/style.css' % options['template_dir'], '%s/style.css' % options['output_dir'])

Loading…
Cancel
Save