From b96fd630bc0862c7a7af68f3b238f2148df5c786 Mon Sep 17 00:00:00 2001 From: Will Larson Date: Mon, 2 Jun 2008 13:38:01 +0900 Subject: [PATCH] Files uplaoded via LFE will no longer risk overwriting each other. --- editor/views.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/editor/views.py b/editor/views.py index 99c5d78..badd985 100644 --- a/editor/views.py +++ b/editor/views.py @@ -15,7 +15,7 @@ """ -import datetime +import datetime, os.path from django.template import RequestContext from django.shortcuts import render_to_response from django.http import Http404, HttpResponseRedirect, HttpResponseServerError @@ -215,7 +215,12 @@ def add_resource(request): title = request.POST['title'] markdown_id = request.POST['markdown_id'] filename = file['filename'] - fd = open('%s/lifeflow/resource/%s' %(settings.MEDIA_ROOT, filename), 'wb') + filebase = '%s/lifeflow/resource/' % settings.MEDIA_ROOT + filepath = "%s%s" % (filebase, filename) + while (os.path.isfile(filepath)): + filename = "_%s" % filename + filepath = "%s%s" % (filebase, filename) + fd = open(filepath, 'wb') fd.write(file['content']) fd.close() rec = Resource(title=title, markdown_id=markdown_id, content="lifeflow/resource/%s" % filename)