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 target site URL
SITE_URL = "https://example.com"


def create_task():
    """Create an AWS WAF solving task."""
    payload = {
        "clientKey": CAPSOLVER_API_KEY,
        "task": {
            "type": "AntiAwsWafTaskProxyLess",
            "websiteURL": SITE_URL,
            # Optional parameters (uncomment if needed):
            # "awsKey": "your_aws_key",
            # "awsIv": "your_aws_iv",
            # "awsContext": "your_aws_context",
            # "awsChallengeJS": "challenge_js_url"
        }
    }

    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 == "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 AWS WAF solving task...")
    print(f"Site URL: {SITE_URL}")

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

    print("Waiting for solution (5-30 seconds)...")
    solution = get_task_result(task_id)

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

    # Use this cookie in your requests to bypass AWS WAF
    # Example:
    # cookies = {"aws-waf-token": solution.get("cookie")}
    # response = requests.get(SITE_URL, cookies=cookies)


if __name__ == "__main__":
    main()