首先,您需要安装 Flask 和 Flask-Uploads。使用 pip 非常容易,只需运行以下命令:
pip install Flask Flask-Uploads
然后,导入必要的库并设置应用程序:
from flask import Flask, render_template, request, redirect, url_for
from flask_uploads import UploadSet, configure_uploads, patch_request_class
app = Flask(__name__)
# 配置文件上传
app.config['UPLOADS_DEFAULT_DEST'] = 'static/uploads'
app.config['UPLOADS_DEFAULT_URL'] = '/static/uploads'
app.config['UPLOADSet'] = ('my_uploads', 'uploads', 'all')
# 初始化上传套件
my_uploads = UploadSet(app)
configure_uploads(app, my_uploads)
patch_request_class(app)
现在,您需要定义一个简单的路由,以便用户可以上传文件:
@app.route('/upload', methods=['GET', 'POST'])
def upload():
if request.method == 'POST':
# 从表单中获取文件
file = request.files['file']
# 将文件保存到上传目录中
file.save(my_uploads.join('uploads', file.filename))
# 重定向到成功页面
return redirect(url_for('uploaded'))
# 如果是 GET 请求,则显示上传表单
return render_template('upload.html')
但是,在此之前,您需要创建另一个路由来显示已上传的文件:
@app.route('/uploads/' )
def uploaded(filename):
return render_template('uploaded.html', filename=filename)
接下来,您需要创建上传表单的 HTML 模板。在 Flask 中,通常将模板保存在 templates 目录中。在这个例子中,我们将文件上传表单模板保存在 upload.html 中,并将已上传文件的模板保存在 uploaded.html 中。
这是 upload.html 的内容:
DOCTYPE html>
<html>
<head>
<title>文件上传title>
head>
<body>
<h1>上传文件h1>
<form method="POST" enctype="multipart/form-data">
<input type="file" name="file">
<input type="submit" value="上传">
form>
body>
html>
这是 uploaded.html 的内容:
DOCTYPE html>
<html>
<head>
<title>已上传文件title>
head>
<body>
<h1>已上传文件h1>
<p>您已上传的文件是:{{ filename }}p>
body>
html>
现在,您的 Flask-Uploads 应用程序已经准备就绪,可以开始上传和下载文件了!
以上是使用 Flask-Uploads 处理文件上传和下载的基本步骤和代码示例。下面我们再深入一步,介绍一些扩展功能和用法。
自定义上传目录
默认情况下,上传的文件会被保存在应用程序的静态目录中。如果您需要将文件保存在其他目录中,可以设置 UploadSet 的自定义目录:
my_uploads = UploadSet(app, 'custom_dir')
这将上传文件保存到名为 “custom_dir” 的目录中。
自定义文件名
默认情况下,上传的文件会保留原始文件名。如果您需要为文件指定一个新的文件名,可以使用 Flask-Uploads 的 save() 方法,并传递一个新的文件名作为参数:
file.save('custom_filename.txt')
这将上传文件保存到上传目录中,并使用 “custom_filename.txt” 作为文件名。
限制文件大小
如果您需要限制上传文件的大小,可以在表单中添加一个 input 元素,并设置 name 属性为 MAX_FILE_SIZE,例如:
<input type="hidden" name="MAX_FILE_SIZE" value="10485760">
这将限制用户上传的文件大小不能超过 10MB。
上传多个文件
如果您需要允许用户上传多个文件,可以在表单中添加多个 input 元素,并设置 name 属性为 file,例如:
<input type="file" name="file">
<input type="file" name="file">
<input type="file" name="file">
然后,在 Flask 路由中,您可以使用 request.files 获取所有上传的文件,并使用循环遍历它们:
for file in request.files.getlist('file'):
# 处理每个文件
使用 Flask-Uploads 蓝图
如果您使用 Flask-Uploads 的蓝图,可以将上传文件保存到不同的目录中,以便更好地组织应用程序的文件。例如,您可以在蓝图目录中创建一个名为 “uploads” 的目录,然后在应用程序配置中设置 UPLOADSet 为 (‘my_uploads’, ‘uploads’, ‘all’)。这将将上传文件保存到蓝图目录中的 “uploads” 目录中。
自定义 URL
如果您需要为上传的文件生成自定义的 URL,可以在应用程序配置中设置 UPLOADSet 的 url 参数:
app.config['UPLOADS_DEFAULT_URL'] = 'http://example.com/uploads/%(path)s'
这将为上传的文件生成自定义的 URL,例如 http://example.com/uploads/myfile.txt。
自定义文件名转换器
如果您需要自定义文件名转换器,可以继承 FileConverter 类,并实现自己的转换器方法。例如,以下代码将文件名中的空格替换为下划线:
from flask_uploads import FileConverter
class MyFileConverter(FileConverter):
@staticmethod
def rename(filename):
return filename.replace(' ', '_')
然后,在应用程序配置中注册自定义转换器:
app.config['UPLOADS_DEFAULT_FILENAME_Converter'] = MyFileConverter
这将使用自定义的转换器重命名上传的文件。
以上是使用 Flask-Uploads 处理文件上传和下载的一些扩展功能和用法。根据您的具体需求,您可以根据自己的情况进一步定制和扩展 Flask-Uploads 应用程序。