import sqlite3
import os

db_path = r"D:\Code\iDRAC_Info\idrac_info\backend\instance\site.db"

if not os.path.exists(db_path):
    print(f"Database not found at {db_path}")
else:
    try:
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        
        # Check if column exists
        cursor.execute("PRAGMA table_info(telegram_bots)")
        columns = [info[1] for info in cursor.fetchall()]
        
        if 'notification_types' not in columns:
            print("Adding notification_types column...")
            cursor.execute("ALTER TABLE telegram_bots ADD COLUMN notification_types VARCHAR(255) DEFAULT 'auth,activity,system'")
            conn.commit()
            print("Column added successfully.")
        else:
            print("Column already exists.")
            
        conn.close()
    except Exception as e:
        print(f"Error: {e}")
