Python Flask上传图片到S3 -随机文件名被覆盖

import os
import uuid
from flask import Flask, request
import boto3

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_image():
    # 获取上传的图片文件
    image_file = request.files['image']
    
    # 生成唯一的随机文件名
    filename = generate_unique_filename(image_file.filename)
    
    # 保存图片到本地临时目录
    image_path = os.path.join('/tmp', filename)
    image_file.save(image_path)
    
    # 保存图片到S3
    save_image_to_s3(image_path, filename)
    
    # 删除本地临时文件
    os.remove(image_path)
    
    return 'Image uploaded successfully!'

def generate_unique_filename(filename):
    s3 = boto3.client('s3')
    bucket_name = 'your_bucket_name'
    
    while True:
        # 生成唯一的随机文件名
        unique_filename = uuid.uuid4().hex + os.path.splitext(filename)[1]
        
        try:
            # 检查S3中是否已存在相同的文件名
            s3.head_object(Bucket=bucket_name, Key=unique_filename)
        except:
            # 文件不存在，返回唯一的文件名
            return unique_filename

def save_image_to_s3(image_path, filename):
    s3 = boto3.client('s3')
    bucket_name = 'your_bucket_name'
    
    # 上传图片到S3
    s3.upload_file(image_path, bucket_name, filename)

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