Refactor filename generation in the public store

Just a small refactoring of the filename setup in the
public store. Very simple.
This commit is contained in:
Elrond 2011-06-08 23:22:11 +02:00
parent 9150244afa
commit 180bdbde93

View File

@ -24,6 +24,13 @@ from mediagoblin.globals import database, queue_store, public_store
THUMB_SIZE = 200, 200
def create_pub_filepath(entry, filename):
return public_store.get_unique_filepath(
['media_entries',
unicode(entry['_id']),
filename])
@task
def process_media_initial(media_id):
entry = database.MediaEntry.one(
@ -36,10 +43,7 @@ def process_media_initial(media_id):
thumb = Image.open(queued_file)
thumb.thumbnail(THUMB_SIZE, Image.ANTIALIAS)
thumb_filepath = public_store.get_unique_filepath(
['media_entries',
unicode(entry['_id']),
'thumbnail.jpg'])
thumb_filepath = create_pub_filepath(entry, 'thumbnail.jpg')
with public_store.get_file(thumb_filepath, 'w') as thumb_file:
thumb.save(thumb_file, "JPEG")
@ -49,15 +53,13 @@ def process_media_initial(media_id):
queued_file = queue_store.get_file(queued_filepath, 'rb')
with queued_file:
main_filepath = public_store.get_unique_filepath(
['media_entries',
unicode(entry['_id']),
queued_filepath[-1]])
main_filepath = create_pub_filepath(entry, queued_filepath[-1])
with public_store.get_file(main_filepath, 'wb') as main_file:
main_file.write(queued_file.read())
queue_store.delete_file(queued_filepath)
entry['queued_media_file'] = []
media_files_dict = entry.setdefault('media_files', {})
media_files_dict['thumb'] = thumb_filepath
media_files_dict['main'] = main_filepath