sqlalchemy - проблема двойного полиморфного наследования - PullRequest
Новая
       9

sqlalchemy - проблема двойного полиморфного наследования

3 голосов
/ 15 августа 2010

У меня есть отображение классов с двумя полиморфными наследованиями: 

#test classes

class AbstractA(Base):

    __tablename__ = "abstract_a"

    id = Column(Integer, primary_key=True)
    class_name = Column('class_name', String(50))

    __mapper_args__ = {
      'polymorphic_on': class_name,
    }


#some stuff here

class AbstractB(AbstractA):

    __tablename__ = "abstract_b"

    id = Column(Integer, ForeignKey('abstract_a.id'), primary_key=True)
    class_name = Column('class_name', String(50))

    __mapper_args__ = {
      'polymorphic_on': class_name,
      'polymorphic_identity': __tablename__,
    }

#some stuff here

class ClassA(AbstractB):

    __tablename__ = "table_a"
    __mapper_args__ = {
      'polymorphic_identity': __tablename__,
    }

    id = Column(Integer, ForeignKey('abstract_b.id'), primary_key=True)
    label = Column('label', String(50))

    def __init__(self, label):
        self.label = label

Я сохраняю объект ClassA: 

object = ClassA('toto')
db_session.add(object)
db_session.commit()

Когда я пытаюсь перезагрузить объект следующим образом: 

reloaded_object = AbstractB.query.first()

Я получаю обратно объект ClassA (просто отлично)

но когда я пытаюсь перезагрузить, как это: 

reloaded_object = AbstractA.query.first()

Я возвращаю объект AbstractA, потому что abstract_a.class_name не было установлено в polymorphic_identity.

Это проблема или ожидаемая работа?

1 Ответ

2 голосов
/ 25 августа 2010

Зачем вам нужен другой дискриминатор для AbstractB подклассов вместо использования одного AbstractA.class_name для всего дерева наследования? Просто удалите 2 строки из AbstractB определения, и ваш тест будет работать: 

class AbstractB(AbstractA):

    __tablename__ = "abstract_b"

    id = Column(ForeignKey('abstract_a.id'), primary_key=True)

    __mapper_args__ = {
        'polymorphic_identity': __tablename__,
    }
...