如何在类定义后设置多态身份？

Question

我有一个类，这个类定义了一个表，并且有一个映射参数：polymorphic_on: column

我还有一些其他的类，它们是从这个类继承而来的，这些类都需要设置__mapper_args__ 'polymorphic_identity'，我希望在类定义之后再进行设置，而不是在类定义的时候就设置好。

我该如何在类定义之后设置这个多态身份的映射参数呢？

谢谢。

Answer 1

这不是一个完全公开的API，但应该可以用：

from sqlalchemy.orm import class_mapper
mapper = class_mapper(MySubClass)
mapper.polymorphic_identity = "some_identity"
mapper.polymorphic_map["some_identity"] = mapper
# in 0.8, might try to fix this for 0.9
mapper._identity_class = mapper.inherits._identity_class

这可以作为一个功能来支持，所以你可以随意提交一个功能请求。

补充：这里有一个使用声明式的测试：

from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy.orm import class_mapper

Base= declarative_base()


class A(Base):
    __tablename__ = "a"

    id = Column(Integer, primary_key=True)
    type = Column(String)
    __mapper_args__ = {"polymorphic_on": type, "polymorphic_identity": "a"}

class B(A):
    pass

mapper = class_mapper(B)
mapper.polymorphic_identity = "b"
mapper.polymorphic_map["b"] = mapper

e = create_engine("sqlite://", echo=True)

Base.metadata.create_all(e)

s = Session(e)

s.add_all([A(), B(), B(), A(), A()])
s.commit()

s.close()

assert [obj.type for obj in s.query(B).order_by(A.id)] \
            == ['b', 'b']

assert [type(obj) for obj in s.query(B).order_by(A.id)] \
            == [B, B]

assert [obj.type for obj in s.query(A).order_by(A.id)] \
            == ['a', 'b', 'b', 'a', 'a']