import os
import sys
import psycopg2
from pathlib import Path
from dotenv import load_dotenv

# Load .env
ROOT_DIR = Path(__file__).resolve().parent.parent.parent
load_dotenv(ROOT_DIR / ".env")

def get_db_connection():
    db_host = os.getenv("DB_HOST", "192.168.3.195")
    db_user = os.getenv("DB_USER", "value")
    db_pass = os.getenv("DB_PASSWORD", "Value609!")
    db_name = os.getenv("DB_NAME", "fa3")
    db_port = os.getenv("DB_PORT", "5432")

    return psycopg2.connect(
        host=db_host, user=db_user, password=db_pass, dbname=db_name, port=db_port
    )

def drop_table(table_name):
    print(f"Dropping table {table_name}...")
    try:
        conn = get_db_connection()
        conn.autocommit = True
        with conn.cursor() as cur:
            cur.execute(f"DROP TABLE IF EXISTS {table_name}")
        print(f"✅ Dropped {table_name}")
        conn.close()
    except Exception as e:
        print(f"❌ Error dropping {table_name}: {e}")

if __name__ == "__main__":
    drop_table("ifind_int_daily_basic")