sqlalchemy - 如何获取 SQLAlchemy 对象的主键名称？

Question

我只想以编程方式确定 SQLalchemy 模型主键的名称。

Answer 1

如果使用 SQLAlchemy 0.8 或更高版本，则应使用 runtime inspection API .

如果模型类是 User 并且唯一的主键是 id，

>>> from sqlalchemy.inspection import inspect
>>> inspect(User).primary_key[0].name
'id'

如果模型类是User并且有很多主键，

>>> from sqlalchemy.inspection import inspect
>>> [key.name for key in inspect(User).primary_key]
['id', 'name']

如果使用 SQLAlchemy 版本低于 0.8，则应使用 class_mapper() 函数。

>>> from sqlalchemy.orm import class_mapper
>>> class_mapper(User).primary_key[0].name
'id'

在本例中，inspect() 函数和 class_mapper() 函数返回相同的对象，但使用 inspect() 是更加面向 future 。