烧瓶:`@after_this_request` 不工作

alb*_*ert 2 python request delete-file flask python-3.x

我想在用户下载由 Flask 应用程序创建的文件后删除一个文件。

为此,我在 SO 上找到了这个答案,它没有按预期工作,并引发了一个after_this_request未定义的错误。

因此,我更深入地研究了Flask 的文档,其中提供了有关如何使用该方法的示例片段。因此,我通过定义after_this_request示例代码段中所示的函数来扩展我的代码。

执行代码resp。运行服务器按预期工作。但是,该文件没有被删除,因为@after_this_request没有被调用,这很明显,因为After request ...它没有打印到终端中 Flask 的输出:

#!/usr/bin/env python3
# coding: utf-8


import os
from operator import itemgetter
from flask import Flask, request, redirect, url_for, send_from_directory, g
from werkzeug.utils import secure_filename

UPLOAD_FOLDER = '.'
ALLOWED_EXTENSIONS = set(['csv', 'xlsx', 'xls'])

app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER


def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS


def after_this_request(func):
    if not hasattr(g, 'call_after_request'):
        g.call_after_request = []
    g.call_after_request.append(func)
    return func


@app.route('/', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        if 'file' not in request.files:
            flash('No file part')
            return redirect(request.url)
        file = request.files['file']
        if file.filename == '':
            flash('No selected file')
            return redirect(request.url)
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
            file.save(filepath)

            @after_this_request
            def remove_file(response):
                print('After request ...')
                os.remove(filepath)
                return response

            return send_from_directory('.', filename=filepath, as_attachment=True)

    return '''
    <!doctype html>
    <title>Upload a file</title>
    <h1>Uplaod new file</h1>
    <form action="" method=post enctype=multipart/form-data>
      <p><input type=file name=file>
         <input type=submit value=Upload>
    </form>
    '''


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=8080, debug=True)
Run Code Online (Sandbox Code Playgroud)

我在这里想念什么?如何确保调用@after_this_request装饰器后面的函数以在用户下载文件后删除文件?

注意:使用 Flask 版本 0.11.1

Mar*_*ers 6

确保从 导入装饰器flask.after_this_request。该装饰器是 Flask 0.9 中的新增功能。

如果您使用的是 Flask 0.8 或更早版本,那么此请求功能之后就没有具体的了。每个请求只有一个钩子,这是代码片段用来处理每个请求回调的方法。

因此,除非您使用 Flask 0.9 或更高版本,否则您需要自己实现记录的钩子:

@app.after_request
def per_request_callbacks(response):
    for func in getattr(g, 'call_after_request', ()):
        response = func(response)
    return response
Run Code Online (Sandbox Code Playgroud)

因此,该钩子在每个请求之后运行,并查找要调用的钩子列表g.call_after_request。装饰after_this_request器在那里注册一个函数。


use*_*654 5

只需要从flask中导入after_this_request,不需要修改after_request或者创建hook。

from flask import after_this_request

@after_this_request
def remove_file(response):
    print('After request ...')
    os.remove(filepath)
    return response
Run Code Online (Sandbox Code Playgroud)