我试图在sqlalchemy中执行此查询
SELECT id, name FROM user WHERE id IN (123, 456)
我想在执行时绑定列表[123,456]。
我试图在sqlalchemy中执行此查询
SELECT id, name FROM user WHERE id IN (123, 456)
我想在执行时绑定列表[123,456]。
当前回答
只是对上述答案的补充。
如果你想用"IN"语句执行SQL,你可以这样做:
ids_list = [1,2,3]
query = "SELECT id, name FROM user WHERE id IN %s"
args = [(ids_list,)] # Don't forget the "comma", to force the tuple
conn.execute(query, args)
两个点:
IN语句不需要括号(如“…(% s)),只要写上“…在% s” 强制id列表为元组的一个元素。不要忘记“,”:(ids_list,)
编辑 注意,如果list的长度为1或0,将会引发一个错误!
其他回答
只是对上述答案的补充。
如果你想用"IN"语句执行SQL,你可以这样做:
ids_list = [1,2,3]
query = "SELECT id, name FROM user WHERE id IN %s"
args = [(ids_list,)] # Don't forget the "comma", to force the tuple
conn.execute(query, args)
两个点:
IN语句不需要括号(如“…(% s)),只要写上“…在% s” 强制id列表为元组的一个元素。不要忘记“,”:(ids_list,)
编辑 注意,如果list的长度为1或0,将会引发一个错误!
如何
session.query(MyUserClass).filter(MyUserClass.id.in_((123,456))).all()
编辑:如果没有ORM,它将是
session.execute(
select(
[MyUserTable.c.id, MyUserTable.c.name],
MyUserTable.c.id.in_((123, 456))
)
).fetchall()
Select()接受两个参数,第一个是要检索的字段列表,第二个是where条件。您可以通过c(或columns)属性访问表对象上的所有字段。
或者也可以使用.in_(list),类似于@Carl已经建议的方法 作为
stmt = select(
id,
name
).where(
id.in_(idlist),
)
完整的代码,假设你在User类中有数据模型:
def fetch_name_ids(engine, idlist):
# create an empty dataframe
df = pd.DataFrame()
try:
# create session with engine
session = Session(engine, future=True)
stmt = select(
User.id,
User.name
).where(
User.id.in_(idlist),
)
data = session.execute(stmt)
df = pd.DataFrame(data.all())
if len(df) > 0:
df.columns = data.keys()
else:
columns = data.keys()
df = pd.DataFrame(columns=columns)
except SQLAlchemyError as e:
error = str(e.__dict__['orig'])
session.rollback()
raise error
else:
session.commit()
finally:
engine.dispose()
session.close()
return df
通过表达式API(基于注释的表达式API是这个问题所要求的),您可以使用相关列的in_方法。
查询
SELECT id, name FROM user WHERE id in (123,456)
use
myList = [123, 456]
select = sqlalchemy.sql.select([user_table.c.id, user_table.c.name], user_table.c.id.in_(myList))
result = conn.execute(select)
for row in result:
process(row)
这里假设user_table和conn已经适当地定义了。
假设你使用声明式风格(即ORM类),这是相当简单的:
query = db_session.query(User.id, User.name).filter(User.id.in_([123,456]))
results = query.all()
db_session是你的数据库会话,而User是ORM类,__tablename__等于"users"。