如何在Python中用SQL查询匹配PostgreSQL表列与列表的文件？

要找出PostgreSQL中xml_joblist表的filename列与Python列表files_name的交集，推荐用参数化查询实现，既安全又高效，以下是两种常见实现方式：

方法1：使用psycopg2直接操作 #

psycopg2（PostgreSQL官方Python驱动）支持直接将Python列表作为参数传入，自动转换为SQL的IN或ANY格式：

import psycopg2

# 你的文件名列表（注意必须是字符串类型）
files_name = ["file_789", "file_456", "file_555", "file_111"]

# 建立数据库连接
conn = psycopg2.connect(
    dbname="你的数据库名",
    user="你的用户名",
    password="你的密码",
    host="你的数据库地址"
)
cur = conn.cursor()

# 使用ANY子句的参数化SQL（IN子句也可以，写法是WHERE filename IN %s）
sql = "SELECT filename FROM xml_joblist WHERE filename = ANY(%s);"

# 执行查询，传入列表作为参数
cur.execute(sql, (files_name,))

# 提取结果并转为列表
matching_files = [row[0] for row in cur.fetchall()]

print(matching_files)  # 输出: ['file_555', 'file_111']

# 关闭资源
cur.close()
conn.close()

方法2：使用SQLAlchemy（ORM方式） #

如果项目用SQLAlchemy，写法更简洁：

from sqlalchemy import create_engine, select, Table, Column, String, MetaData

# 初始化连接引擎
engine = create_engine('postgresql://用户名:密码@地址/数据库名')
metadata = MetaData()

# 映射数据库表
xml_joblist = Table(
    'xml_joblist',
    metadata,
    Column('filename', String)
)

files_name = ["file_789", "file_456", "file_555", "file_111"]

# 执行查询
with engine.connect() as conn:
    query = select(xml_joblist.c.filename).where(xml_joblist.c.filename.in_(files_name))
    result = conn.execute(query)
    matching_files = [row[0] for row in result]

print(matching_files)

关键注意事项 #

• 必须用参数化查询：绝对不要手动拼接SQL字符串，否则会引发SQL注入风险，且无法处理文件名中的特殊字符（如引号、空格）。
• 字符串类型检查：Python列表中的文件名必须是字符串（示例中已加引号），否则会导致类型错误。
• 大数据量场景：如果files_name列表长度超过数千条，建议改用临时表批量插入后关联查询，避免IN/ANY子句的长度限制。

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