From a91204745eeb81958099480afd029cc15cb8b8d0 Mon Sep 17 00:00:00 2001 From: Kenneth Jao Date: Sat, 28 Apr 2018 23:34:45 -0400 Subject: [PATCH] Added file extensions --- SmearcarDB/server.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/SmearcarDB/server.py b/SmearcarDB/server.py index e9689a4..b3fb68c 100644 --- a/SmearcarDB/server.py +++ b/SmearcarDB/server.py @@ -154,8 +154,9 @@ def initial(): # GET method for files @app.route("/server/", methods=["GET"]) def file_return(lang_id): + extensions = dict(x.split(".") for x in os.listdir("F:/Programming Projects/SmearcarDB/files")) try: - return send_file('files/' + lang_id) + return send_file('files/' + lang_id+"."+extensions[str(lang_id)]) except Exception as e: return str(e) @@ -168,7 +169,7 @@ def source_add(): "password": request.form["password"] }, 2): f = request.files['file'] - write_path = os.path.join(working_dir, "files", request.form['lang_id']) + write_path = os.path.join(working_dir, "files", request.form['lang_id']+"."+f.filename.split(".")[-1]) # with open(write_path, 'w') as a: # a.write() f.save(write_path)