9

我正在尝试在 Sanic 上执行文件上传,但它无法正常工作,flask 的正常语法似乎不适用于 sanic。

我什至无法访问文件名或保存方法以将上传的文件保存到给定目录。

4

3 回答 3

12

经过长时间的努力,我发现以下代码可以正常工作

@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)
于 2018-02-22T14:51:43.860 回答
8

上面的答案很棒。一些小的改进:

(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/

于 2019-06-18T20:46:31.060 回答
1

这是特定文件类型的文件上传示例(这个是针对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

于 2019-04-24T10:35:35.547 回答