我正在尝试在 Sanic 上执行文件上传,但它无法正常工作,flask 的正常语法似乎不适用于 sanic。
我什至无法访问文件名或保存方法以将上传的文件保存到给定目录。
我正在尝试在 Sanic 上执行文件上传,但它无法正常工作,flask 的正常语法似乎不适用于 sanic。
我什至无法访问文件名或保存方法以将上传的文件保存到给定目录。
经过长时间的努力,我发现以下代码可以正常工作
@app.route("/upload", methods=['POST'])
async def omo(request):
from sanic import response
import os
import aiofiles
if not os.path.exists(appConfig["upload"]):
os.makedirs(appConfig["upload"])
async with aiofiles.open(appConfig["upload"]+"/"+request.files["file"][0].name, 'wb') as f:
await f.write(request.files["file"][0].body)
f.close()
return response.json(True)
上面的答案很棒。一些小的改进:
(1)由于我们使用的是Sanic,我们尝试异步做文件io:
async def write_file(path, body):
async with aiofiles.open(path, 'wb') as f:
await f.write(body)
f.close()
(2) 确保文件不要太大以免服务器崩溃:
def valid_file_size(file_body):
if len(file_body) < 10485760:
return True
return False
(3) 检查文件名和文件类型是否正确:
def valid_file_type(file_name, file_type):
file_name_type = file_name.split('.')[-1]
if file_name_type == "pdf" and file_type == "application/pdf":
return True
return False
(4) 确保文件名没有危险/不安全的字符。您可以在 werkzeug.utils 中使用 secure_filename 函数:http: //flask.pocoo.org/docs/0.12/patterns/fileuploads/
(5) 这段代码将所有内容结合在一起:
async def process_upload(request):
# Create upload folder if doesn't exist
if not os.path.exists(app.config.UPLOAD_DIR):
os.makedirs(app.config.UPLOAD_DIR)
# Ensure a file was sent
upload_file = request.files.get('file_names')
if not upload_file:
return redirect("/?error=no_file")
# Clean up the filename in case it creates security risks
filename = secure_filename(upload_file.name)
# Ensure the file is a valid type and size, and if so
# write the file to disk and redirect back to main
if not valid_file_type(upload_file.name, upload_file.type):
return redirect('/?error=invalid_file_type')
elif not valid_file_size(upload_file.body):
return redirect('/?error=invalid_file_size')
else:
file_path = f"{app.config.UPLOAD_DIR}/{str(datetime.now())}.pdf"
await write_file(file_path, upload_file.body)
return redirect('/?error=none')
我创建了一篇关于如何在 Sanic 中处理文件上传的博客文章。我添加了一些文件验证和异步文件写入。我希望其他人觉得这有帮助:
https://blog.fcast.co/2019/06/16/file-upload-handling-using-asynchronous-file-writing/
这是特定文件类型的文件上传示例(这个是针对pdf文件的)
from sanic import Sanic
from sanic.response import json
from pathlib import os
from datetime import datetime
app = Sanic()
config = {}
config["upload"] = "./tests/uploads"
@app.route("/upload", methods=['POST'])
def post_json(request):
if not os.path.exists(config["upload"]):
os.makedirs(config["upload"])
test_file = request.files.get('file')
file_parameters = {
'body': test_file.body,
'name': test_file.name,
'type': test_file.type,
}
if file_parameters['name'].split('.')[-1] == 'pdf':
file_path = f"{config['upload']}/{str(datetime.now())}.pdf"
with open(file_path, 'wb') as f:
f.write(file_parameters['body'])
f.close()
print('file wrote to disk')
return json({ "received": True, "file_names": request.files.keys(), "success": True })
else:
return json({ "received": False, "file_names": request.files.keys(), "success": False, "status": "invalid file uploaded" })
有关其他请求类型的示例,请参阅官方文档(https://sanic.readthedocs.io/en/latest/sanic/request_data.html)