Postgresql在循环中意外中断时无法将CSV写入表格

在处理大型CSV文件时，PostgreSQL中的COPY命令是一种快速将数据加载到表中的方法。但是，当循环中断时，无法将CSV写入表格。以下是一个解决方法的示例代码：

import psycopg2
import csv

def insert_csv_data(conn, table_name, csv_file_path):
    try:
        cursor = conn.cursor()
        # 创建临时表
        cursor.execute(f"CREATE TEMP TABLE temp_{table_name} AS SELECT * FROM {table_name} LIMIT 0")
        
        with open(csv_file_path, 'r') as file:
            # 读取CSV文件
            csv_data = csv.reader(file)
            next(csv_data)  # 跳过标题行
            # 执行循环插入
            for row in csv_data:
                cursor.execute(f"INSERT INTO temp_{table_name} VALUES (%s, %s, %s)", row)
                
        # 将临时表中的数据插入正式表中
        cursor.execute(f"INSERT INTO {table_name} SELECT * FROM temp_{table_name}")
        
        # 提交事务
        conn.commit()
        
    except (Exception, psycopg2.Error) as error:
        print("Error while inserting data from CSV file", error)
        conn.rollback()
        
    finally:
        # 关闭游标和数据库连接
        if cursor:
            cursor.close()

        if conn:
            conn.close()

上述代码使用psycopg2库来连接到PostgreSQL数据库。insert_csv_data函数负责将CSV文件的数据插入到表中。首先，它创建了一个临时表，然后使用csv.reader读取CSV文件的数据。在循环中，它将每一行数据插入到临时表中。最后，它将临时表中的数据插入到正式表中。

请注意，上述代码中使用的表名为table_name，CSV文件的路径为csv_file_path。您需要将这些参数替换为您自己的表名和CSV文件的路径。

此方法可以确保在循环中断时不会丢失数据，并且仅在整个CSV文件的所有行都成功插入到临时表后，才将数据插入到正式表中。