python - 动态查询sqlalchemy中的列子集

Question

假设表中只需要两列(name 和 id)。我会编写如下代码:

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

但是如果列名是动态的，

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

但是上面那个看起来很难看。有没有更好的推荐方式？

Answer 1

您可以使用select() :

from sqlalchemy.sql import select

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

希望对您有所帮助。