python - 列出多态继承选择并根据该选择创建实例

Question

我有一个连接表继承，例如 Animal、Cat 和 Dog。我想允许用户创建一个新的动物。

有没有办法将所有类型(猫、狗)放入用户列表中？有没有办法接受用户的选择并从中创建一个实例？ (我不想硬编码 if choice == "cat": Cat()。)

Answer 1

SQLAlchemy 跟踪父模型的多态子模型。使用inspect在 Animal 模型上找到 mapping of polymorphic identities .

>>> from sqlalchemy import inspect
>>> i = inspect(Animal)
>>> i.polymorphic_map
{'cat': <Mapper at 0x7fb4c7a8b390; Cat>,
 'dog': <Mapper at 0x7fb4c7a98240; Dog>}

您可以使用class_从映射器获取类。属性。

>>> i.polymorphic_map['cat'].class_
__main__.Cat

有了这两个部分，您现在可以获得要呈现给用户的类型列表，并且可以根据所选值实例化模型。

types = inspect(Animal).polymorphic_map.keys()
user_input = 'cat'
new_cat = inspect(Animal).polymorphic_map[user_input]()

如果层次结构有更多级别，例如 CalicoCat 和 HairlessCat，您可以使用 polymorphic_iterator查看整个层次结构。