如何从 SQLAlchemy 获取列名？

SQLAlchemy是一个流行的Python ORM库，它可以让我们以Python对象的方式操作关系型数据库，使用它可以更加方便和高效地操作数据库。在使用SQLAlchemy时，有时需要获取表的列名，本文将介绍如何从SQLAlchemy获取列名。

使用反射

SQLAlchemy提供了反射机制，可以通过反射获取表结构信息，包括列名、列类型等。以下示例演示了如何使用反射获取表users的列名：

from sqlalchemy import create_engine, Table, MetaData

engine = create_engine('mysql://user:password@host/dbname', echo=True)
metadata = MetaData()
table = Table('users', metadata, autoload=True, autoload_with=engine)

print(table.columns.keys())

以上代码先创建了一个数据库连接引擎engine，然后创建一个MetaData对象metadata，接着创建一个表对象table，使用autoload参数可以自动从数据库中读取表结构信息，并生成相关的映射关系。最后使用table.columns.keys()方法获取列名列表，并打印输出。

手动指定表结构

还可以手动指定表结构信息，使用Column对象来定义列，然后使用table.columns.keys()方法获取列名。以下示例演示了如何手动指定users表的列信息：

from sqlalchemy import create_engine, Column, Integer, String, MetaData

engine = create_engine('mysql://user:password@host/dbname', echo=True)
metadata = MetaData()
table = Table('users', metadata,
              Column('id', Integer, primary_key=True),
              Column('name', String(50)),
              Column('email', String(50)))

print(table.columns.keys())

以上代码定义了id、name、email三个列信息，并将它们与表users关联起来，最后使用table.columns.keys()输出列名列表。

总结

SQLAlchemy提供了反射和手动指定表结构两种方式来获取列名，使用反射需要注意数据库表和表结构一致，手动指定表结构则需要手动定义每个列信息，可以根据实际需求选择合适的方式。