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

时间:2018-07-25 14:59:56

标签: python postgresql dataframe psycopg

我有以下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 |

我正在使用psycopgdf上传到表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 |

1 个答案:

答案 0 :(得分:0)

您可以指定这样的列:

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