#系统参数地址 https://192.168.2.1/proxy/network/api/s/default/stat/device/

import json, yaml, requests
from datetime import datetime
from urllib3 import disable_warnings
disable_warnings()

SECRETS_FILE = "/config/secrets.yaml"
def get_secret(secret):
    try:
        with open(SECRETS_FILE, "r", encoding="utf8") as file:
            secrets = yaml.full_load(file)
            for key, value in secrets.items():
                if key == secret:
                    return value
    except FileNotFoundError:
        print("secrets.yaml not found")
        exit()

IP = get_secret("unifi_ip")
PORT = get_secret("unifi_port")
USER = get_secret("unifi_user")
PASS = get_secret("unifi_pass")
URL = f"https://{IP}:{PORT}"

login = requests.request("POST", f"{URL}/api/auth/login", \
    headers={"Content-Type": "application/json"}, \
    data=json.dumps({"username": USER, "password": PASS}), verify=False)
response = requests.request("GET", f"{URL}/proxy/network/api/s/default/stat/device/", \
    cookies=login.cookies, verify=False)
data = response.json()["data"][3]   #注意修改获得参数的序号

print(json.dumps({
    "cpu": data["system-stats"]["cpu"],
    "cpu_temp": round(data["temperatures"][1]["value"], 1),
    "mem": data["system-stats"]["mem"],
    "disk": round(data["storage"][1]["used"] / data["storage"][1]["size"] * 100, 1),
#    "internet": data["internet"],
    "internet": response.json()["data"][4]["internet"],
    "uptime": datetime.fromtimestamp(data["startup_timestamp"]).isoformat(),
    "version": data["displayable_version"]
}))