要实现保存用户发布的图像并删除旧图像的功能,可以使用以下代码示例:
import os
from flask import Flask, request
app = Flask(__name__)
UPLOAD_FOLDER = 'path/to/upload/folder'
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@app.route('/', methods=['POST'])
def upload_image():
# 检查是否有文件被上传
if 'image' not in request.files:
return 'No file uploaded'
file = request.files['image']
# 检查文件名是否合法
if file.filename == '':
return 'No file selected'
if file and allowed_file(file.filename):
# 保存新图像
file.save(os.path.join(app.config['UPLOAD_FOLDER'], file.filename))
# 删除旧图像
delete_old_images()
return 'Image uploaded and old images deleted'
return 'Invalid file type'
def delete_old_images():
# 获取上传文件夹中的所有文件
files = os.listdir(UPLOAD_FOLDER)
# 删除上传文件夹中的所有文件
for file in files:
os.remove(os.path.join(UPLOAD_FOLDER, file))
if __name__ == '__main__':
app.run()
上述代码使用Flask框架创建了一个简单的Web应用,当用户上传图像时,会将图像保存到指定的文件夹中,并删除该文件夹中的所有旧图像文件。需要注意的是,你需要根据你的实际需求修改UPLOAD_FOLDER
的值为你想要保存图像的文件夹路径,并确保该文件夹存在。
此外,你还需要在HTML页面中添加一个表单,用于让用户选择并上传图像。以下是一个简单的HTML表单代码示例:
Upload Image
将上述HTML代码保存为index.html
文件,并将其放置在与上述Python代码相同的目录中。然后,你可以通过运行Python代码启动应用,并访问http://localhost:5000/
来上传图像。