From d9bb875a27526be5acea8d7d364333515204780a Mon Sep 17 00:00:00 2001
From: Daniel Markstedt <markstedt@gmail.com>
Date: Thu, 23 Sep 2021 08:45:46 -0700
Subject: [PATCH] Unzip and delete zip files after upload; validate upload file
 suffix

---
 src/web/requirements.txt     | 1 +
 src/web/templates/index.html | 1 +
 src/web/web.py               | 9 ++++++++-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/web/requirements.txt b/src/web/requirements.txt
index e853db3d..bd0bdec8 100644
--- a/src/web/requirements.txt
+++ b/src/web/requirements.txt
@@ -12,3 +12,4 @@ zope.event==4.5.0
 zope.interface==5.1.2
 protobuf==3.17.3
 pydrop==0.0.6
+zipfile
\ No newline at end of file
diff --git a/src/web/templates/index.html b/src/web/templates/index.html
index 96afc8c3..12300b01 100644
--- a/src/web/templates/index.html
+++ b/src/web/templates/index.html
@@ -180,6 +180,7 @@
     <script type="application/javascript">
     Dropzone.options.dropper = {
         paramName: 'file',
+        acceptedFiles: '{{valid_file_suffix}}'
         chunking: true,
         forceChunking: true,
         url: '/files/upload',
diff --git a/src/web/web.py b/src/web/web.py
index 96367802..9b43c88d 100644
--- a/src/web/web.py
+++ b/src/web/web.py
@@ -526,7 +526,14 @@ def upload_file():
         log.debug(f"Chunk {current_chunk + 1} of {total_chunks} "
                   f"for file {file.filename} completed.")
 
-    return make_response(("File upload successful!", 200))
+    if file.filename.endswith("zip"):
+        from zipfile import ZipFile
+        with ZipFile(file_path, 'r') as zip:
+            zip.extractall()
+        delete_file(file_path)
+        return make_response(("File upload and unzip successful!", 200))
+    else:
+        return make_response(("File upload successful!", 200))
 
 
 @app.route("/files/create", methods=["POST"])