调用psycopg2查询单字段如何得到扁平列表？拆包位置有何最佳实践？

1. psycopg2 类似 sqlite3 row_factory 的配置方案 #

psycopg2 提供了cursor_factory（游标工厂）能力，功能和 sqlite3 的row_factory完全对等，你可以自定义游标返回格式：

• 官方自带了RealDictCursor等预设工厂，返回字典格式的查询结果
• 你也可以自定义仅返回第一个字段的专用游标，避免每次查询都手动拆包

如果只是临时查询单字段不想改游标配置，也可以用更简洁的方式直接拆包：

# 方式1：直接遍历游标（比fetchall内存效率更高，适合大数据量场景）
photos = [row[0] for row in cursor]

# 方式2：用itertools快速压平嵌套结构
from itertools import chain
photos = list(chain.from_iterable(cursor.fetchall()))

2. 拆包位置的最优选择 #

推荐优先选择在CRUD函数内部完成拆包，你提到的开发者预期不符的问题，完全可以通过规范接口定义解决：

• CRUD函数的核心作用就是封装数据库底层逻辑，给上层业务返回语义明确的结构，而不是把DB-API的嵌套元组格式暴露到业务层，导致每处调用都要写重复的拆包逻辑
• 只要把函数名、返回类型标注写清楚，就不会出现预期歧义：比如把函数命名为list_user_photo_ids，返回类型标注为-> list[str]（对应你实际的字段类型），任何人调用都能立刻知道返回的是扁平列表
• 如果有部分场景需要保留原始嵌套元组格式，可以加一个可选参数控制，兼顾灵活性：

def list_user_photo_ids(user_id: int, cursor, flatten: bool = True) -> list[str] | list[tuple[str]]:
    cursor.execute('SELECT photo FROM photos WHERE user_id = %s', (user_id,))
    raw_result = cursor.fetchall()
    if flatten:
        return [row[0] for row in raw_result]
    return raw_result

这种方案比在业务层调用后再拆包的可维护性高得多，后续如果查询字段调整，只需要修改CRUD内部逻辑即可，不需要改所有调用位置的拆包代码。

内容的提问来源于stack exchange，提问作者Давид Шико