如何通过Lambda将Base64编码文件上传至S3并返回URL？

嘿，我来帮你搞定这个问题。先看看你之前代码里的几个问题：

• 你把文件对象f传给了put_object的Body参数，但此时文件指针已经移到了末尾，而且其实完全没必要先写入本地文件——直接用解码后的二进制数据上传更高效
• 你定义了convertedFileString但没用到，处理URL安全Base64的步骤应该作用于你的new_string变量

下面是修正后的完整代码，附带详细说明：

先确认boto3配置 #

首先得确保boto3能访问你的S3账户：

• 如果是本地运行，配置~/.aws/credentials文件或者设置AWS_ACCESS_KEY_ID和AWS_SECRET_ACCESS_KEY环境变量
• 如果是在AWS服务（比如Lambda、EC2）上运行，给对应的IAM角色添加S3上传权限就行

完整可运行代码 #

import base64
import boto3
from botocore.exceptions import ClientError

def upload_base64_img_to_s3(new_string, bucket_name='ag-grid', filename='image.jpg'):
    try:
        # 处理URL安全的Base64编码（替换特殊字符）
        converted_string = new_string.replace('-', '+').replace('_', '/')
        # 解码Base64字符串为二进制图片数据
        img_data = base64.b64decode(converted_string)
        
        # 初始化S3资源
        s3 = boto3.resource('s3')
        bucket = s3.Bucket(bucket_name)
        
        # 上传图片到S3，指定Content-Type保证浏览器能正确识别
        bucket.put_object(
            Key=filename,
            Body=img_data,
            ContentType='image/jpeg',
            # 如果需要公开访问，添加下面这行（注意桶权限也要配合）
            # ACL='public-read'
        )
        
        # 生成访问URL，分两种情况：
        # 情况1：桶/对象是公开可读的
        public_url = f"https://{bucket_name}.s3.amazonaws.com/{filename}"
        
        # 情况2：桶是私有的，生成预签名临时URL（默认有效期1小时）
        # s3_client = boto3.client('s3')
        # presigned_url = s3_client.generate_presigned_url(
        #     'get_object',
        #     Params={'Bucket': bucket_name, 'Key': filename},
        #     ExpiresIn=3600
        # )
        
        # 返回对应的URL，根据你的需求选上面其中一种
        return {
            'statusCode': 200,
            'body': public_url
        }
    except ClientError as e:
        # 捕获S3相关的具体错误
        return {
            'statusCode': 500,
            'body': f"S3上传失败: {e.response['Error']['Message']}"
        }
    except Exception as e:
        # 捕获其他通用错误
        return {
            'statusCode': 500,
            'body': f"出错了: {str(e)}"
        }

关键改进点说明 #

• 跳过本地文件写入：直接使用解码后的img_data作为上传主体，避免了文件IO的开销和指针位置问题
• 正确处理Base64：把URL安全格式的Base64转换回标准格式，确保解码不会出错
• 指定Content-Type：上传时设置image/jpeg，这样S3会正确返回文件类型，浏览器访问URL时能直接显示图片而不是下载
• 灵活的URL生成：根据桶的权限选择公开URL或者预签名临时URL，满足不同场景需求

额外提示 #

• 如果要使用公开URL，记得给桶或者对象设置公开读取权限（可以在put_object里加ACL='public-read'，或者在桶的权限策略里配置）
• 文件名最好用唯一值（比如UUID），避免覆盖已有文件

内容的提问来源于stack exchange，提问作者user8022517