main.py

import os
import time
import requests
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

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

# Target website configuration
# Replace with your actual DataDome captcha URL (must contain t=fe parameter)
CAPTCHA_URL = "https://geo.captcha-delivery.com/captcha/?initialCid=yourCid&t=fe&..."

# Your proxy (required for DataDome)
# Format: host:port:user:pass
PROXY = os.getenv("PROXY", "host:port:user:pass")

# User agent must match between API request and your actual browser/request
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():
    """Create a DataDome Slider solving task."""
    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"Failed to create task: {result.get('errorDescription')}")

    return result.get("taskId")


def get_task_result(task_id):
    """Poll for the task result until it's ready."""
    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"Failed to get result: {result.get('errorDescription')}")

        status = result.get("status")

        if status == "ready":
            return result.get("solution")
        elif status == "failed":
            raise Exception("Task failed")
        elif status == "processing":
            print("Task is still processing, waiting...")
            time.sleep(2)
        else:
            raise Exception(f"Unknown status: {status}")


def main():
    if not CAPSOLVER_API_KEY:
        print("Error: CAPSOLVER_API_KEY not found in .env file")
        print("Please create a .env file with your API key:")
        print("CAPSOLVER_API_KEY=your_api_key_here")
        return

    print("Creating DataDome Slider solving task...")
    print(f"Using proxy: {PROXY}")
    print(f"Using user agent: {USER_AGENT}")

    task_id = create_task()
    print(f"Task created successfully! Task ID: {task_id}")

    print("Waiting for solution (may take 1-20 seconds)...")
    solution = get_task_result(task_id)

    print("\nSolution received!")
    print(f"DataDome Cookie: {solution.get('cookie')}")

    # You can now use this cookie in your requests to bypass DataDome
    # Example:
    # cookies = {"datadome": solution.get("cookie")}
    # response = requests.get("https://target-site.com", cookies=cookies, headers={"User-Agent": USER_AGENT})


if __name__ == "__main__":
    main()