import time
import random

def api_request_with_retry(api_func, max_retries=3):
    """API request example with retry"""
    for attempt in range(max_retries + 1):
        try:
            response = api_func()
            if response.status_code == 429:
                if attempt < max_retries:
                    # Exponential retraction + random jitter
                    delay = (2 ** attempt) + random.uniform(0, 1)
                    time.sleep(delay)
                    continue
                else:
                    raise Exception("Request frequency limit, maximum retry count has been reached")
            return response
        except Exception as e:
            if attempt == max_retries:
                raise e
            time.sleep(1)

import hmac
import hashlib
import time
import json
import requests
from typing import Dict, Any, Optional

class ArmCloudSignatureV2:
    def __init__(self, access_key_id: str, secret_key: str, base_url: str = "https://api.xiaosuanyun.com"):
        self.access_key_id = access_key_id
        self.secret_key = secret_key
        self.base_url = base_url  # Mainland China: https://api.xiaosuanyun.com; overseas: https://openapi-hk.armcloud.net

    def _calculate_signature(self, timestamp: str, path: str, body: str = "") -> str:
        """Calculate HMAC-SHA256 signature"""
        string_to_sign = timestamp + path + (body if body else "")
        signature = hmac.new(
            self.secret_key.encode('utf-8'),
            string_to_sign.encode('utf-8'),
            hashlib.sha256
        ).hexdigest()
        return signature

    def sign_get_request(self, path: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, str]:
        """Generates a signature header for the GET request"""
        timestamp = str(int(time.time() * 1000))
        query_string = ""

        if params:
            query_string = "&".join([f"{k}={v}" for k, v in params.items()])
            body = query_string
        else:
            body = ""

        signature = self._calculate_signature(timestamp, path, body)

        return {
            "authver": "2.0",
            "x-ak": self.access_key_id,
            "x-timestamp": timestamp,
            "x-sign": signature
        }

    def sign_post_request(self, path: str, data: Dict[str, Any]) -> Dict[str, str]:
        """Generates signature headers for POST requests"""
        timestamp = str(int(time.time() * 1000))
        request_body = json.dumps(data, ensure_ascii=False, separators=(',', ':'))
        signature = self._calculate_signature(timestamp, path, request_body)

        return {
            "authver": "2.0",
            "x-ak": self.access_key_id,
            "x-timestamp": timestamp,
            "x-sign": signature,
            "Content-Type": "application/json"
        }

    def get(self, path: str, params: Optional[Dict[str, Any]] = None) -> requests.Response:
        """Initiate a GET request"""
        headers = self.sign_get_request(path, params)
        url = self.base_url + path
        return requests.get(url, params=params, headers=headers)

    def post(self, path: str, data: Dict[str, Any]) -> requests.Response:
        """Initiate a POST request"""
        headers = self.sign_post_request(path, data)
        url = self.base_url + path
        request_body = json.dumps(data, ensure_ascii=False, separators=(',', ':'))
        return requests.post(url, data=request_body, headers=headers)

# Use examples
if __name__ == "__main__":
    # Initialize the client - select the domain name based on the region
    # Domestic users
    client = ArmCloudSignatureV2("LTAI4FzK8888888888888", "your_secret_key", "https://api.xiaosuanyun.com")
    # Overseas users
    # client = ArmCloudSignatureV2("LTAI4FzK8888888888888", "your_secret_key", "https://openapi-hk.armcloud.net")

    # POST Request Example - Board List Query (For interface paths, please refer to the OpenAPI documentation)
    device_list_data = {
        "page": 1,
        "rows": 10,
        "padCodes": ["AC21020010391"],
        "vmStatus": "1",
        "deviceStatus": "0"
    }

    try:
        # Note: The path shown here is for example only. Please refer to the OpenAPI documentation for the actual path
        response = client.post("/openapi/open/device/list", device_list_data)
        print(f"Board List Query Response: {response.status_code}")
        print(f"Response Body: {response.text}")
    except Exception as e:
        print(f"Board list query Error: {e}")

    # POST Request Example - Creating a Cloud Instance (For interface paths and parameters, please refer to the OpenAPI documentation)
    create_instance_data = {
        "clusterCode": "001",
        "specificationCode": "m2-3",
        "imageId": "img-25021828327",
        "screenLayoutCode": "realdevice_1440x3120x560",
        "number": 2,
        "dns": "8.8.8.8",
        "storageSize": 16,
        "realPhoneTemplateId": 36
    }

    try:
        # Note: The path shown here is for example only. Please refer to the OpenAPI documentation for the actual path
        response = client.post("/openapi/open/pad/net/storage/res/create", create_instance_data)
        print(f"Create Cloud Instance Response: {response.status_code}")
        print(f"Response Body: {response.text}")
    except Exception as e:
        print(f"Create Cloud Instance Error: {e}")