Session()如何从两个带约束的不同表中向上插入

# 导入相关的库和模块
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.exc import IntegrityError
from sqlalchemy.ext.declarative import declarative_base

# 创建数据库引擎和Session工厂
engine = create_engine('数据库连接字符串')
Session = sessionmaker(bind=engine)
Base = declarative_base()

# 定义两个表的ORM模型
class Table1(Base):
    __tablename__ = 'table1'
    id = Column(Integer, primary_key=True)
    data = Column(String, nullable=False, unique=True)

class Table2(Base):
    __tablename__ = 'table2'
    id = Column(Integer, primary_key=True)
    data = Column(String, nullable=False, unique=True)

# 创建Session实例
session = Session()

try:
    # 创建数据对象并插入表1
    data1 = Table1(data='data1')
    session.add(data1)

    # 创建数据对象并插入表2
    data2 = Table2(data='data2')
    session.add(data2)

    # 提交事务
    session.commit()

    # 打印成功消息
    print("数据插入成功！")

except IntegrityError as e:
    # 回滚事务，打印错误消息
    session.rollback()
    print("插入数据失败：", str(e))

finally:
    # 关闭Session
    session.close()