フォルダ内のCSVファイルを一括でDBに取り込む

GTFS-JPを取り込む際、フォルダ内のCSVファイルをすべてDB(PostgreSQL)に取り込む必要性に駆られました。

バイト先の方が、Pandasに取り込んでからDBにそのまま入れられると教えてくださったので、その時のメモです。

├─csv_importer.py
└─GTFS
  ├─file1.csv
  └─file2.csv

csv_importer.py

import pandas as pd
import glob
from sqlalchemy import create_engine

engine = create_engine('DB接続用文字列')
file_list = glob.glob("./GTFS/*")

for filename in file_list:
    print(filename[7:])
    df  = pd.read_csv(filename)
    df.to_sql(name=filename[7:],con=engine,if_exists='replace',index=None)