def upload_file(): # check if the post request has the file part if 'file' not in request.files: return jsonify(code=1002, result="File can't empty!") file_instance = request.files['file'] if file_instance.filename == '': return jsonify(code=1002, result="File name can't empty!") if file_instance and common.allowed_file(file_instance.filename): filename = secure_filename(file_instance.filename) file_instance.save(os.path.join(os.path.join(config.Config('upload', 'directory').value, 'uploads'), filename)) # scan job code, result = scan.Scan(filename).compress() return jsonify(code=code, result=result) else: return jsonify(code=1002, result="This extension can't support!")
def upload_file(): """ Scan by uploading compressed files :return: """ if 'file' not in request.files: return jsonify(code=1002, result="File can't empty!") file_instance = request.files['file'] if file_instance.filename == '': return jsonify(code=1002, result="File name can't empty!") if file_instance and common.allowed_file(file_instance.filename): filename = secure_filename(file_instance.filename) file_instance.save(os.path.join(os.path.join(config.Config('upload', 'directory').value, 'uploads'), filename)) # Start scan code, result = scan.Scan(filename).compress() return jsonify(code=code, result=result) else: return jsonify(code=1002, result="This extension can't support!")
def test_allowed_file(): assert common.allowed_file('test01/test02/file.rar') is True