解决Python对Postgres批量操作解决单值元组引发的报错

解决Python对Postgres批量操作解决单值元组引发的报错

背景

Python对Postgres批量操作时,若传入的条件变量为单元素的元组时,会引发语法报错,原因是python中单值元组为(“xxx”,)而逗号在sql语句中将会被识别出错,故使用代码示例中方法三中的写法来解决此问题

代码示例
import psycopg2.extras


pg_conn = psycopg2.connect(database="postgres", user="postgres", password="xxxxx", host="10.255.xx.xxx", port="5432")
cursor = pg_conn.cursor()


# 方法一:当变量为多值时,不会引发报错
id_list = [66, 70]
cursor.execute(f"select id,name,data from table_test where id in {tuple(id_list)}")
result_1 = cursor.fetchall()
pg_conn.commit()
print(result_1)

# 方法一:当变量为单值时会引发报错
id_list = [66]
cursor.execute(f"select id,name,data from table_test where id in {tuple(id_list)}")
result_2 = cursor.fetchall()
pg_conn.commit()
print(result_2)

# 方法三:正确写法
id_list = [66]
sql = "select id,name,data from table_test where id in %(id_list)s"
values = {"id_list": tuple(id_list)}
cursor.execute(sql, values)
result_3 = cursor.fetchall()
pg_conn.commit()
print(result_3)

cursor.close()
pg_conn.close()
方法二报错

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值