flask - 从flask-sqlalchemy db获取所有模型

Question

在 SQLAlchemy 中构建数据库后，我想获取所有模型，也就是说，代表它们的类

>>> db
<SQLAlchemy engine='postgresql:///example'>
>>> ???
[<db.Model 'User'>, <db.Model 'Comment'>, <db.Model 'Article'>]

这能实现吗？怎么样？

我可以体面地拿到 table ，只是没有模型。例如:

>>> for t in db.metadata.tables.items():
        print(t)

这给出了表格而不是模型本身

Answer 1

有几个相关的问题，但我没有看到任何完全重复的问题。

使用您的代码获取表名，以及 this question的接受的答案来获取类，我们可以将类与在您的数据库中注册的表名进行匹配。

classes, models, table_names = [], [], []
for clazz in db.Model._decl_class_registry.values():
    try:
        table_names.append(clazz.__tablename__)
        classes.append(clazz)
    except:
        pass
for table in db.metadata.tables.items():
    if table[0] in table_names:
        models.append(classes[table_names.index(table[0])])

哪里 models是在您的数据库中注册的模型列表。

try catch 是必需的，因为 <sqlalchemy.ext.declarative.clsregistry._ModuleMarker object>将包含在 for clazz in ... 中循环，它没有 __tablename__属性。