签名查询模拟盘交易账户余额

import base64, datetime, hashlib, hmac, json, os, urllib.request, urllib.error
# 只读模拟盘示例，环境与密钥不能混用。
key = os.environ["OKX_DEMO_API_KEY"]
secret = os.environ["OKX_DEMO_SECRET_KEY"]
passphrase = os.environ["OKX_DEMO_PASSPHRASE"]
host = os.environ.get("OKX_API_HOST", "https://www.okx.com")
allowed_hosts = {"https://www.okx.com", "https://us.okx.com"}
if host not in allowed_hosts:
    raise SystemExit("请按所属地区核对文档中的官方 API 域名")
# 获取服务器时间，降低本机时钟漂移造成的错误。
try:
    with urllib.request.urlopen(host + "/api/v5/public/time", timeout=20) as response:
        clock = json.load(response)
    if clock.get("code") != "0":
        raise ValueError("服务器时间查询失败")
    server_ms = int(clock["data"][0]["ts"])
    timestamp = datetime.datetime.fromtimestamp(server_ms / 1000, datetime.timezone.utc).isoformat(timespec="milliseconds").replace("+00:00", "Z")
    path = "/api/v5/account/balance?ccy=USDT"
    message = timestamp + "GET" + path  # GET 没有 body，query 必须包含在 path 中。
    signature = base64.b64encode(hmac.new(secret.encode(), message.encode(), hashlib.sha256).digest()).decode()
    req = urllib.request.Request(host + path, headers={
        "OK-ACCESS-KEY": key,
        "OK-ACCESS-SIGN": signature,
        "OK-ACCESS-TIMESTAMP": timestamp,
        "OK-ACCESS-PASSPHRASE": passphrase,
        "x-simulated-trading": "1",
        "Accept": "application/json",
    })
    with urllib.request.urlopen(req, timeout=20) as response:
        result = json.load(response)
except urllib.error.HTTPError as exc:
    raise SystemExit(f"HTTP {exc.code}: {exc.read().decode()}")
if result.get("code") != "0":
    raise SystemExit(json.dumps(result, ensure_ascii=False))
print(json.dumps(result, ensure_ascii=False, indent=2))