Flask 文件上传与下载

曾经终败给现在 2022-12-24 10:59 268阅读 0赞

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

### 相关配置 ###

# 设置文件上传保存路径
    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:
        	# 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>

上传效果。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2_size_16_color_FFFFFF_t_70 1]  
上传的路径设置的为项目目录下的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地址即可下载对应的文件。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2_size_16_color_FFFFFF_t_70 2]  
**【注】一般开发中会把上传文件的信息，如文件名、路径等信息保存在数据库中。可以在上传文件视图函数中添加保存文件信息方法。**

上传文件的完整代码：

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格式，存储在数据库中，这样就不用写入到本地了。**

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2_size_16_color_FFFFFF_t_70]: /images/20221120/fa2a51ea7c7a4d40b575c8dd6a1e5b2c.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20201127213505246.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20201127213940744.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjA1MTE2,size_16,color_FFFFFF,t_70