import requests

class ZenithTweakingAPI:
    def __init__(self, api_key):
        self.api_key = api_key
        self.base_url = "https://api.zenithtweaking.com"

    def optimize_system(self, system_id):
        endpoint = f"{self.base_url}/optimize"
        headers = {"Authorization": f"Bearer {self.api_key}"}
        params = {"system_id": system_id}

        response = requests.post(endpoint, headers=headers, json=params)

        if response.status_code == 200:
            return response.json()
        else:
            print(f"Error: {response.status_code}, {response.text}")

    def get_optimization_status(self, system_id):
        endpoint = f"{self.base_url}/status"
        headers = {"Authorization": f"Bearer {self.api_key}"}
        params = {"system_id": system_id}

        response = requests.get(endpoint, headers=headers, params=params)

        if response.status_code == 200:
            return response.json()
        else:
            print(f"Error: {response.status_code}, {response.text}")