Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import time
 import logging
 import requests
+from datetime import datetime, timedelta
 from apscheduler.schedulers.background import BackgroundScheduler
 from flask import Flask, request, jsonify, Response, stream_with_context
 
@@ -326,27 +327,79 @@ def list_models():
         "free_models": free_models
     })
 
-@app.route('/handsome/v1/dashboard/billing', methods=['GET'])
-def billing_info():
+def get_billing_info():
     """
-    处理 /handsome/v1/dashboard/billing 路由的请求，返回所有余额大于0的key的总和。
+    获取所有KEY的额度信息。
     """
-    if not check_authorization(request):
-        return jsonify({"error": "Unauthorized"}), 401
-
     total_balance = 0
-    # 遍历 valid_keys_global 和 unverified_keys_global 获取总余额
     for key in valid_keys_global + unverified_keys_global:
         credit_summary = get_credit_summary(key)
-        if credit_summary and credit_summary.get("total_balance", 0) > 0:
-            total_balance += credit_summary["total_balance"]
-    
+        if credit_summary:
+            total_balance += credit_summary.get("total_balance", 0)
+    return total_balance
+
+@app.route('/handsome/v1/dashboard/billing/usage', methods=['GET'])
+def billing_usage():
+    """
+    处理 /handsome/v1/dashboard/billing/usage 路由的请求，返回用量信息（模拟）。
+    """
+    if not check_authorization(request):
+        return jsonify({"error": "Unauthorized"}), 401
+
+    # 获取当前时间和30天前的时间
+    end_date = datetime.now()
+    start_date = end_date - timedelta(days=30)
+
+    # 构造用量数据（模拟）
+    daily_usage = []
+    current_date = start_date
+    while current_date <= end_date:
+        # 假设每天的用量是总额度的 1/1000
+        daily_usage.append({
+            "timestamp": int(current_date.timestamp()),
+            "daily_usage": get_billing_info() / 1000
+        })
+        current_date += timedelta(days=1)
+
+    return jsonify({
+        "object": "list",
+        "data": daily_usage,
+        "total_usage": sum(day["daily_usage"] for day in daily_usage)
+    })
+
+@app.route('/handsome/v1/dashboard/billing/subscription', methods=['GET'])
+def billing_subscription():
+    """
+    处理 /handsome/v1/dashboard/billing/subscription 路由的请求，返回订阅信息。
+    """
+    if not check_authorization(request):
+        return jsonify({"error": "Unauthorized"}), 401
+
+    total_balance = get_billing_info()
+
     return jsonify({
-        "total_granted": total_balance,  # 使用 total_granted 字段
-        "total_usage": 0,  # 暂时设置为0，因为没有使用量数据
-        "total_available": total_balance,
+        "object": "billing_subscription",
         "has_payment_method": False,
-        "object": "billing_summary"
+        "canceled": False,
+        "canceled_at": None,
+        "delinquent": None,
+        "access_until": int(datetime(9999, 12, 31).timestamp()),  # 设置一个较远的未来时间
+        "soft_limit": 0,
+        "hard_limit": total_balance,
+        "system_hard_limit": total_balance,
+        "soft_limit_usd": 0,
+        "hard_limit_usd": total_balance,
+        "system_hard_limit_usd": total_balance,
+        "plan": {
+            "name": "SiliconFlow API",
+            "id": "siliconflow-api"
+        },
+        "account_name": "SiliconFlow User",
+        "po_number": None,
+        "billing_email": None,
+        "tax_ids": [],
+        "billing_address": None,
+        "business_address": None
     })
 
 if __name__ == '__main__':