Python Sqlalchemy一次将数据插入多个表

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.exc import IntegrityError

# 创建数据库连接
engine = create_engine('数据库连接地址')

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

try:
    # 开始事务
    session.begin()

    # 插入数据到第一个表
    table1_data = {
        'column1': 'value1',
        'column2': 'value2'
    }
    session.execute(table1.insert().values(**table1_data))

    # 插入数据到第二个表
    table2_data = {
        'column1': 'value3',
        'column2': 'value4'
    }
    session.execute(table2.insert().values(**table2_data))

    # 提交事务
    session.commit()

except IntegrityError:
    # 回滚事务，处理插入冲突等错误
    session.rollback()
    raise

finally:
    # 关闭会话
    session.close()