Flask多图上传不覆盖旧图技巧

小伙伴们有没有觉得学习文章很有意思？有意思就对了！今天就给大家带来《Flask 多图上传不覆盖旧图实现方法》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

本文档旨在解决 Flask 应用中上传多个图片时，新图片替换旧图片的问题。通过结合 Flask 的 session 功能，我们将展示如何修改 Python 代码和 HTML模板，实现同时显示多个用户上传的图片，从而构建更完善的图像编辑应用。

问题分析

在原代码中，上传图片后，文件名存储在全局变量中，并在 HTML 模板中使用该变量来显示图片。每次上传新图片，全局变量都会被更新，导致页面上只显示最后上传的图片。

解决方案：使用 Flask Session

Flask Session 允许我们在服务器端存储用户特定的数据，并在后续的请求中访问这些数据。 我们可以使用 Session 来存储每个上传图片的文件名，并在 HTML 模板中分别显示它们。

代码实现

1. 修改 Python 代码 (app.py)

首先，确保你的 Flask 应用已经安装了 Flask-Session。如果没有，可以使用 pip install Flask-Session 安装。

然后，修改你的 Flask 应用代码，如下所示：

from flask import Flask, render_template, request, redirect, url_for, flash, session
from werkzeug.utils import secure_filename
import os
import cv2

app = Flask(__name__)
app.secret_key = "your_secret_key" # 强烈建议修改为你自己的密钥
app.config['UPLOAD_FOLDER'] = 'static/uploads'
app.config['SESSION_TYPE'] = 'filesystem'  # 使用文件系统存储session
#from flask_session import Session #如果使用这种方式，需要去掉下面的注释，并把app.config注释掉
#Session(app) #如果使用这种方式，需要去掉上面的注释，并把app.config注释掉

ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}

def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/', methods=['POST'])
def upload_content():
    if not any(f in ['content_file', 'style_file'] for f in request.files):
        flash('No file part')
        return redirect(request.url)

    submitted_file = 'content_file' if 'content_file' in request.files else 'style_file'
    file = request.files[submitted_file]

    if file.filename == '':
        flash('No image selected for uploading')
        return redirect(request.url)

    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
        session[submitted_file] = filename
        print('upload_image filename: ' + filename, os.path.join(app.config['UPLOAD_FOLDER'], filename))
        flash('Image successfully uploaded and displayed below')
        return render_template('index.html')
    else:
        flash('Allowed image types are - png, jpg, jpeg, gif')
        return redirect(request.url)


@app.route('/display/')
def display_image(filename):
    return redirect(url_for('static', filename='uploads/' + filename), code=301)

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

代码解释：

• app.secret_key: 设置 Flask 应用的密钥。 务必替换为你自己的安全密钥。
• app.config['SESSION_TYPE'] = 'filesystem': 配置 Session 的存储类型为文件系统。你也可以选择其他存储类型，如 Redis 或数据库。
• session[submitted_file] = filename: 将上传的文件名存储到 Session 中。 submitted_file 变量用于区分是内容图片还是样式图片。
• if not any(f in ['content_file', 'style_file'] for f in request.files):: 检查是否上传了任何文件。
• submitted_file = 'content_file' if 'content_file' in request.files else 'style_file': 确定是哪个文件上传表单提交了文件。

2. 修改 HTML 模板 (index.html)

修改 HTML 模板，从 Session 中获取文件名并显示图片：

    


    

        

            
Select a content image to upload
            

                {% with messages = get_flashed_messages() %}
                  {% if messages %}
                    

                    {% for message in messages %}
                      
{{ message }}
                    {% endfor %}
                    
                  {% endif %}
                {% endwith %}
            

            

                {% if session['content_file'] %}
                    
                {% endif %}
            
            

                

                    

                        <input type="file" name="content_file" class="form-control" autocomplete="off" required>
                    
                
                

                    <input type="submit" value="Submit" class="btn btn-info">
                
            
        

        

            
Select a style image to upload
            

                {% with messages = get_flashed_messages() %}
                  {% if messages %}
                    

                    {% for message in messages %}
                      
{{ message }}
                    {% endfor %}
                    
                  {% endif %}
                {% endwith %}
            

            

                {% if session['style_file'] %}
                    
                {% endif %}
            
            

                

                    

                        <input type="file" name="style_file" class="form-control" autocomplete="off" required>
                    
                
                

                    <input type="submit" value="Submit" class="btn btn-info">
                
            
        
    

代码解释：

• name="content_file" 和 name="style_file": 修改了上传表单中文件输入框的 name 属性，以便在 Python 代码中区分上传的文件类型。
• {% if session['content_file'] %} 和 {% if session['style_file'] %}: 只有当 Session 中存在相应的文件名时，才显示图片。
• url_for('display_image', filename=session['content_file']) 和 url_for('display_image', filename=session['style_file']): 从 Session 中获取文件名，并生成正确的图片 URL。

总结

通过使用 Flask Session，我们可以轻松地在多个请求之间存储和访问数据，从而实现更复杂的功能，例如同时显示多个上传图片。 这种方法避免了使用全局变量，使代码更清晰、更易于维护。

注意事项

• 安全性: 务必设置一个安全的 secret_key，以保护 Session 数据的安全。
• Session 存储: 选择合适的 Session 存储类型，根据你的应用需求和服务器环境进行选择。文件系统存储适用于开发和小型应用，对于大型应用，建议使用 Redis 或数据库等更可靠的存储方式。
• 错误处理: 添加更完善的错误处理机制，例如处理 Session 中不存在文件名的错误。
• 文件大小限制: 根据实际需求，限制上传文件的大小，以防止服务器资源被滥用。

通过以上步骤，你就可以在 Flask 应用中同时渲染多个上传图片，而不会替换之前的图片。

以上就是《Flask多图上传不覆盖旧图技巧》的详细内容，更多关于的资料请关注golang学习网公众号！