set_leverage_migration.py

"""
set_leverage_migration.py
─────────────────────────
Yeh script existing SQLite database mein do naye columns add karta hai:
  1. trades.leverage       (INTEGER, default 1)
  2. challenges.max_leverage (INTEGER, default 10)

Ek baar chalao — project root se:
    python set_leverage_migration.py
"""

import sqlite3
import os

DB_PATH = os.path.join(os.path.dirname(__file__), "prop_platform.db")

def column_exists(cursor, table, column):
    cursor.execute(f"PRAGMA table_info({table})")
    cols = [row[1] for row in cursor.fetchall()]
    return column in cols

def run_migration():
    if not os.path.exists(DB_PATH):
        print(f"❌ Database not found at: {DB_PATH}")
        print("   Sahi path set karo ya is file ko project root mein rakho.")
        return

    conn = sqlite3.connect(DB_PATH)
    cur  = conn.cursor()

    # ── 1. trades.leverage ──
    if not column_exists(cur, "trades", "leverage"):
        cur.execute("ALTER TABLE trades ADD COLUMN leverage INTEGER DEFAULT 1")
        print("✅ trades.leverage column added (default=1)")
    else:
        print("ℹ️  trades.leverage already exists — skip")
        
    # ── 2. pending_orders.leverage ──
    if not column_exists(cur, "pending_orders", "leverage"):
        cur.execute("ALTER TABLE pending_orders ADD COLUMN leverage INTEGER DEFAULT 1")
        print("✅ pending_orders.leverage column added (default=1)")
    else:
        print("ℹ️  pending_orders.leverage already exists — skip")    

    # ── 2. challenges.max_leverage ──
    if not column_exists(cur, "challenges", "max_leverage"):
        cur.execute("ALTER TABLE challenges ADD COLUMN max_leverage INTEGER DEFAULT 10")
        print("✅ challenges.max_leverage column added (default=10)")
    else:
        print("ℹ️  challenges.max_leverage already exists — skip")

    conn.commit()
    conn.close()
    print("\n🎉 Migration complete! Server restart karo.")

if __name__ == "__main__":
    run_migration()