Flask 文件上传下载

最新推荐文章于 2024-05-19 16:02:09 发布

ㅤ予

最新推荐文章于 2024-05-19 16:02:09 发布

阅读量2.5k

点赞数

分类专栏： Flask-learn 文章标签： flask python 后端

本文链接：https://blog.csdn.net/qq_44791003/article/details/126818927

版权

Flask-learn 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

Flask 文件上传

在 Flask 中处理文件上传非常简单。它需要一个 HTML 表单，其 enctype 属性设置为“multipart/form-data”，将文件发布到 URL。

URL 处理程序从 request.files[] 对象中提取文件，并将其保存到所需的位置。

每个上传的文件首先会保存在服务器上的临时位置，然后将其实际保存到它的最终位置。

目标文件的名称可以是硬编码的，也可以从 request.files[file] 对象的filename属性中获取。

但是，建议使用 secure_filename() 函数获取它的安全版本。

可以在 Flask 对象的配置设置中定义默认上传文件夹的路径和上传文件的最大大小。

app.config[‘UPLOAD_FOLDER’] 定义上传文件夹的路径

app.config[‘MAX_CONTENT_LENGTH’] 指定要上传的文件的最大大小（以字节为单位）

以下代码具有 '/upload' URL 规则，该规则在get请求下- templates 文件夹中显示 'upload.html'，在post请求下用于调用 upload() 函数处理上传过程，在'/download/<filename>'请求下可以通过文件名下载对应提交文件。

'upload.html' 有一个文件选择器按钮和一个提交按钮。（界面较为简单，有需求可自己美化界面）

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
    {# enctype设置不对字符编码 #}
    <form action="upload" method="POST" enctype="multipart/form-data">
        <input type="file" name="file" accept=".jpg, .png" />
        <input type="submit" />
    </form>
</body>
</html>

以下是 Flask 应用程序的 Python 代码。

from flask import Flask, render_template, request, send_from_directory
from werkzeug.utils import secure_filename
import os
import uuid

app = Flask(__name__)

# 设置文件上传保存路径
app.config['UPLOAD_FOLDER'] = 'static/upload/'
# MAX_CONTENT_LENGTH设置上传文件的大小，单位字节
app.config['MAX_CONTENT_LENGTH'] = 1 * 1024 * 1024

@app.route('/upload', methods=['GET', 'POST'])
def upload():
    # 如果是get请求响应上传视图，post请求响应上传文件
    if(request.method == 'GET'):
        return render_template('upload.html');
    else:
        f = request.files['file'];
        fname = secure_filename(f.filename);
        ext = fname.rsplit('.')[-1];
        # 生成一个uuid作为文件名
        fileName = str(uuid.uuid4()) + "." + ext;
        # os.path.join拼接地址，上传地址，f.filename获取文件名
        f.save(os.path.join(app.config['UPLOAD_FOLDER'], fileName))
        return 'ok';

# 图片下载
@app.route('/download/<filename>', methods=['GET'])
def download(filename):
    if request.method == "GET":
        #通过文件名下载文件
        path = os.path.isfile(os.path.join(app.config['UPLOAD_FOLDER'], filename));
        if path:
            return send_from_directory(app.config['UPLOAD_FOLDER'], filename, as_attachment=True)

if __name__ == '__main__':
    app.run(debug=True)