Flask 文件上传与下载

最新推荐文章于 2024-08-11 16:13:55 发布

小马同学

最新推荐文章于 2024-08-11 16:13:55 发布

阅读量6.8k

点赞数 2

分类专栏： python 文章标签： python web flask

本文链接：https://blog.csdn.net/qq_40205116/article/details/110247395

版权

python 专栏收录该内容

14 篇文章 9 订阅

订阅专栏

文件上传与下载在平时开发中很常见，所以也是很重要的一部分。下面来一起看看flask中的文件上传与下载。

上传文件

定义一个视图函数，用于返回上传页面和保存上传文件。

@app.route('/upload', methods=['GET', 'POST'])
def upload():
    if(request.method == 'GET'):
        return render_template('upload.html');
    else:
    	# file为上传表单的name属性值
        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';

uuid.uuid4()生成一个uuid，可以理解为生成一个不会重复的字符串，当做文件名（避免文件名重复）。

上传页面，这里上传页面比较简陋，可以自行更改。

<!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>

上传效果。
在这里插入图片描述

上传的路径设置的为项目目录下的static/upload下，上传成功后，可以看到upload文件夹下多了一张图片，名字为生成的uuid。

下载

定义一个视图函数，用于根据文件名下载文件。

# 图片下载
@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)

输入http://localhost:5000/download/e19ddc98-d317-4f11-aad6-3e73677fe066.jpg地址即可下载对应的文件。
在这里插入图片描述
【注】一般开发中会把上传文件的信息，如文件名、路径等信息保存在数据库中。可以在上传文件视图函数中添加保存文件信息方法。

上传文件的完整代码：

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():
    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()

除上述方法外，还可以在视图函数中把接收到的图片转为base64格式，存储在数据库中，这样就不用写入到本地了。