easyctf/easyctf-2017
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix file uploading

Browse source
This commit is contained in:
James Wang 2016-01-02 17:44:32 -05:00
parent 25fb22bc5c
commit 7532a327b8
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
server/api/problem.py
View file

@ -1,5 +1,6 @@
import hashlib import hashlib
import logger import logger
import os
from flask import Blueprint, session, request from flask import Blueprint, session, request
from flask import current_app as app from flask import current_app as app
@ -27,8 +28,9 @@ def problem_add():
problem = Problems(name, category, description, hint, flag, value) problem = Problems(name, category, description, hint, flag, value)
db.session.add(problem) db.session.add(problem)
db.session.commit()
files = request.files["imp-files"] files = request.files.getlist("files[]")
for _file in files: for _file in files:
filename = secure_filename(_file.filename) filename = secure_filename(_file.filename)
@ -36,11 +38,11 @@ def problem_add():
continue continue
folder = problem.name.replace(" ", "-") folder = problem.name.replace(" ", "-")
folder_path = os.path.join(os.path.normpath(app["UPLOAD_FOLDER"], folder)) folder_path = os.path.join(os.path.normpath(app.config["UPLOAD_FOLDER"]), folder)
if not folder_path: if not os.path.exists(folder_path):
os.makedirs(folder_path) os.makedirs(folder_path)
file_path = os.path.join(path, filename) file_path = os.path.join(folder_path, filename)
_file.save(file_path) _file.save(file_path)
db_file = Files(problem.pid, file_path) db_file = Files(problem.pid, file_path)
db.session.add(db_file) db.session.add(db_file)