db_helper.py
import sqlite3
def check_table_exist(db_name, table_name):
with sqlite3.connect('{}.db'.format(db_name)) as con:
cur = con.cursor()
sql = "SELECT name FROM sqlite_master WHERE type='table' and name=:table_name"
cur.execute(sql, {"table_name": table_name})
if len(cur.fetchall()) > 0:
return True
else:
return False
def insert_df_to_db(db_name, table_name, df, option="replace"):
with sqlite3.connect('{}.db'.format(db_name)) as con:
df.to_sql(table_name, con, if_exists=option)
insert_df_to_db 함수에서 사용하는 df.to_sql을 보면 to_sql은 DataFrame 객체가 사용할 수 있는 함수로, 데이터베이스 연결 객체(con)를 매개변수로 전달하면 해당 데이터베이스로 DataFrame을 저장하는 아주 편리한 기능이 있습니다.