import requests
import time
import json

# API Configuration
API_KEY = "your_api_key_here"  # Replace with your API key
BASE_URL = "https://dev.capunlock.com/openapi"

def create_task():
    """Create task"""
    url = f"{BASE_URL}/task/create"
    headers = {
        "Content-Type": "application/json"
    }
    data = {
        "api_key": API_KEY,
        "type": "reCaptcha_v2",
        "website_key": "6Le-wvkSAAAAAPBMRTvw0Q4Muexq9bi0DJwx_mJ-",
        "website_url": "https://www.google.com/recaptcha/api2/demo"
    }
    
    response = requests.post(url, headers=headers, json=data)
    result = response.json()
    
    if result.get("code") == 0:
        print(f"Task created successfully, Task ID: {result['data']['task_id']}")
        return result['data']['task_id']
    else:
        print(f"Task creation failed: {result.get('message')}")
        return None

def get_task_result(task_id):
    """Get task result"""
    url = f"{BASE_URL}/task/query"
    params = {
        "api_key": API_KEY,
        "task_id": task_id
    }
    
    # Polling for results
    max_attempts = 30
    for attempt in range(max_attempts):
        response = requests.get(url, params=params)
        result = response.json()
        
        if result.get("code") == 0:
            if result['data']['status'] == 'completed':
                print(f"Task completed, Result: {result['data']['result']}")
                return result['data']['result']
            elif result['data']['status'] == 'failed':
                print(f"Task failed: {result['data'].get('error_message')}")
                return None
            else:
                print(f"Task in progress... (Attempt {attempt + 1}/{max_attempts})")
                time.sleep(2)
        else:
            print(f"Query failed: {result.get('message')}")
            return None
    
    print("Task timeout")
    return None

# Complete workflow example
if __name__ == "__main__":
    # Create task
    task_id = create_task()
    
    if task_id:
        # Get task result
        result = get_task_result(task_id)
        if result:
            print(f"Final result: {result}")