烧瓶：`@after_this_request` 不工作

Question

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

为此，我发现这个 answer on SO 没有按预期工作，并引发了一个错误，提示 after_this_request 未定义。

因此，我更深入地研究了 Flask's documentation providing a sample snippet 如何使用该方法。因此，我通过定义示例代码段中所示的 after_this_request 函数来扩展我的代码。

分别执行代码。运行服务器按预期运行。但是，文件没有被删除，因为 @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)

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

注意：使用 Flask 版本 0.11.1

Answer 1

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

如果您使用的是 Flask 0.8 或更早版本，则在该请求后没有特定的功能。只有一个 after every request 挂钩，这是代码段用来处理每个请求回调的内容。

因此，除非您使用的是 Flask 0.9 或更新版本，否则您需要自己实施记录的钩子：

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

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

Answer 2

只需从 flask 中导入 after_this_request，无需修改 after_request 或创建挂钩。

from flask import after_this_request

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

烧瓶：`@after_this_request` 不工作

flask: `@after_this_request` not working

python

request

delete-file

flask

python-3.x