import sqlite3
import os

DB_PATH = "db/history.db"

"""
cursor.execute('select id, name for users')
cursor.fetchone() => (1, 'alice', '010-1234-5678')

row['id']
"""

def get_connection():
    coon = sqlite3.connect(DB_PATH, isolation_level=None)
    # 조회 결과를 어떻게 반활할지 설정
    coon.row_factory = sqlite3.Row
    return coon

def init_db():
    # 디렉토리 생성
    os.makedirs("db", exist_ok=True)
    conn = get_connection()
    cursor = conn.cursor()
    cursor.execute("""create table if not exists transactions (
                    id integer primary key AUTOINCREMENT,
                    date text, 
                    category text,
                    merchant text,
                    amount integer
                    )
""")
    conn.commit()
    conn.close()

    print("DB 초기화")

def get_table_columns(table_name):
    conn = get_connection()
    cursor = conn.cursor()
    cursor.execute("PRAGMA table_info({})".format(table_name))
    columns = cursor.fetchall()
    conn.close()
    return [column[1] for column in columns]