Flask 文件和流
当我们要往客户端发送大量的数据比较好的方式是使用流,通过流的方式来将响应内容发送给客户端,实现文件的上传功能，以及如何获取上传后的文件。
响应流的生成
Flask响应流的实现原理就是通过Python的生成器，也就是大家所熟知的yield的表达式，将yield的内容直接发送到客户端。下面就是一个简单的实现：
from flask import Flask, Response app = Flask(__name__) @app.route('/large.csv') def generate_large_csv(): def generate(): for row in range(50000): line = [] for col in range(500): line.append(str(col)) if row % 1000 == 0: print 'row: %d' % row yield ','.join(line) + '\n' return Response(generate(), mimetype='text/csv')
这段代码会生成一个5万行100M的csv文件，每一行会通过yield表达式分别发送给客户端。
运行时你会发现文件行的生成与浏览器文件的下载是同时进行的，而不是文件全部生成完毕后再开始下载
这里我们用到了响应类”flask.Response”，它的初始化方法第一个参数就是我们定义的生成器函数，第二个参数指定了响应类型
我们将上述方法应用到模板中，如果模板的内容很大，怎么采用流的方式呢？这里我们要自己写个流式渲染模板的方法。
# 流式渲染模板 def stream_template(template_name, **context): # 将app中的请求上下文内容更新至传入的上下文对象context， # 这样确保请求上下文会传入即将被渲染的模板中 app.update_template_context(context) # 获取Jinja2的模板对象 template = app.jinja_env.get_template(template_name) # 获取流式渲染模板的生成器 generator = template.stream(context) # 启用缓存，这样不会每一条都发送，而是缓存满了再发送 generator.enable_buffering(5) return generator
这段代码的核心，就是通过”app.jinja_env”来访问Jinja2的Environment对象，然后调用Environment对象的”get_template()”方法来获得模板对象，再调用模板对象的”stream()”方法生成一个”StreamTemplate”的对象。这个对象实现了”__next__()”方法，可以作为一个生成器使用，如果你看了Jinja2的源码，你会发现模板对象的”stream()”方法的实现就是使用了yield表达式，所以原理同上例一样。另外，我们启用了缓存”enable_buffering()”来避免客户端发送过于频繁，其参数的默认值就是5。
现在我们就可以在视图方法中，采用”stream_template()”，而不是以前介绍的”render_template()”来渲染模板了：
@app.route('/stream.html') def render_large_template(): file = open('server.log') return Response(stream_template('stream-view.html',logs=file.readlines()))
上例的代码会将本地的”server.log”日志文件内容传入模板，并以流的方式渲染在页面上。
”stream_with_context()”方法，它允许生成器在运行期间获取请求上下文：
from flask import request, stream_with_context @app.route('/method') def streamed_response(): def generate(): yield 'Request method is: ' yield request.method yield '.' return Response(stream_with_context(generate()))
因为我们初始化Response对象时调用了”stream_with_context()”方法，所以才能在yield表达式中访问request对象。
文件上传
我们分下面4个步骤来实现文件上传功能：
1、首先建立一个让用户上传文件的页面，我们将其放在模板”upload.html”中
<!DOCTYPE html> <title>Upload File</title> <h1>Upload new File</h1> <form action="" method="post" enctype="multipart/form-data"> <p><input type="file" name="file"> <input type="submit" value="Upload">
</p> </form>
这里主要就是一个enctype=”multipart/form-data”的form表单；一个类型为file的input框，即文件选择框；还有一个提交按钮。
2、定义一个文件合法性检查函数
# 设置允许上传的文件类型 ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg']) # 检查文件类型是否合法 def allowed_file(filename): # 判断文件的扩展名是否在配置项ALLOWED_EXTENSIONS中 return '.' in filename and \ filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
3、文件提交后，在POST请求的视图函数中，通过request.files获取文件对象
这个request.files是一个字典，字典的键值就是之前模板中文件选择框的”name”属性的值，上例中是”file”；键值所对应的内容就是上传过来的文件对象。
4、检查文件对象的合法性后，通过文件对象的save()方法将文件保存在本地
我们将第3和第4步都放在视图函数中，代码如下：
import os from flask import flask, render_template from werkzeug import secure_filename app = Flask(__name__) # 设置请求内容的大小限制，即限制了上传文件的大小 app.config['MAX_CONTENT_LENGTH'] = 5 * 1024 * 1024 # 设置上传文件存放的目录 UPLOAD_FOLDER = './uploads' @app.route('/upload', methods=['GET', 'POST']) def upload_file(): if request.method == 'POST': # 获取上传过来的文件对象 file = request.files['file'] # 检查文件对象是否存在，且文件名合法 if file and allowed_file(file.filename): # 去除文件名中不合法的内容 filename = secure_filename(file.filename) # 将文件保存在本地UPLOAD_FOLDER目录下 file.save(os.path.join(UPLOAD_FOLDER, filename)) return 'Upload Successfully' else: # 文件不合法 return 'Upload Failed' else: # GET方法 return render_template('upload.html')
- Flask的MAX_CONTENT_LENGTH配置项可以限制请求内容的大小，默认是没有限制，上例中我们设为5M。
- 必须调用”werkzeug.secure_filename()”来使文件名安全
- Flask处理文件上传的方式：如果上传的文件很小，那么会把它直接存在内存中。否则就会把它保存到一个临时目录下，通过”tempfile.gettempdir()”方法可以获取这个临时目录的位置。
一个简便的方法来让用户获取已上传的文件：
from flask import send_from_directory @app.route('/uploads/<filename>') def uploaded_file(filename): return send_from_directory(UPLOAD_FOLDER, filename)
这个帮助方法”send_from_directory()”可以安全地将文件发送给客户端，它还可以接受一个参数”mimetype”来指定文件类型，和参数”as_attachment=True”来添加响应头”Content-Disposition: attachment”
