should store uploads per-episode now

1 files changed, 8 insertions, 2 deletions

diff --git a/animism-align/cli/app/controllers/upload_controller.py b/animism-align/cli/app/controllers/upload_controller.py
index 7122e53..634f15d 100644
--- a/animism-align/cli/app/controllers/upload_controller.py
+++ b/animism-align/cli/app/controllers/upload_controller.py
@@ -64,6 +64,12 @@ class UploadView(FlaskView):
     except:
       raise APIError('No tag specified')
 
+    try:
+      episode_id = request.form.get('episode_id')
+      # print(episode_id)
+    except:
+      raise APIError('No episode_id specified')
+
     if 'image' in request.files:
       file = request.files['image']
       # print(fn)
@@ -85,7 +91,7 @@ class UploadView(FlaskView):
     file.seek(0)
 
     uploaded_im_fn = secure_filename(file.filename)
-    uploaded_im_abspath = os.path.join(app_cfg.DIR_UPLOADS, tag)
+    uploaded_im_abspath = os.path.join(app_cfg.DIR_UPLOADS, str(episode_id), tag)
     uploaded_im_fullpath = os.path.join(uploaded_im_abspath, uploaded_im_fn)
 
     session = Session()
@@ -107,7 +113,7 @@ class UploadView(FlaskView):
     os.makedirs(uploaded_im_abspath, exist_ok=True)
     file.save(uploaded_im_fullpath)
 
-    upload = Upload(username=username, tag=tag, fn=uploaded_im_fn, sha256=sha256, ext=ext)
+    upload = Upload(username=username, tag=tag, fn=uploaded_im_fn, sha256=sha256, ext=ext, episode_id=episode_id)
     session.add(upload)
     session.commit()
     response = {