使用 SQLAlchemy 动态获取列名

Question

我正在使用 SQLAlchemy 查询 sqlite db，如下所示：

import db

...

results = session.query(table).all()
        for result in results:
            print result
            print "how to print column name?"

这是一个来自 db 类的 sn-p：

class Device(Base):
    __tablename__ = "devices"
    name = Column(String(250), primary_key=True)
    location = Column(String(250), nullable=False)

    def __init__(self, name, location):
        self.name = name
        self.location = location

尝试根据文档在结果上使用“.column_descriptions”会引发“'list' object has no attribute 'column_descriptions'”错误。

动态获取列名和值的正确方法是什么？我想要这个，所以我可以构建一个函数来处理所有查询的 json 转换，而不是到处重复代码。

Answer 1

您可以在代码中使用Device.__table__.columns 获取Device 模型中的列列表。这些是Column 对象，您可以在迭代时在它们上使用.name 来获取名称。然后在查询的时候，就可以使用getattr或者其他的东西来获取你需要的值。

编辑：

例子：

col_names = Device.__table__.columns.keys()  # This gets the column names
sample_row = Device.query.first()  # I just query for a random row
required_values = {name: getattr(sample_row, name) for name in col_names}
# This is a dictionary comprehension

也许您也可以使用select 语法。

EDIT2：

这是dictionary comprehension上的一篇文章

好的，所以这个字典理解的作用是它获取键作为列的名称，并将值作为我查询的示例行中相应列的值。

您想打印list，所以我们可以这样做：

用这个替换字典理解行。

print([getattr(sample_row, name) for name in col_names])

这将在列表中打印该行的值，但您不会获得列名。

但为此，您可以创建一个元组列表。

print([(name, getattr(sample_row, name)) for name in col_names])

但您更简单的方法可能是使用 SQLAlchemy 提供的 select 语法。

这里的文档：http://docs.sqlalchemy.org/en/rel_0_9/core/tutorial.html#selecting