Update router_wallet.py

router_wallet.py

@@ -45,25 +45,49 @@ router = APIRouter()
 # ==========================================
 # 🚨 替换这里的支付宝初始化逻辑 🚨
 # ==========================================
+def format_pem_key(key_str, key_type="PRIVATE"):
+    """自动给'裸奔'的秘钥穿上符合 Python 规范的外套"""
+    # 如果用户已经带有 BEGIN 标签，说明格式没问题，直接返回
+    if "BEGIN" in key_str:
+        return key_str.replace("\\n", "\n").strip()
+        
+    # 去除所有的空格和换行符，拿到最纯净的字符串
+    clean_key = key_str.replace(" ", "").replace("\n", "").replace("\\n", "").replace("\r", "")
+    
+    # 按照国际标准，每 64 个字符强制切断换行
+    chunks = [clean_key[i:i+64] for i in range(0, len(clean_key), 64)]
+    formatted_body = "\n".join(chunks)
+    
+    # 穿上头尾外套
+    if key_type == "PRIVATE":
+        return f"-----BEGIN RSA PRIVATE KEY-----\n{formatted_body}\n-----END RSA PRIVATE KEY-----"
+    else:
+        return f"-----BEGIN PUBLIC KEY-----\n{formatted_body}\n-----END PUBLIC KEY-----"
+
 alipay_error_msg = "未知错误"
 try:
     from alipay import AliPay
     from alipay.utils import AliPayConfig
     
-    # 获取环境变量并清理可能存在的首尾空格
-    appid = os.environ.get("ALIPAY_APPID", "").strip()
-    priv_key = os.environ.get("ALIPAY_PRIVATE_KEY", "").replace("\\n", "\n").strip()
-    pub_key = os.environ.get("ALIPAY_PUBLIC_KEY", "").replace("\\n", "\n").strip()
+    # 1. 抓取原生态的环境变量
+    raw_appid = os.environ.get("ALIPAY_APPID", "").strip()
+    raw_priv_key = os.environ.get("ALIPAY_PRIVATE_KEY", "").strip()
+    raw_pub_key = os.environ.get("ALIPAY_PUBLIC_KEY", "").strip()
     
-    if not appid or not priv_key or not pub_key:
-        alipay_error_msg = f"缺少环境变量。当前读取到: APPID={bool(appid)}, PRIV_KEY={bool(priv_key)}, PUB_KEY={bool(pub_key)}"
+    if not raw_appid or not raw_priv_key or not raw_pub_key:
+        alipay_error_msg = f"缺少环境变量。当前读取到: APPID={bool(raw_appid)}, PRIV_KEY={bool(raw_priv_key)}, PUB_KEY={bool(raw_pub_key)}"
         alipay = None
     else:
+        # 2. 扔进我们的格式化引擎进行自动包装！
+        priv_key_formatted = format_pem_key(raw_priv_key, "PRIVATE")
+        pub_key_formatted = format_pem_key(raw_pub_key, "PUBLIC")
+        
+        # 3. 完美加载
         alipay = AliPay(
-            appid=appid,
+            appid=raw_appid,
             app_notify_url="https://zhiwei666-comfyui-ranking-api.hf.space/api/wallet/alipay_notify",
-            app_private_key_string=priv_key,
-            alipay_public_key_string=pub_key,
+            app_private_key_string=priv_key_formatted,
+            alipay_public_key_string=pub_key_formatted,
             sign_type="RSA2",
             debug=False,
             config=AliPayConfig(timeout=15)