99 lines
4.0 KiB
Python
99 lines
4.0 KiB
Python
import time
|
||
from typing import Any, Dict, List, Optional
|
||
|
||
import httpx
|
||
from pydantic import BaseModel
|
||
from tenacity import retry, stop_after_attempt, wait_exponential
|
||
|
||
from .config import settings
|
||
|
||
|
||
class AuthResponse(BaseModel):
|
||
token: str
|
||
exp: Optional[int] = None # epoch seconds
|
||
|
||
|
||
class EipClient:
|
||
def __init__(self, base_url: str, username: str, password: str) -> None:
|
||
self.base_url = base_url.rstrip("/")
|
||
self.username = username
|
||
self.password = password
|
||
self._token: Optional[str] = None
|
||
self._token_exp: Optional[int] = None
|
||
self._client = httpx.Client(timeout=15.0)
|
||
|
||
def _is_token_valid(self) -> bool:
|
||
if not self._token or not self._token_exp:
|
||
return False
|
||
# refresh 60s earlier
|
||
return time.time() < (self._token_exp - 60)
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=0.5, min=0.5, max=2))
|
||
def authenticate(self) -> AuthResponse:
|
||
url = f"{self.base_url}/client/auth"
|
||
resp = self._client.post(url, json={"username": self.username, "password": self.password})
|
||
resp.raise_for_status()
|
||
data = resp.json()
|
||
# 尝试从响应中提取 exp；如果没有，设置 1 小时后过期
|
||
exp = data.get("exp") or int(time.time()) + 3600
|
||
token = data.get("token") or data.get("X-Token") or data.get("access_token")
|
||
if not token:
|
||
# 兼容后端直接返回字符串 token 的情况
|
||
if isinstance(data, str) and data:
|
||
token = data
|
||
else:
|
||
raise ValueError("Auth response missing token")
|
||
self._token = token
|
||
self._token_exp = exp
|
||
return AuthResponse(token=token, exp=exp)
|
||
|
||
def _get_headers(self) -> Dict[str, str]:
|
||
if not self._is_token_valid():
|
||
self.authenticate()
|
||
assert self._token
|
||
return {"X-Token": self._token, "Accept": "application/json"}
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=0.5, min=0.5, max=2))
|
||
def list_cities(self) -> List[Dict[str, Any]]:
|
||
url = f"{self.base_url}/edge/city"
|
||
resp = self._client.get(url, headers=self._get_headers())
|
||
resp.raise_for_status()
|
||
data = resp.json()
|
||
return data if isinstance(data, list) else data.get("data", [])
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=0.5, min=0.5, max=2))
|
||
def list_devices(self, geo: str, offset: int, num: int) -> List[Dict[str, Any]]:
|
||
url = f"{self.base_url}/edge/device"
|
||
payload = {"geo": geo, "offset": offset, "num": num}
|
||
resp = self._client.post(url, headers=self._get_headers(), json=payload)
|
||
resp.raise_for_status()
|
||
data = resp.json()
|
||
return data if isinstance(data, list) else data.get("data", [])
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=0.5, min=0.5, max=2))
|
||
def gateway_config_get(self, macaddr: str) -> Dict[str, Any]:
|
||
url = f"{self.base_url}/gateway/config/get"
|
||
resp = self._client.post(url, headers=self._get_headers(), json={"macaddr": macaddr})
|
||
resp.raise_for_status()
|
||
return resp.json()
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=0.5, min=0.5, max=2))
|
||
def gateway_config_set(self, macaddr: str, config: Dict[str, Any]) -> Dict[str, Any]:
|
||
url = f"{self.base_url}/gateway/config/set"
|
||
payload = {"macaddr": macaddr, "config": config}
|
||
resp = self._client.post(url, headers=self._get_headers(), json=payload)
|
||
resp.raise_for_status()
|
||
return resp.json()
|
||
|
||
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=0.5, min=0.5, max=2))
|
||
def gateway_status(self, macaddr: str) -> Dict[str, Any]:
|
||
url = f"{self.base_url}/gateway/status"
|
||
resp = self._client.post(url, headers=self._get_headers(), json={"macaddr": macaddr})
|
||
resp.raise_for_status()
|
||
return resp.json()
|
||
|
||
|
||
client_singleton = EipClient(settings.eip_base_url, settings.eip_username, settings.eip_password)
|
||
|
||
|