Psycopg2中多列结合=ANY实现批量查询的问题

要解决这个问题，核心是让PostgreSQL明确识别传入的复合参数类型，同时利用Psycopg2的参数绑定机制避免动态SQL。以下是具体方案：

解决方案原理 #

PostgreSQL的=ANY操作符需要匹配同类型的复合值数组，之前的报错都是因为参数类型未正确转换：

• 传入多组元组时，Psycopg2默认将其包装为record类型，无法与表中的text列组成的复合值匹配；
• 直接传列表时，PostgreSQL无法推断数组元素的具体类型，导致类型不匹配。

通过在SQL中显式指定参数的复合类型转换，就能让Psycopg2和PostgreSQL正确协同工作。

代码实现 #

基础写法（推荐） #

直接在SQL中对占位符进行类型转换，指定复合类型为(text, text)（根据你的表字段类型调整）：

import psycopg2

# 未知数量的账户对列表
account_pairs = [('name', 'number'), ('name1', 'number2')]

# 数据库连接（替换为你的连接参数）
conn = psycopg2.connect("dbname=your_db user=your_user password=your_pwd host=your_host")
cur = conn.cursor()

# 执行查询，通过::(text,text)[]指定参数类型
cur.execute(
    """
    SELECT * FROM accounts
    WHERE (account_name, account_number) = ANY(%s::(text, text)[]);
    """,
    (account_pairs,)  # 注意参数是一个元组，包裹列表
)

# 获取结果
matches = cur.fetchall()
for row in matches:
    print(row)

cur.close()
conn.close()

进阶写法（利用Psycopg2适配器） #

如果不想在SQL中写类型转换，可以使用Psycopg2的NamedTupleList适配器，自动将列表元组转换为PostgreSQL复合数组：

import psycopg2
from psycopg2.extras import NamedTupleList

account_pairs = [('name', 'number'), ('name1', 'number2')]

conn = psycopg2.connect("dbname=your_db user=your_user")
cur = conn.cursor()

cur.execute(
    """
    SELECT * FROM accounts
    WHERE (account_name, account_number) = ANY(%s);
    """,
    (NamedTupleList(account_pairs),)
)

matches = cur.fetchall()

cur.close()
conn.close()

为什么之前的方法失败 #

1. 传入多组元组报错：
   当你尝试用VALUES结合参数绑定时，Psycopg2无法将多组元组拆分为VALUES的行，反而会把整个列表当成一个record类型参数，导致text = record的操作符不存在。
2. 移除VALUES传列表报错：
   直接传列表时，PostgreSQL无法推断数组元素的具体类型（默认是unknown），无法与表中的text类型字段组成的复合值进行比较，因此报类型不匹配。

内容的提问来源于stack exchange，提问作者Brooks