import os
import pymysql
from dotenv import load_dotenv

# Carrega .env
load_dotenv()

# Função de teste
def test_query():
    try:
        connection = pymysql.connect(
            host=os.getenv("DB_HOST"),
            user=os.getenv("DB_USER"),
            password=os.getenv("DB_PASSWORD"),
            database=os.getenv("DB_NAME"),
            port=int(os.getenv("DB_PORT")),
            cursorclass=pymysql.cursors.DictCursor
        )

        with connection.cursor() as cursor:
            cursor.execute("SELECT sku FROM lumixdb.catalog_product_entity LIMIT 5;")
            results = cursor.fetchall()
            print("✅ Conexão OK. Resultados:")
            for row in results:
                print(row)

    except Exception as e:
        print("❌ Erro na conexão ou consulta:")
        print(e)

    finally:
        if 'connection' in locals():
            connection.close()

if __name__ == "__main__":
    test_query()