diff options
| author | rubenwardy <rw@rubenwardy.com> | 2020-01-18 01:20:32 +0000 |
|---|---|---|
| committer | rubenwardy <rw@rubenwardy.com> | 2020-01-18 01:20:32 +0000 |
| commit | 095494f96f00b33dc9e956b094105c49b5bc558b (patch) | |
| tree | 017cbacda779f691ac0e38fca83e86e25864b0d9 /app/utils.py | |
| parent | 6f230ee4b27445a4d487b9166660dbafaf2f7912 (diff) | |
| download | cheatdb-095494f96f00b33dc9e956b094105c49b5bc558b.tar.xz | |
Improve Docker configurations
Diffstat (limited to 'app/utils.py')
| -rw-r--r-- | app/utils.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/app/utils.py b/app/utils.py index 102219d..5d1c8da 100644 --- a/app/utils.py +++ b/app/utils.py @@ -46,6 +46,8 @@ def randomString(n): return ''.join(random.choice(string.ascii_lowercase + \ string.ascii_uppercase + string.digits) for _ in range(n)) +assert(os.path.isdir(app.config["UPLOAD_DIR"]), "UPLOAD_DIR must exist") + def doFileUpload(file, fileType, fileTypeDesc): if not file or file is None or file.filename == "": flash("No selected file", "error") @@ -73,7 +75,7 @@ def doFileUpload(file, fileType, fileTypeDesc): file.stream.seek(0) filename = randomString(10) + "." + ext - file.save(os.path.join("app/public/uploads", filename)) + file.save(os.path.join(app.config["UPLOAD_DIR"], filename)) return "/uploads/" + filename def make_flask_user_password(plaintext_str): |