使用psycopg2的copy_from导入CSV时报UndefinedTable表不存在错误如何解决

你遇到的报错是因为 psycopg2 的 copy_from() 方法不会自动解析带 schema 前缀的表名字符串：当你直接传入 'test.new_table' 作为表名参数时，程序会将整段字符串识别为表名，在当前连接的默认 schema（通常是 public）下查找名为 test.new_table 的表，自然匹配不到你在 test schema 下创建的 new_table。

两种修改方式任选其一即可：

方案1：单独指定schema参数 #

copy_from() 方法提供了专门的 schema 参数来声明表所属的 schema，修改导入代码的对应行即可：

with open(csv_file, 'r') as f:
    next(f) # Skip the header row.
    # 单独通过schema参数指定test schema
    cur.copy_from(f, 'new_table', sep=',', schema='test')

方案2：修改当前连接的搜索路径 #

在执行导入操作前，先设置当前连接的 search_path 包含 test schema，后续操作就可以直接用表名访问 test 下的表：

conn = psycopg2.connect(dbname='mydb', user='postgres', password='mypassword', host='www.mydbserver.com', port='5432', sslmode='require')
cur = conn.cursor()
# 先设置搜索路径
cur.execute("SET search_path TO test, public")
with open(csv_file, 'r') as f:
    next(f) # Skip the header row.
    cur.copy_from(f, 'new_table', sep=',')
conn.commit()

• 如果创建表之后没有主动关闭连接，不需要重复创建数据库连接，复用原有连接即可，减少不必要的资源开销
• 导入前请确认CSV文件的列数、列顺序和表结构完全一致，避免出现导入数据错位/报错的问题

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