Python使用boto3时如何优雅处理S3存储桶列表空/None的异常

原代码存在的核心问题 #

• 异常捕获过于宽泛：直接捕获所有Exception会掩盖不可预见的错误，不利于问题排查
• 空值判断存在漏洞：如果调用s3.list_buckets()抛出异常，buckets_list会保持None，后续直接访问buckets_list['Buckets']会触发TypeError
• 输出冗余：同时使用print和logging输出重复信息，生产环境通常只用logging即可
• 字符串格式化写法过时：使用%和str.format()的写法可以替换为Python 3.6+支持的f-string，可读性更高
• 异常抛出不规范：直接抛出泛型Exception，无法和其他异常做区分，建议使用更具体的内置异常类
• 嵌套层级过深：两层嵌套的try块可以合并简化，降低代码维护成本
• 业务逻辑存在误区：s3.list_buckets()返回的是当前AWS账号下全局所有存储桶，和客户端初始化的区域无关，想要过滤指定区域的桶需要额外调用接口判断桶的所属区域

优化后的Pythonic写法 #

import os
import logging
import boto3
from botocore.exceptions import ClientError

# 提前配置日志即可，不需要混用print
logging.basicConfig(level=logging.ERROR)

ALLOWED_REGION = "us-east-1"

try:
    my_region = os.environ["AWS_REGION"]
    if my_region != ALLOWED_REGION:
        raise ValueError(f"Region {my_region} not available, only {ALLOWED_REGION} is supported")

    s3 = boto3.client("s3", region_name=my_region)
    buckets_resp = s3.list_buckets()
    buckets = buckets_resp.get("Buckets", [])

    if not buckets:
        print("No buckets found in current account")
    else:
        for bucket in buckets:
            bucket_name = bucket["Name"]
            # 如果需要过滤当前区域的桶，加上下面的逻辑
            # loc_resp = s3.get_bucket_location(Bucket=bucket_name)
            # if loc_resp["LocationConstraint"] == my_region:
            # 你的业务逻辑

except KeyError:
    logging.error("AWS_REGION environment variable not found")
    raise
except ClientError as e:
    logging.error(f"S3 operation failed: {e}")
    raise
except ValueError as e:
    logging.error(e)
    raise
except Exception as e:
    logging.error(f"Unexpected error occurred: {e}")
    raise

额外优化点说明 #

• 用dict.get()方法获取Buckets字段，默认返回空列表，避免KeyError
• 异常分支按类型拆分，不同错误对应不同的处理逻辑和提示信息
• 提前退出不符合条件的分支，减少嵌套层级，符合「早返回」的编码规范
• 固定常量单独抽离，后续修改更方便

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