极客笔记Flask框架上传文件
1. 介绍
Flask是一个基于Python的轻量级Web应用框架，它简洁而灵活，适用于开发小型的Web应用程序或原型。在Flask框架中，文件上传是一个常见的需求，可以用于用户上传头像、文件分享或其他文件处理任务。本文将详细介绍如何使用Flask框架实现文件上传功能。
2. 准备工作
在开始编写文件上传功能之前，我们需要先安装Flask框架并创建一个新的Flask应用。你可以使用以下命令安装Flask框架：
pip install Flask
然后，创建一个新的Python文件，例如app.py，用于编写Flask应用。在app.py中导入Flask模块：
from flask import Flask
接下来创建一个Flask应用实例：
app = Flask(__name__)
这样我们就完成了Flask的基本设置。
3. 文件上传
要实现文件上传，我们首先需要在Flask应用中创建一个HTML表单，允许用户选择并上传文件。在HTML文件中，可以使用元素来实现文件选择功能。以下是一个简单的文件上传表单的示例：
<!DOCTYPE html>
<html>
<body>
<h2>文件上传</h2>
<form action="/upload" method="post" enctype="multipart/form-data">
<input type="file" name="file"><br><br>
<input type="submit" value="上传">
</form>
</body>
</html>
在上述HTML文件中，我们设置了一个表单，表单的action属性指向”/upload”，表示将表单数据提交到Flask应用的/upload路由上。enctype属性设置为”multipart/form-data”，表示表单中包含文件数据。文件选择框的name属性设置为”file”，这个name属性值将在后端代码中用于接收文件。
在Flask应用中，我们需要为/upload路由编写一个处理函数，用于接收并保存上传的文件。可以使用Flask提供的request对象来获取上传的文件数据。以下是一个简单的文件上传处理函数的示例代码：
from flask import request
@app.route('/upload', methods=['POST'])
def upload_file():
file = request.files['file']
file.save(file.filename)
return '文件上传成功！'
在上述代码中，我们首先使用request.files[‘file’]获取上传的文件对象，其中’file’对应HTML文件上传表单中文件选择框的name属性值。接着，使用file.save()方法保存文件到当前目录下，其中file.filename表示文件名。最后，返回一个字符串表示文件上传成功。
为了让Flask应用能够处理文件上传请求，我们还需要使用app.run()启动应用：
if __name__ == '__main__':
app.run()
启动Flask应用后，你可以在浏览器中访问http://localhost:5000/，将看到一个带有文件上传功能的网页。选择并上传一个文件后，上传的文件将被保存到当前目录下。
4. 文件上传限制
为了保护服务器和应用程序，我们常常需要对上传的文件进行限制。Flask框架提供了一些方法来限制上传文件的大小、类型和数量。
4.1 文件大小限制
默认情况下，Flask框架没有对上传文件大小进行限制。但你可以通过设置app.config中的MAX_CONTENT_LENGTH选项来限制上传文件的大小，单位为字节。以下是一个限制文件大小为4MB的示例代码：
app.config['MAX_CONTENT_LENGTH'] = 4 * 1024 * 1024 # 4MB
如果上传的文件大小超过了设置的最大限制，Flask框架将返回一个413错误，表示请求实体过大。
4.2 文件类型限制
除了限制文件大小，我们还可以限制上传文件的类型。Flask框架提供了一个upload_extensions选项，用于指定允许上传的文件扩展名。以下是一个限制只允许上传jpg、png和gif格式图片文件的示例代码：
from werkzeug.utils import secure_filename
app.config['UPLOAD_EXTENSIONS'] = ['.jpg', '.png', '.gif']
@app.route('/upload', methods=['POST'])
def upload_file():
file = request.files['file']
if file and file.filename.rsplit('.', 1)[1].lower() in app.config['UPLOAD_EXTENSIONS']:
filename = secure_filename(file.filename)
file.save(filename)
return '文件上传成功！'
else:
return '不允许上传该文件类型！'
在上述代码中，我们首先设置了UPLOAD_EXTENSIONS变量来定义可上传的文件扩展名。当接收到上传文件时，我们通过file.filename.rsplit(‘.’, 1)[1].lower()获取到上传文件的扩展名，并将其转换为小写。然后，通过判断上传的文件扩展名是否在允许的扩展名列表中，来判断是否允许上传该文件类型。
4.3 文件数量限制
一个常见的需求是限制用户一次性上传文件的数量。Flask框架没有提供直接的方法来限制文件数量，但我们可以通过一些检查和计数来实现。以下是一个限制用户最多上传3个文件的示例代码：
app.config['MAX_FILE_COUNT'] = 3
@app.route('/upload', methods=['POST'])
def upload_file():
files = request.files.getlist('file')
if len(files) <= app.config['MAX_FILE_COUNT']:
for file in files:
filename = secure_filename(file.filename)
file.save(filename)
return '文件上传成功！'
else:
return '最多只能上传{}个文件！'.format(app.config['MAX_FILE_COUNT'])
在上述代码中，我们首先设置了MAX_FILE_COUNT变量来定义最多允许上传的文件数量。当接收到上传文件时，我们通过request.files.getlist(‘file’)获取到上传的多个文件对象。然后，通过判断上传文件列表的长度是否小于等于允许的文件数量，来限制文件数量。
5. 文件保存路径
在前面的示例代码中，我们将上传的文件保存到了当前目录下。但实际应用中，通常需要将文件保存到指定的路径下，或者将文件命名为唯一的名称以避免覆盖。Flask框架通过app.config[‘UPLOAD_FOLDER’]选项来设置文件保存的路径。以下是一个将上传的文件保存到指定路径下的示例代码：
import os
app.config['UPLOAD_FOLDER'] = '/path/to/your/upload/folder'
@app.route('/upload', methods=['POST'])
def upload_file():
file = request.files['file']
if file:
filename = secure_filename(file.filename)
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
return '文件上传成功！'
else:
return '请选择上传文件！'
在上述代码中，我们首先设置了UPLOAD_FOLDER变量来指定文件保存的路径，你需要将其修改为你实际的