Flask接收请求，将指定字符串作为文件返回，浏览器端实现下载功能 Flask文件下载功能

2022年6月21日 111次阅读 来源: 匆匆流年。 
最近项目中有这样的需求，用户上传一个xml的文件，服务端对它进行处理，返回一个TXT文件
需要将用户上传的文件保存后，再进行处理，生成TXT文件，返回给用户，
但这样会产生文件冗余的情况，用户上传的文件可以即时删除，但是返回的这个文件什么时候删除呢？
于是想到如果返回一个文件流呢，那样就不会生成新的文件了。
参阅flask官方文档 http://www.pythondoc.com/flask/patterns/streaming.html
做了以下修改，成功实现功能
@login_required
def cast_xml_file():
    form = XMLFileForm()

    if form.validate_on_submit():
        cur_file = form.data['xml_file']

        # 获取原文件名后缀
        suffix = os.path.splitext(cur_file.filename)

        if len(suffix) != 2 or suffix[1][1:] != 'xml':
            flash('文件名不合法！')
        else:
            rand_str = ''.join(random.sample(string.ascii_letters + string.digits, 12))
            new_file_name = "xml-{rand_str}{suffix}".format(rand_str=rand_str, suffix=suffix[1])
            text_file.save(cur_file, name=new_file_name)
            # 文件的保存路径
            save_path = text_file.path(new_file_name)
            
            # do something
            # result_str =['1','2']

            def generate():
                for row in result_str:
                    yield row + '\n'

            response = make_response(Response(generate(), mimetype='text/xml'))
            response.headers["Content-Disposition"] = "attachment; filename=download.txt;")
            return response

    return render_template('home/home_cast_xml.html', form=form)

 

    原文作者：匆匆流年。
    原文地址: https://blog.csdn.net/m0_37932636/article/details/105430275
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞