You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
3.2 KiB
Python
81 lines
3.2 KiB
Python
#!/usr/bin/python
|
|
|
|
import os, argparse, hashlib, sqlite3, time, shutil
|
|
|
|
parser = argparse.ArgumentParser(description='Renders static HTML pages and indexes from an SQLite database of emails and an attachment folder.')
|
|
|
|
parser.add_argument('-o', dest='output_dir', action='store', default='rendered_files',
|
|
help='path of the directory where rendered files should be stored')
|
|
|
|
parser.add_argument('-t', dest='template_dir', action='store', default='templates',
|
|
help='path where the template files are')
|
|
|
|
parser.add_argument('-i', dest='title', action='store', default='Inbox',
|
|
help='title for the rendered pages')
|
|
|
|
parser.add_argument('-d', dest='database', action='store', default='emails.db',
|
|
help='path of the database that should be used to render the e-mail files')
|
|
|
|
parser.add_argument('-a', dest='attachment_dir', action='store', default='attachments',
|
|
help='path where attachments are stored')
|
|
|
|
args = parser.parse_args()
|
|
options = vars(args)
|
|
|
|
def escape_html_chars(text):
|
|
return text.replace("&", "&").replace('"', """).replace("<", "<").replace(">", ">")
|
|
|
|
if os.path.isfile(options['database']) == False:
|
|
print "Database file not found. Use the -d switch to specify a custom database path."
|
|
exit(1)
|
|
|
|
# Connect to database
|
|
database = sqlite3.connect(options['database'])
|
|
cursor = database.cursor()
|
|
|
|
# Load templates
|
|
template_message = open('%s/message.html' % options['template_dir']).read()
|
|
|
|
# Create output directory if necessary
|
|
try:
|
|
os.makedirs("%s/messages" % options['output_dir'])
|
|
except OSError:
|
|
pass
|
|
|
|
for message_id, sender, recipient, subject, timestamp, textbody, htmlbody, sha1_hash in cursor.execute("SELECT * FROM emails"):
|
|
|
|
sender = escape_html_chars(sender)
|
|
recipient = escape_html_chars(recipient)
|
|
subject = escape_html_chars(subject)
|
|
message_id = escape_html_chars(message_id)
|
|
|
|
versions = {}
|
|
|
|
if textbody != "":
|
|
available_text = True
|
|
versions['Plaintext'] = '%s_text.html' % sha1_hash
|
|
else:
|
|
available_text = False
|
|
|
|
if htmlbody != "":
|
|
available_html = True
|
|
versions['HTML'] = '%s_html.html' % sha1_hash
|
|
else:
|
|
available_html = False
|
|
|
|
version_list = "".join('<a href="%s" class="version">%s</a>' % (value, key) for key, value in versions.viewitems())
|
|
|
|
if available_text == True:
|
|
# Text version
|
|
body = "<pre>%s</pre>" % textbody
|
|
generated = template_message % {'subject': subject, 'date': timestamp, 'from': sender, 'to': recipient, 'body': body, 'title': options['title'], 'version': "Plaintext version", 'index': "../index.html", 'versions': version_list}
|
|
open('%s/messages/%s_text.html' % (options['output_dir'], sha1_hash), 'w').write(generated.encode('UTF-8'))
|
|
|
|
if available_html == True:
|
|
# HTML version
|
|
body = htmlbody
|
|
generated = template_message % {'subject': subject, 'date': timestamp, 'from': sender, 'to': recipient, 'body': body, 'title': options['title'], 'version': "HTML version", 'index': "../index.html", 'versions': version_list}
|
|
open('%s/messages/%s_html.html' % (options['output_dir'], sha1_hash), 'w').write(generated.encode('UTF-8'))
|
|
|
|
shutil.copy('%s/style.css' % options['template_dir'], '%s/style.css' % options['output_dir'])
|