Flask服务后删除文件

Question

我有一个Flask视图,它生成数据并将其保存为带有Pandas的CSV文件,然后显示数据.第二个视图用于生成的文件.我想在下载后删除该文件.我当前的代码引发了权限错误,可能是因为after_request在文件提供之前删除了该文件send_from_directory.如何在提供文件后删除文件？

def process_data(data)
    tempname = str(uuid4()) + '.csv'
    data['text'].to_csv('samo/static/temp/{}'.format(tempname))
    return file

@projects.route('/getcsv/<file>')
def getcsv(file):
    @after_this_request
    def cleanup(response):
        os.remove('samo/static/temp/' + file)
        return response

    return send_from_directory(directory=cwd + '/samo/static/temp/', filename=file, as_attachment=True)

Answer 1

after_request在视图返回之后但在发送响应之前运行.发送文件可以使用流式响应; 如果你在完全阅读之前删除它,你可能会遇到错误.

这主要是Windows上的一个问题,其他平台可以将文件标记为已删除并保留,直到无法访问.但是,只有在您确定已发送文件时才删除该文件仍然有用,无论平台如何.

将文件读入内存并提供服务,以便以后删除时不会被读取.如果文件太大而无法读入内存,请使用生成器为其提供服务然后将其删除.

@app.route('/download_and_remove/<filename>')
def download_and_remove(filename):
    path = os.path.join(current_app.instance_path, filename)

    def generate():
        with open(path) as f:
            yield from f

        os.remove(path)

    r = current_app.response_class(generate(), mimetype='text/csv')
    r.headers.set('Content-Disposition', 'attachment', filename='data.csv')
    return r