【发布时间】:2021-04-21 07:21:21
【问题描述】:
新的每日 csv 文件不断生成并写入目录。任务是每天自动将新的 csv 导入 postgres 表(因此 PG 表将每天附加一条新记录)。我有一个适用于导入所有 csv 文件的 python 脚本。如何只导入最新的 csv?
import glob
import psycopg2
file_names = glob.glob('path/to/directory/*.csv')
con = psycopg2.connect(database="XXXX", user="XXXX", password="XXXX", host="XXXX")
for file_name in file_names:
with open(file_name, 'r') as file_in:
next(file_in)
with con.cursor() as cur:
cur.copy_from(file_in, "tbl_name", columns=('objectid', 'starttime', 'endtime', 'comments'), sep=",")
con.commit()
con.close()
【问题讨论】:
标签: python postgresql automation