database.py

# database.py
import sqlite3
import json

DB_PATH = 'service.db'
JSON_PATH = 'data/service_slots.json'

def init_db():
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS service_slots (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            date TEXT NOT NULL,
            time TEXT NOT NULL,
            booked INTEGER NOT NULL DEFAULT 0
        )
    ''')
    conn.commit()

    # Optionally clear previous data
    cursor.execute('DELETE FROM service_slots')
    conn.commit()

    # Seed from JSON
    with open(JSON_PATH, 'r') as f:
        slots = json.load(f)
        for slot in slots:
            cursor.execute(
                'INSERT INTO service_slots (date, time, booked) VALUES (?, ?, ?)',
                (slot['date'], slot['time'], int(slot['booked']))
            )
    conn.commit()
    conn.close()
    print("Database seeded from JSON.")

def seed_router_plans():
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS router_plans (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            speed INTEGER NOT NULL,
            price INTEGER NOT NULL
        )
    ''')
    conn.commit()

    # Optionally clear previous data
    cursor.execute('DELETE FROM router_plans')
    conn.commit()

    # Seed from JSON
    with open(r'D:\void\data\router_plans.json', 'r') as f:
        plans = json.load(f)
        for plan in plans:
            cursor.execute(
                'INSERT INTO router_plans (name, speed, price) VALUES (?, ?, ?)',
                (plan['name'], plan['speed'], plan['price'])
            )
    conn.commit()
    conn.close()
    print("Router plans seeded from JSON.")

if __name__ == "__main__":
    #init_db()
    seed_router_plans()