AWS Lambda运行时使用sqlacodegen生成SQLAlchemy类报错求助

问题根源在于sqlacodegen的API在新版本（v2.0+）中发生了变更，原有的CodeGenerator.from_engine类方法已被移除，需要手动构建数据库元数据并初始化CodeGenerator实例。

以下是适配新版本的Lambda代码实现：

import os
from sqlalchemy import create_engine, MetaData, declarative_base
from sqlacodegen.codegen import CodeGenerator

def lambda_handler(event, context):
    # 读取环境变量中的数据库配置
    db_user = os.environ['DB_USER']
    db_password = os.environ['DB_PASSWORD']
    db_name = os.environ['DB_NAME']
    db_host = os.environ['DB_HOST']

    # 创建数据库引擎（注意：若使用pymysql驱动，需改为mysql+pymysql://）
    engine = create_engine(f'mysql+pymysql://{db_user}:{db_password}@{db_host}/{db_name}')

    # 从数据库加载元数据（包含所有表结构）
    metadata = MetaData()
    metadata.reflect(bind=engine)

    # 声明SQLAlchemy的Declarative Base类
    base_class = declarative_base()

    # 初始化代码生成器，可根据需求调整参数
    generator = CodeGenerator(
        metadata,
        base_class=base_class,
        generate_types=True,  # 生成类型注解
        skip_plural_names=False  # 不跳过复数表名处理
    )

    # 生成模型代码
    code = generator.render()

    # 处理生成的代码（示例：打印或上传到S3）
    print(code)
    return {"statusCode": 200, "body": "Model code generated successfully"}

关键细节说明 #

1. 元数据加载：通过MetaData.reflect(bind=engine)从数据库中读取所有表、字段等结构信息，这是生成模型的核心数据来源。
2. Declarative Base：必须传入base_class参数，生成的模型会继承这个基类，才能符合SQLAlchemy的使用规范。
3. 数据库驱动：Lambda环境默认不包含MySQL驱动，需在部署包或Lambda层中安装pymysql或mysql-connector-python，同时修改连接URL的前缀（如mysql+pymysql://）。
4. 部署注意事项：
   • 将sqlacodegen、sqlalchemy和MySQL驱动打包到Lambda层，避免部署包过大。
   • 配置Lambda的VPC和安全组，确保其能访问目标MySQL数据库的3306端口。

内容的提问来源于stack exchange，提问作者Leonardo Soares Bezerra