SQLAlchemy批量保存后无法获取默认UUID主键问题求助

问题原因 #

SQLAlchemy的bulk_save_objects是为性能优化设计的低层级批量操作，它不会触发模型实例的完整生命周期钩子，也不会将Python端默认值生成的主键值同步回内存中的实例对象。这就是单个创建实例时能正常获取UUID，但批量插入后实例uuid属性仍为None的原因。

解决方案 #

1. 创建实例时手动生成UUID（推荐） #

绕过Column的default配置，在创建Car实例时直接生成并赋值UUID，让内存中的实例本身持有UUID值，无需依赖插入后的同步逻辑。

from uuid import uuid4

# 手动为每个实例生成并赋值UUID
cars = [Car(uuid=str(uuid4()), **car_data) for car_data in raw_cars_data]

session.bulk_save_objects(cars)
session.commit()

# 直接访问实例的uuid属性即可获取值
for car in cars:
    print(car.uuid)

2. 使用add_all替代bulk_save_objects #

add_all属于SQLAlchemy的高层级操作，会完整触发对象的生命周期管理，包括执行默认值生成逻辑并将值同步到实例。缺点是处理超大量数据时，性能略低于bulk_save_objects。

cars = [Car(**car_data) for car_data in raw_cars_data]

session.add_all(cars)
session.commit()

# 提交后实例的uuid已被自动赋值
for car in cars:
    print(car.uuid)

3. 利用INSERT语句的RETURNING特性同步主键 #

如果必须使用批量插入且希望保留Column的default逻辑，可以通过insert语句结合returning语法获取插入后的UUID，再手动赋值给实例。该方案依赖数据库支持RETURNING语法（如PostgreSQL、MySQL 8.0+等）。

from sqlalchemy import insert

# 构建INSERT语句并指定返回uuid字段
insert_stmt = insert(Car).values(raw_cars_data).returning(Car.uuid)
# 执行语句并获取所有插入的UUID
inserted_uuids = session.execute(insert_stmt).scalars().all()
session.commit()

# 将返回的UUID对应赋值给内存中的实例
for car, uuid_val in zip(cars, inserted_uuids):
    car.uuid = uuid_val

# 现在可正常访问实例的uuid属性
for car in cars:
    print(car.uuid)

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