Update app.py

app.py

@@ -1,35 +1,69 @@
 import os
 import time
 import logging
-from threading import Thread
+import requests
+from apscheduler.schedulers.background import BackgroundScheduler
 
-# 配置日志记录器
+# 配置日志记录
 logging.basicConfig(level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(message)s')
 
-def load_keys():
+API_ENDPOINT = "https://api.siliconflow.cn/v1/usage/status"
+
+def get_credit_usage(key):
+    """
+    获取单个 key 的额度使用情况。
+
+    Args:
+        key: 要查询的 API key。
+
+    Returns:
+        包含额度信息的字典，如果出错则返回 None。
+    """
+    headers = {
+        "Authorization": f"Bearer {key}"
+    }
+    try:
+        response = requests.get(API_ENDPOINT, headers=headers)
+        response.raise_for_status()  # 检查请求是否成功
+        data = response.json()
+        return data
+    except requests.exceptions.RequestException as e:
+        logging.error(f"请求 API 时出错，key：{key}, 错误信息：{e}")
+    except ValueError as e:
+        logging.error(f"解析 JSON 响应时出错，key：{key}, 错误信息：{e}")
+    return None
+
+def load_keys_and_get_usage():
     """
-    从环境变量中加载 API 密钥，并在日志中输出。
+    从环境变量中加载 keys，并获取每个 key 的额度使用情况，然后记录到日志中。
     """
     keys_str = os.environ.get("KEYS")
     if keys_str:
         keys = [key.strip() for key in keys_str.split(',')]
-        logging.info("已加载的 API 密钥：")
+        logging.info(f"加载的 keys：{keys}")
         for key in keys:
-            logging.info(key)
+            usage_data = get_credit_usage(key)
+            if usage_data:
+                logging.info(f"Key: {key}, 额度信息: {usage_data}")
+            else:
+                logging.warning(f"获取 key：{key} 的额度信息失败。")
     else:
-        logging.warning("未找到名为 KEYS 的环境变量")
+        logging.warning("环境变量 KEYS 未设置。")
 
-def scheduled_task():
-    """
-    每小时执行一次的定时任务。
-    """
-    while True:
-        load_keys()
-        # 休眠一小时
-        time.sleep(3600)
-
-# 启动定时任务
-thread = Thread(target=scheduled_task)
-thread.daemon = True
-thread.start()
+# 创建一个后台调度器
+scheduler = BackgroundScheduler()
+
+# 添加定时任务，每小时执行一次 load_keys_and_get_usage 函数
+scheduler.add_job(load_keys_and_get_usage, 'interval', hours=1)
+
+# 启动调度器
+scheduler.start()
+
+# 为了保持程序运行，添加一个循环
+if __name__ == '__main__':
+    try:
+        while True:
+            time.sleep(120)
+    except (KeyboardInterrupt, SystemExit):
+        scheduler.shutdown()