SQLAlchemy多态性：如何将父类Room对象转换为子类Office对象

嘿，我来帮你搞定把父类Room对象转换成子类Office对象的问题！先看看你给出的基础模型代码，我帮你补全了截断的部分：

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, declarative_base
from sqlalchemy import Column, Integer, String, ForeignKey

Base = declarative_base()

engine = create_engine('sqlite:///:memory:')
Session = sessionmaker(bind=engine)

class Room(Base):
    __tablename__ = 'rooms'
    id = Column(Integer, primary_key=True)
    floor = Column(Integer)
    usage = Column(String(50))
    __mapper_args__ = {'polymorphic_on': usage,
                       'polymorphic_identity': 'empty'}
    
    def __repr__(self):
        return f"<Room(id={self.id}, floor={self.floor}, usage='{self.usage}')>"
    
class Office(Room):
    __tablename__ = 'offices'
    id = Column(Integer, ForeignKey('rooms.id'), primary_key=True)
    computers = Column(Integer)
    desks = Column(Integer)
    __mapper_args__ = {'polymorphic_identity': 'office'}
    
    def __repr__(self):
        return f"<Office(id={self.id}, floor={self.floor}, computers={self.computers}, desks={self.desks})>"

接下来咱们一步步实现转换逻辑：

转换的核心思路 #

你用的是关联表继承（Joined Table Inheritance），父类Room和子类Office分别对应不同的表，要完成转换需要做两件关键的事：

1. 在子类对应的offices表中插入一条与目标Room关联的记录，填充子类特有的属性（比如computers、desks）
2. 更新父类Room的usage字段为子类的多态标识office，让SQLAlchemy识别这个对象的类型变化

具体代码实现 #

# 初始化会话并创建表结构
session = Session()
Base.metadata.create_all(engine)

# 先创建一个普通的Room对象并提交到数据库
room = Room(floor=3, usage='empty')
session.add(room)
session.commit()

# 开始转换操作
# 1. 给offices表添加关联记录，用merge避免主键冲突
office_record = Office(id=room.id, computers=5, desks=8)
session.merge(office_record)

# 2. 更新Room的usage字段为子类的多态标识
room.usage = 'office'
session.commit()

# 刷新会话，让ORM重新识别对象类型
session.refresh(room)

# 验证结果
print(type(room))  # 输出: <class '__main__.Office'>
print(room)        # 输出: <Office(id=1, floor=3, computers=5, desks=8)>

注意事项 #

• 确保要转换的Room对象在offices表中没有对应的记录，否则会触发主键冲突错误
• 必须更新usage字段，这是SQLAlchemy区分多态类型的核心依据，不修改的话ORM依然会把它当作Room对象加载
• 如果需要批量转换，可以用SQL语句直接操作两张表，效率会更高，不过要记得同步ORM会话的状态

备注：内容来源于stack exchange，提问作者Javier Perez