import sqlite3
import pandas as pd
import os

DB_PATH = "stock_data.db"


def init_db():
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    cursor.execute("""
    CREATE TABLE IF NOT EXISTS stock_daily (
        code TEXT,
        date TEXT,
        open REAL,
        close REAL,
        high REAL,
        low REAL,
        volume REAL,
        PRIMARY KEY (code, date)
    )
    """)

    conn.commit()
    conn.close()