https://www.apexrouter.top/api/data/self

"""
================================================================================
知能API 模型用量查询工具
================================================================================

功能说明:
    查询指定时间范围内的模型调用消耗数据，并以表格形式展示。

使用方法:
    1. 将 SYSTEM_TOKEN 替换为你的系统令牌
    2. 将 USER_ID 替换为你的用户 ID（在个人资料中获取）
    3. 设置查询时间范围（可选）:
       - start_time_text: 开始时间，格式 "2025-12-09 11:00"，留空则默认今天零点
       - end_time_text: 结束时间，格式 "2025-12-09 15:00"，留空则默认当前时间
    4. 运行脚本: python test.py

================================================================================
"""

import requests
from datetime import datetime

# ============================================================================
# 配置信息
# ============================================================================

# API 平台根地址
BASE_URL = "https://www.apexrouter.top"

# 认证信息 - 请替换为你自己的信息
SYSTEM_TOKEN = "YOUR_SYSTEM_TOKEN"  # 系统令牌，用于身份验证
USER_ID = "YOUR_USER_ID"                                  # 用户 ID，在「个人资料」中获取

# ----------------------------------------------------------------------------
# 查询时间范围配置（可选）
# ----------------------------------------------------------------------------
# 格式: "YYYY-MM-DD HH:MM"，例如 "2025-12-09 11:00"
# 留空 "" 则使用默认值:
#   - start_time_text 为空: 默认为今天 00:00
#   - end_time_text 为空: 默认为当前时间
# ----------------------------------------------------------------------------
start_time_text = "2026-05-09 11:00"  # 开始时间，例如: "2025-12-09 11:00"
end_time_text = "2026-06-08 11:00"    # 结束时间，例如: "2025-12-09 15:00"


# ============================================================================
# 核心函数
# ============================================================================

def parse_time(time_text: str) -> int:
    """
    将时间字符串解析为时间戳

    Args:
        time_text (str): 时间字符串，格式 "YYYY-MM-DD HH:MM"

    Returns:
        int: Unix 时间戳（秒级）
    """
    return int(datetime.strptime(time_text, "%Y-%m-%d %H:%M").timestamp())


def get_model_usage(start_timestamp: int, end_timestamp: int, time_granularity: str = "day"):
    """
    获取模型消耗分布数据

    通过调用 知能API 接口，查询指定时间范围内的模型调用记录。

    Args:
        start_timestamp (int): 查询开始时间戳（秒级）
        end_timestamp (int): 查询结束时间戳（秒级）
        time_granularity (str): 时间粒度，可选值:
                                - "day": 按天统计（默认）
                                - "hour": 按小时统计

    Returns:
        list: 包含模型使用数据的字典列表，每个字典包含:
              - model_name: 模型名称
              - quota: 消耗额度
              - created_time_text: 格式化后的日期时间
    """
    # 构建请求头
    headers = {
        "Accept": "application/json",
        "Authorization": f"{SYSTEM_TOKEN}",  # Token 认证
        "Dmx-Api-User": USER_ID,                    # 用户标识
    }

    # 构建查询参数
    params = {
        "start_timestamp": start_timestamp,
        "end_timestamp": end_timestamp,
        "default_time": time_granularity
    }

    # 发送 GET 请求
    response = requests.get(f"{BASE_URL}/api/data/self", headers=headers, params=params)

    # 检查 HTTP 状态码
    if response.status_code != 200:
        print(f"请求失败: {response.status_code}")
        return []

    # 解析 JSON 响应
    result = response.json()

    # 检查业务状态
    if not result.get("success"):
        print(f"接口返回错误: {result.get('message')}")
        return []

    # 提取并格式化数据
    data = []
    for item in result.get("data", []):
        data.append({
            "model_name": item.get("model_name"),          # 模型名称
            "quota": item.get("quota"),                    # 消耗额度（原始值）
            "created_time_text": datetime.fromtimestamp(  # 格式化时间戳
                item.get("created_at")
            ).strftime("%Y-%m-%d %H:%M")
        })

    return data


def get_display_width(text: str) -> int:
    """
    计算字符串的显示宽度（中文字符占2个宽度）

    Args:
        text (str): 输入字符串

    Returns:
        int: 显示宽度
    """
    width = 0
    for char in text:
        # 中文字符及全角字符占2个宽度
        if '\u4e00' <= char <= '\u9fff' or '\u3000' <= char <= '\u303f' or '\uff00' <= char <= '\uffef':
            width += 2
        else:
            width += 1
    return width


def pad_to_width(text: str, width: int) -> str:
    """
    将字符串填充到指定的显示宽度

    Args:
        text (str): 输入字符串
        width (int): 目标显示宽度

    Returns:
        str: 填充后的字符串
    """
    current_width = get_display_width(text)
    padding = width - current_width
    return text + ' ' * max(0, padding)


def print_usage(data: list):
    """
    格式化输出模型使用数据

    以表格形式打印查询结果，额度值会按 500000 原始额度 = 1 元 转换为 CNY。

    Args:
        data (list): get_model_usage() 返回的数据列表
    """
    if not data:
        print("没有数据")
        return

    # 定义列宽
    col_model = 40   # 模型名称列宽
    col_quota = 10   # 额度列宽
    col_date = 18    # 日期列宽

    # 打印表头
    header_model = pad_to_width("模型名称", col_model)
    header_quota = "额度".rjust(col_quota)
    header_date = pad_to_width("日期", col_date)
    print(f"{header_model} {header_quota} {header_date}")
    print("-" * (col_model + col_quota + col_date + 2))

    # 打印数据行（额度除以 500000 转换单位）
    for item in data:
        model_name = pad_to_width(item['model_name'], col_model)
        consumed_quota_cny = f"{item['quota'] / 500000:>12.4f}"  # 500000 原始额度 = 1 元
        created_time_text = pad_to_width(item['created_time_text'], col_date)
        print(f"{model_name} {consumed_quota_cny} {created_time_text}")


# ============================================================================
# 主程序入口
# ============================================================================

if __name__ == "__main__":
    import time

    # ------------------------------------------------------------------------
    # 设置查询时间范围
    # ------------------------------------------------------------------------

    # 解析开始时间：如果配置了 start_time_text 则使用，否则默认今天零点
    if start_time_text:
        start_timestamp = parse_time(start_time_text)
    else:
        start_datetime = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
        start_timestamp = int(start_datetime.timestamp())

    # 解析结束时间：如果配置了 end_time_text 则使用，否则默认当前时间
    if end_time_text:
        end_timestamp = parse_time(end_time_text)
    else:
        end_timestamp = int(time.time())

    # ------------------------------------------------------------------------
    # 执行查询并输出结果
    # ------------------------------------------------------------------------

    print(f"查询时间范围: {datetime.fromtimestamp(start_timestamp)} 至 {datetime.fromtimestamp(end_timestamp)}")
    print()

    # 调用 API 获取数据
    data = get_model_usage(start_timestamp, end_timestamp)

    # 格式化输出结果
    print_usage(data)

查询时间范围: 2026-05-09 11:00:00 至 2026-06-08 11:00:00

模型名称                                         额度 日期
----------------------------------------------------------------------
gpt-5.5                                        0.1783 2026-05-09 20:00
gpt-5.5-pro-ssvip                              1.3994 2026-05-09 20:00
claude-opus-4-7-cc                             3.6593 2026-05-09 20:00
doubao-seed-2-0-pro-260215                     0.0174 2026-05-09 21:00
doubao-seed-2-0-pro-260215                     0.0066 2026-05-09 20:00
claude-opus-4-7-cc                             2.2162 2026-05-09 21:00
gpt-5.5                                        0.3094 2026-05-12 03:00
claude-haiku-4-5-20251001-cc                   0.0010 2026-05-12 21:00
claude-opus-4-7-cc                             0.3121 2026-05-12 21:00
gpt-5.5                                        0.1273 2026-05-15 21:00