def test_store_files():
"""
Stores a file in the file system and creates a database record.
"""
user = mailinglist.create_user(text_msg['from'])
dbmessage = archive.store_message(list_name, text_msg)
dbfile = files.store_file(list_name, text_msg, "customerinterviews.txt", dbmessage)
assert dbfile.name == "customerinterviews.txt"
user = mailinglist.create_user(two_msg['from'])
dbmessage = archive.store_message(list_name, two_msg)
dbfile = files.store_file(list_name, two_msg, "sean-text.txt", dbmessage)
assert dbfile.name == "sean-text.txt"
def START(message, list_name=None, host=None):
"""
This is the central logic for doing most of the stuff
that makes Postosaurus worth paying money for. Right
now we pull a message off of the work queue, archive it
and extract any links.
"""
try:
#db message is the message record created in the database. The
#id generated is used as a key to extract the actual message
#from tokyo tyrant. We do this so that it's easy to maintain
# relational data, like what links are related to a message? A
# key value store is not good at storing this kind of data.
dbmessage = archive.store_message(list_name, message)
# store attached files for retrieval
for name in files.file_names(message):
files.store_file(list_name, message, name, dbmessage)
body = message.body()
if body:
urls = links.extract_urls_from_text(body)
for url in urls:
links.add_link(list_name, url, dbmessage)
transaction.commit()
except:
#queue up any messages that failed so we can diagnose
#and try later.
transaction.rollback()
q = queue.Queue("run/error")
q.push(message)
def test_add_message():
"""
Adds a message to the archive.
"""
dbmessage = archive.store_message(list_name, message)
assert dbmessage.subject == message['Subject']
assert json.dumps(archive.get_message(str(dbmessage.id)), sort_keys=True) == archive.to_json(message.base)