더북(TheBook)

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_sqlDataFrame 객체가 사용할 수 있는 함수로, 데이터베이스 연결 객체(con)를 매개변수로 전달하면 해당 데이터베이스로 DataFrame을 저장하는 아주 편리한 기능이 있습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.