python - 将数据框复制到具有默认值的列的 postgres 表

Question

我有以下 postgreSql 表stock，结构如下，列insert_time有一个默认值now()

|    column   |  pk |    type   |
+-------------+-----+-----------+
| id          | yes | int       |
| type        | yes | enum      |
| c_date      |     | date      |
| qty         |     | int       |
| insert_time |     | timestamp |

我正在尝试copy以下df

|  id | type |    date    | qty  |
+-----+------+------------+------+
| 001 | CB04 | 2015-01-01 |  700 |
| 155 | AB01 | 2015-01-01 |  500 |
| 300 | AB01 | 2015-01-01 | 1500 |

我psycopg用来上传df到表stock

cur.copy_from(df, stock, null='', sep=',')
conn.commit()

收到此错误。

DataError: missing data for column "insert_time"
CONTEXT:  COPY stock, line 1: "001,CB04,2015-01-01,700"

我期待使用 psycopg copy_from 函数，我的 postgresql 表将自动填充插入时间旁边的行。

|  id | type |    date    | qty  |     insert_time     |
+-----+------+------------+------+---------------------+
| 001 | CB04 | 2015-01-01 |  700 | 2018-07-25 12:00:00 |
| 155 | AB01 | 2015-01-01 |  500 | 2018-07-25 12:00:00 |
| 300 | AB01 | 2015-01-01 | 1500 | 2018-07-25 12:00:00 |

Answer 1

您可以像这样指定列：

cur.copy_from(df, stock, null='', sep=',', columns=('id', 'type', 'c_date', 'qty'))