flask中request对象以及文件上传详解

最新推荐文章于 2022-12-02 13:02:45 发布

FatPuffer

最新推荐文章于 2022-12-02 13:02:45 发布

阅读量5.6k

点赞数 6

分类专栏： flask 文章标签： request对象文件上传详解

本文链接：https://blog.csdn.net/qq_42517220/article/details/88710861

版权

flask 专栏收录该内容

22 篇文章 2 订阅

订阅专栏

request获取请求参数：

from flask import request

与Django不同的是，flask是不需要将request对象作为第一个参数传入视图函数，他的request对象是来自于flask，是一个请求上下文对象（全局变量 --> 线程局部变量，使用起来就像线程全局变量一样，具有较高的隔离性），我们只需导入即可使用，request对象中保存了一次HTTP请求的一切信息。

request原型猜想

request = {
	"线程A"：{
		form:
		args:
		data:
	},
	"线程B"：{
		form:
		args:
		data:
	},
	...
}

属性	说明（类型）
data	记录请求的数据，并转换为字符串（*）
form	记录请求中的表单数据（MultiDict）
args	记录请求（url）中的查询参数（MultiDict）
cookies	记录请求中的cookie信息（Dict）
headers	记录请求中的报文头（EnvironHeaders）
method	记录请求使用的HTTP方法（GET/POST）
url	记录请求的url地址（string）例如：http://127.0.0.1:5000/index
path	记录请求路径（string）例如：/index
files	记录请求上传的文件（*）

表单格式的数据，一个类字典的对象，通过request.form.get()方法只能拿到多个同名参数的第一个，要想拿到所有同名的参数，必须使用request.form.getlist()来获取。

from flask import Flask, request

app = Flask(__name__)

@app.route('/index', methods=['POST'])
def index():
    # 从request中获取表单数据，一个类字典的对象
    name = request.form.get('name')
    age = request.form.get('age')
    gender = request.form.get('gender')
    return 'hello name=%s, age=%s, gender=%s' % (name, age)

上传文件示例：

from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
	if request.method == 'POST':
		f_obj = request.files['the_file']
		if f_obj is None:
			return '文件未上传'
		# 将文件保存至本地
		# 1.创建一个文件
		f = open('./demo.jpg', 'wb')
		# 2.读取用户上传文件内容，写入本地创建的文件
		data = f_obj.read()
		f.write(data)
		f.close()
		return '上传成功'

flask内部已经将保存文件封装了一个save方法，我们直接调用即可。

from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
	if request.method == 'POST':
		f_obj = request.files['the_file']
		if f_obj is None:
			return '文件未上传'
		f_obj.save('./demo.jpg')
		return '上传成功'

文件类型的数据，可以是任何格式，如果你想知道上传前文件在客户端的文件名是什么，你可以访问filename属性，但永远不要信任这个值，这个值是可以伪造的，如果你要把文件按客户端提供的文件名上传至服务器，那么请把它传递给Werkzeug提供的secure_filename()函数。

from flask import request
from werkzeug.utils import secure_filename

@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        f = request.files['the_file']
        if f is None:
			return '文件未上传'
        f.save('/var/www/uploads/' + secure_filename(f.filename))
    ...

我们从一个基本的应用开始，这个应用上传文件到一个指定目录，并把文件显示给用户

import os
from flask import Flask, flash, request, redirect, url_for
from werkzeug.utils import secure_filename
from flask import send_from_directory


# 上传文件要储存的目录
UPLOAD_FOLDER = '/path/to/the/uploads'
# 允许上传的文件扩展名的集合，确保用户不能上传html格式文件，防止xss攻击
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])

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


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

@app.route('/uploads/<filename>')
def uploaded_file(filename):
	# send_from_directory:使用send_file函数，将指定上传目录中的文件发送到客户端
    return send_from_directory(app.config['UPLOAD_FOLDER'],
                               filename)

@app.route('/', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        # 检查POST请求是否包含文件部分
        if 'file' not in request.files:
        	# get_flashed_messages()来获取后台堆积的flash消息
            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)
            # 将用户上传文件保存到我们定义好的路径下
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
            # 将用户上传文件返回
            return redirect(url_for('uploaded_file',
                                    filename=filename))
                    
	return '''
    <!doctype html>
    <title>Upload new File</title>
    <h1>Upload new File</h1>
    <form method=post enctype=multipart/form-data>
      <input type=file name=file>
      <input type=submit value=Upload>
    </form>
    '''

`werkzeug.secure_filename()` 说明：

那么 secure_filename() 函数到底是有什么用？有一条原则是“永远不要信任用户输入”。这条原则同样适用于已上传文件的文件名。所有提交的表单数据可能是伪造的，文件名也可以是危险的。此时要谨记：在把文件保存到文件系统之前总是要使用这个函数对文件名进行安检。

你可以会好奇 secure_filename() 做了哪些工作，如果不使用它会有什么后果。假设有人把下面的信息作为 filename 传递给你的应用:

filename = "../../../../home/username/.bashrc"

假设 ../ 的个数是正确的，你会把它和 UPLOAD_FOLDER 结合在一起，那么用户就可能有能力修改一个服务器上的文件，这个文件本来是用户无权修改的。这需要了解应用是如何运行的，但是请相信我，黑客都是很变态的 ?

现在来看看函数是如何工作的：

>>> secure_filename('../../../../home/username/.bashrc')
'home_username_.bashrc'

改进上传

Flask 到底是如何处理文件上传的呢？如果上传的文件很小，那么会把它们储存在内存中。否则就会把它们保存到一个临时的位置（通过 tempfile.gettempdir() 可以得到这个位置）。但是，如何限制上传文件的尺寸呢？缺省情况下， Flask 是不限制上传文件的尺寸的。可以通过设置配置的 MAX_CONTENT_LENGTH 来限制文件尺寸:

from flask import Flask, Request

app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024

上面的代码会把尺寸限制为 16 M 。如果上传了大于这个尺寸的文件， Flask 会抛出一个 RequestEntityTooLarge 异常，我们可以进行捕获。

FatPuffer

关注

6
点赞
踩
29

收藏

觉得还不错? 一键收藏
0
评论
flask中request对象以及文件上传详解

request获取请求参数：from flask import request与Django不同的是，flask是不需要将request对象作为第一个参数传入视图函数，他的request对象是来自于flask，是一个全局对象，我们只需导入即可使用，request对象中保存了一次HTTP请求的一切信息。属性说明（类型）data记录请求的数据，并转换为字符串（*）...
复制链接

扫一扫