sqlalchemy - 给定一个声明性 SQLAlchemy 类，我如何以编程方式检索其作为主键的 Column 对象的列表？

Question

假设我在 SQLAlchemy 中有一个类

class Note(Base):
    __tablename__ = 'notes'
    slug = Column('short_title', String, primary_key=True)
    date = Column('day', Date, primary_key=True)
    contents = Column(Text, nullable=True)

仅给定对 Note 类的引用，函数如何构建基于它们构成相应表的主键的列表 ['Note.slug', 'Note.date'] ？

此外，有没有办法根据列属性过滤类属性？ (例如可为空、类型、唯一性、索引中的成员资格等)

Answer 1

试试这个来找到主要字段。

from sqlalchemy import create_engine

engine = create_engine('sqlite:///:memory:', echo=True)

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, Date, Text
from sqlalchemy.orm.attributes import InstrumentedAttribute


Base = declarative_base()

class Note(Base):
    __tablename__ = 'notes'
    slug = Column('short_title', String, primary_key=True)
    date = Column('day', Date, primary_key=True)
    contents = Column(Text, nullable=True)

if __name__ == '__main__':
    import inspect
    predicate = lambda x: isinstance(x, InstrumentedAttribute)

    fields = inspect.getmembers(Note, predicate=predicate)

    fields_vars = [vars(y.property.columns[0]) for x,y in fields]

    primary_fields = [x['name'] for x in fields_vars if x['primary_key']]

    print "All Fields :", fields_vars

    print "Primary Fields :", primary_fields

通过这种方式，您可以从模型中找到任何类型的字段。