diff options
Diffstat (limited to 'app/views/utils.py')
| -rw-r--r-- | app/views/utils.py | 28 |
1 files changed, 25 insertions, 3 deletions
diff --git a/app/views/utils.py b/app/views/utils.py index 7a5303e..488e32e 100644 --- a/app/views/utils.py +++ b/app/views/utils.py @@ -1,9 +1,31 @@ -from flask import request +from flask import request, flash +from app import app +import random, string, os + +def getExtension(filename): + return filename.rsplit(".", 1)[1].lower() if "." in filename else None def isFilenameAllowed(filename, exts): - return "." in filename and \ - filename.rsplit(".", 1)[1].lower() in exts + return getExtension(filename) in exts def shouldReturnJson(): return "application/json" in request.accept_mimetypes and \ not "text/html" in request.accept_mimetypes + +def randomString(n): + return ''.join(random.choice(string.ascii_lowercase + \ + string.ascii_uppercase + string.digits) for _ in range(n)) + +def doFileUpload(file, allowedExtensions, fileTypeName): + if not file or file is None or file.filename == "": + flash("No selected file", "error") + return None + + ext = getExtension(file.filename) + if ext is None or not ext in allowedExtensions: + flash("Please upload load " + fileTypeName, "error") + return None + + filename = randomString(10) + "." + ext + file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename)) + return "/uploads/" + filename |