动态查询 sqlalchemy 中的列子集

假设表中只需要两列（名称和 ID）。我会编写如下代码：

session.query(User.id, User.name).all()

但如果列名是动态的，

def get_data(table, columns):
    return session.query(*(getattr(table, column) for column in columns)).all()

但上面那张看起来很丑。有更好推荐的方法吗？

您可以使用select():

from sqlalchemy.sql import select

columns = ['id', 'name']
print(session.query(select(from_obj=User, columns=columns)).all())

希望有帮助。