使用Flask处理大文件上传

本文介绍了使用Flask处理大文件上传的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！

问题描述

使用Flask处理超大文件上传(大于1 GB)的最佳方法是什么?

What would be the best way to handle very large file uploads (1 GB +) with Flask?

我的应用程序本质上采用多个文件，为它们分配一个唯一的文件号，然后根据用户选择的位置将其保存在服务器上.

My application essentially takes multiple files assigns them one unique file number and then saves it on the server depending on where the user selected.

我们如何将文件上传作为后台任务运行，以使用户在1小时内没有浏览器旋转，而是可以立即进入下一页?

Flask开发服务器能够处理大量文件(50gb耗时1.5个小时，上传速度很快，但将文件写入空白文件的速度却很慢)
如果我用Twisted包装应用程序，则该应用程序在大文件上崩溃
我曾尝试将Celery与Redis一起使用，但这似乎不是发布上传的选项
我在Windows上，网络服务器的选项更少

推荐答案

我认为解决该问题的超级简单方法就是将文件分批发送.因此，要完成这项工作将需要两个部分，即前端(网站)和后端(服务器).对于前端部分，您可以使用Dropzone.js之类的东西，它没有附加的依赖关系，还包括不错的CSS.您所要做的就是将dropzone类添加到表单中，它会自动将其变成其特殊的拖放字段之一(您也可以单击并选择).

I think the super simple way to get around that simply sends the file in lots of small parts/chunks. So there are going to be two parts to making this work, the front-end (website) and backend (server).For the front-end part, you can use something like Dropzone.js which has no additional dependencies and decent CSS included. All you have to do is add the class dropzone to a form and it automatically turns it into one of their special drag and drop fields (you can also click and select).

但是，默认情况下，dropzone不会对文件进行分块.幸运的是，它确实很容易启用.这是启用了DropzoneJS和chunking的示例文件上传表单:

However, by default, dropzone does not chunk files. Luckily, it is really easy to enable.Here's a sample file upload form with DropzoneJS and chunking enabled:

<html lang="en">
<head>

    <meta charset="UTF-8">

    <link rel="stylesheet"
     href="https://cdnjs.cloudflare.com/ajax/libs/dropzone/5.4.0/min/dropzone.min.css"/>

    <link rel="stylesheet"
     href="https://cdnjs.cloudflare.com/ajax/libs/dropzone/5.4.0/min/basic.min.css"/>

    <script type="application/javascript"
     src="https://cdnjs.cloudflare.com/ajax/libs/dropzone/5.4.0/min/dropzone.min.js">
    </script>

    <title>File Dropper</title>
</head>
<body>

<form method="POST" action='/upload' class="dropzone dz-clickable"
      id="dropper" enctype="multipart/form-data">
</form>

<script type="application/javascript">
    Dropzone.options.dropper = {
        paramName: 'file',
        chunking: true,
        forceChunking: true,
        url: '/upload',
        maxFilesize: 1025, // megabytes
        chunkSize: 1000000 // bytes
    }
</script>
</body>
</html>

这是使用flask的后端部分:

And Here's the Back-end part using flask:

import logging
import os

from flask import render_template, Blueprint, request, make_response
from werkzeug.utils import secure_filename

from pydrop.config import config

blueprint = Blueprint('templated', __name__, template_folder='templates')

log = logging.getLogger('pydrop')


@blueprint.route('/')
@blueprint.route('/index')
def index():
    # Route to serve the upload form
    return render_template('index.html',
                           page_name='Main',
                           project_name="pydrop")


@blueprint.route('/upload', methods=['POST'])
def upload():
    file = request.files['file']

    save_path = os.path.join(config.data_dir, secure_filename(file.filename))
    current_chunk = int(request.form['dzchunkindex'])

    # If the file already exists it's ok if we are appending to it,
    # but not if it's new file that would overwrite the existing one
    if os.path.exists(save_path) and current_chunk == 0:
        # 400 and 500s will tell dropzone that an error occurred and show an error
        return make_response(('File already exists', 400))

    try:
        with open(save_path, 'ab') as f:
            f.seek(int(request.form['dzchunkbyteoffset']))
            f.write(file.stream.read())
    except OSError:
        # log.exception will include the traceback so we can see what's wrong
        log.exception('Could not write to file')
        return make_response(("Not sure why,"
                              " but we couldn't write the file to disk", 500))

    total_chunks = int(request.form['dztotalchunkcount'])

    if current_chunk + 1 == total_chunks:
        # This was the last chunk, the file should be complete and the size we expect
        if os.path.getsize(save_path) != int(request.form['dztotalfilesize']):
            log.error(f"File {file.filename} was completed, "
                      f"but has a size mismatch."
                      f"Was {os.path.getsize(save_path)} but we"
                      f" expected {request.form['dztotalfilesize']} ")
            return make_response(('Size mismatch', 500))
        else:
            log.info(f'File {file.filename} has been uploaded successfully')
    else:
        log.debug(f'Chunk {current_chunk + 1} of {total_chunks} '
                  f'for file {file.filename} complete')

    return make_response(("Chunk upload successful", 200))

这篇关于使用Flask处理大文件上传的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！