From e3b5a6f93982310c36b36db7e677a115e34065e4 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 7 Jul 2020 11:10:07 +0200 Subject: build --- .../cli/app/controllers/upload_controller.py | 24 +++++++++++++--------- 1 file changed, 14 insertions(+), 10 deletions(-) (limited to 'animism-align/cli/app/controllers/upload_controller.py') diff --git a/animism-align/cli/app/controllers/upload_controller.py b/animism-align/cli/app/controllers/upload_controller.py index aa29553..0b3b4c5 100644 --- a/animism-align/cli/app/controllers/upload_controller.py +++ b/animism-align/cli/app/controllers/upload_controller.py @@ -93,14 +93,17 @@ class UploadView(FlaskView): session.close() return jsonify(response) + tag = request.form.get('tag') uploaded_im_fn = secure_filename(file.filename) - uploaded_im_abspath = os.path.join(app_cfg.DIR_UPLOADS, sha256_tree(sha256)) + uploaded_im_abspath = os.path.join(app_cfg.DIR_UPLOADS, tag) uploaded_im_fullpath = os.path.join(uploaded_im_abspath, uploaded_im_fn) + uploaded_im_stored_fn = os.path.join(tag, uploaded_im_fn) os.makedirs(uploaded_im_abspath, exist_ok=True) - nparr.tofile(uploaded_im_fullpath) + # nparr.tofile(uploaded_im_fullpath) + file.save(uploaded_im_fullpath) - upload = Upload(username=username, fn=uploaded_im_fn, sha256=sha256, ext=ext) + upload = Upload(username=username, fn=uploaded_im_stored_fn, sha256=sha256, ext=ext) session.add(upload) session.commit() response = { @@ -124,13 +127,14 @@ class UploadView(FlaskView): }) sha256 = upload.sha256 - - uploaded_im_fn = secure_filename(sha256 + upload.ext) - uploaded_im_abspath = os.path.join(app_cfg.DIR_UPLOADS, sha256_tree(sha256)) - uploaded_im_fullpath = os.path.join(uploaded_im_abspath, uploaded_im_fn) - if os.path.exists(uploaded_im_fullpath): - print("Removing " + uploaded_im_fullpath) - os.remove(uploaded_im_fullpath) + fn = upload.fn + + # uploaded_im_fn = secure_filename(fn) + uploaded_im_abspath = os.path.join(app_cfg.DIR_UPLOADS, fn) + # uploaded_im_fullpath = os.path.join(uploaded_im_abspath, fn) + if os.path.exists(uploaded_im_abspath): + print("Removing " + uploaded_im_abspath) + os.remove(uploaded_im_abspath) session.delete(upload) session.commit() -- cgit v1.2.3-70-g09d2