import mysql.connector, os
from dotenv import load_dotenv
load_dotenv()


def get_db_connection():
    try:
        db_host = os.getenv("DB_HOST")
        db_user = os.getenv("DB_USER")
        db_pass = os.getenv("DB_PASS")
        db_name = os.getenv("DB_NAME")
        
        db = mysql.connector.connect(host=db_host, user=db_user, password=db_pass, database=db_name)
    except mysql.connector.Error as err:
        print(f"❌ Failed to connect to DB: {err}")
        return  # don’t continue if DB failed
    return db

def preload_source_urls(cursor):
    cursor.execute("SELECT source_url FROM videos")
    return set(row[0] for row in cursor.fetchall())