main.py

import os
import time
import requests
from dotenv import load_dotenv

# Cargar variables de entorno desde el archivo .env
load_dotenv()

# Configuracion de la API de CapSolver
CAPSOLVER_API_KEY = os.getenv("CAPSOLVER_API_KEY")
CREATE_TASK_URL = "https://api.capsolver.com/createTask"
GET_RESULT_URL = "https://api.capsolver.com/getTaskResult"

# Configuracion del sitio web objetivo
# Reemplaza con tu URL real del captcha de DataDome (debe contener el parametro t=fe)
CAPTCHA_URL = "https://geo.captcha-delivery.com/captcha/?initialCid=yourCid&t=fe&..."

# Tu proxy (requerido para DataDome)
# Formato: host:puerto:usuario:contraseña
PROXY = os.getenv("PROXY", "host:puerto:usuario:contraseña")

# El user agent debe coincidir entre la peticion API y tu navegador/peticion real
USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/142.0.0.0 Safari/537.36"


def create_task():
    """Crear una tarea de resolucion de DataDome Slider."""
    payload = {
        "clientKey": CAPSOLVER_API_KEY,
        "task": {
            "type": "DatadomeSliderTask",
            "captchaUrl": CAPTCHA_URL,
            "proxy": PROXY,
            "userAgent": USER_AGENT
        }
    }

    response = requests.post(CREATE_TASK_URL, json=payload)
    result = response.json()

    if result.get("errorId") != 0:
        raise Exception(f"Error al crear la tarea: {result.get('errorDescription')}")

    return result.get("taskId")


def get_task_result(task_id):
    """Consultar el resultado de la tarea hasta que este listo."""
    payload = {
        "clientKey": CAPSOLVER_API_KEY,
        "taskId": task_id
    }

    while True:
        response = requests.post(GET_RESULT_URL, json=payload)
        result = response.json()

        if result.get("errorId") != 0:
            raise Exception(f"Error al obtener el resultado: {result.get('errorDescription')}")

        status = result.get("status")

        if status == "ready":
            return result.get("solution")
        elif status == "failed":
            raise Exception("La tarea fallo")
        elif status == "processing":
            print("La tarea aun se esta procesando, esperando...")
            time.sleep(2)
        else:
            raise Exception(f"Estado desconocido: {status}")


def main():
    if not CAPSOLVER_API_KEY:
        print("Error: CAPSOLVER_API_KEY no encontrada en el archivo .env")
        print("Por favor, crea un archivo .env con tu clave API:")
        print("CAPSOLVER_API_KEY=tu_clave_api_aqui")
        return

    print("Creando tarea de resolucion de DataDome Slider...")
    print(f"Usando proxy: {PROXY}")
    print(f"Usando user agent: {USER_AGENT}")

    task_id = create_task()
    print(f"Tarea creada exitosamente! ID de tarea: {task_id}")

    print("Esperando la solucion (puede tomar 1-20 segundos)...")
    solution = get_task_result(task_id)

    print("\nSolucion recibida!")
    print(f"Cookie de DataDome: {solution.get('cookie')}")

    # Ahora puedes usar esta cookie en tus peticiones para evadir DataDome
    # Ejemplo:
    # cookies = {"datadome": solution.get("cookie")}
    # response = requests.get("https://sitio-objetivo.com", cookies=cookies, headers={"User-Agent": USER_AGENT})


if __name__ == "__main__":
    main()