aboutsummaryrefslogtreecommitdiff
path: root/gn3/file_utils.py
diff options
context:
space:
mode:
authorBonfaceKilz2021-03-01 16:51:57 +0300
committerBonfaceKilz2021-03-08 21:09:58 +0300
commitd0347cecce5655caba96d330536a83279df28f0b (patch)
tree5d1420b10e2631b5891614011e3de1588ea3ad90 /gn3/file_utils.py
parentff29ce04a1f10af819770ccafd347daaba6b2bf7 (diff)
downloadgenenetwork3-d0347cecce5655caba96d330536a83279df28f0b.tar.gz
Extract files to "/TMPDIR/TOKEN"
TOKEN is the user token * gn3/file_utils.py (extract_uploaded_file): Add extra param "token". If a token is empty, create a new directory based off that token. * tests/unit/test_file_utils.py: Update failing tests.
Diffstat (limited to 'gn3/file_utils.py')
-rw-r--r--gn3/file_utils.py29
1 files changed, 14 insertions, 15 deletions
diff --git a/gn3/file_utils.py b/gn3/file_utils.py
index cda4f2b..16b2f94 100644
--- a/gn3/file_utils.py
+++ b/gn3/file_utils.py
@@ -2,7 +2,6 @@
import hashlib
import json
import os
-import shutil
import random
import string
import tarfile
@@ -59,27 +58,27 @@ def generate_random_n_string(n_length: int) -> str:
for _ in range(n_length))
-def extract_uploaded_file(gzipped_file, target_dir: str) -> Dict:
+def extract_uploaded_file(gzipped_file, target_dir: str, token="") -> Dict:
"""Get the (directory) hash of extracted contents of GZIPPED_FILE; and move
contents to TARGET_DIR/<dir-hash>.
"""
- tar_target_loc = os.path.join(target_dir,
- secure_filename(gzipped_file.filename))
- gzipped_file.save(tar_target_loc)
+ if not token:
+ token = (f"{generate_random_n_string(6)}-"
+ f"{generate_random_n_string(6)}")
+ tar_target_loc = os.path.join(
+ target_dir,
+ token,
+ secure_filename(gzipped_file.filename))
try:
- # Extract to "tar_target_loc/tempdir"
+ if not os.path.exists(os.path.join(target_dir, token)):
+ os.mkdir(os.path.join(target_dir, token))
+ gzipped_file.save(tar_target_loc)
+ # Extract to "tar_target_loc/token"
tar = tarfile.open(tar_target_loc)
- tar.extractall(
- path=os.path.join(target_dir, "tempdir"))
+ tar.extractall(path=tar_target_loc)
tar.close()
# pylint: disable=W0703
except Exception:
return {"status": 128, "error": "gzip failed to unpack file"}
- dir_hash = get_dir_hash(tar_target_loc)
- if os.path.exists(os.path.join(target_dir, dir_hash)):
- shutil.rmtree(os.path.join(target_dir, 'tempdir'))
- else:
- os.rename(os.path.join(target_dir, "tempdir"),
- os.path.join(target_dir, dir_hash))
- return {"status": 0, "token": dir_hash}
+ return {"status": 0, "token": token}