import requests


class Demo:
    def __init__(self):
        self.api_key = "bX7pK9mR2nL4qF6s"  # 用户的 API Key
        self.api_secret = "eJ8wT3hY5cA2vN9uZ1xD7fG4"  # 用户的 API Secret
        self.host = "http://10.0.220.110"  # 接口主地址
        self.headers = {"Content-Type": "application/json", "Authorization": ""}

    # 获取认证token
    def getAuthToken(self):
        # print("开始登录")
        url = f"{self.host}/api/openapi/authenticate"
        data = {"apiKey": self.api_key, "apiSecret": self.api_secret}
        try:
            response = requests.post(url, json=data)
            response.raise_for_status()
            result = response.json()

            if result["success"]:
                self.headers["Authorization"] = f"Bearer {result['data']['token']}"
                print("登录vlm成功")
                return result["data"]["token"]
            else:
                raise Exception(f"登录失败: {result['message']}")
        except requests.RequestException as e:
            print(f"登录请求失败: {e}")
            raise