from Requests to HTTPX (Higher performance, I guess!)

API_TitleServer.py

@@ -3,16 +3,13 @@
 
 import zlib
 import hashlib
-import requests
+import httpx
 from loguru import logger
 import random
 import time
-
 from ctypes import c_int32
-
 from Crypto.Cipher import AES
 from Crypto.Util.Padding import unpad
-
 from Config import *
 
 # 舞萌DX 2024
@@ -20,10 +17,16 @@ AesKey = "n7bx6:@Fg_:2;5E89Phy7AyIcpxEQ:R@"
 AesIV = ";;KjR1C3hgB1ovXa"
 ObfuscateParam = "BEs2D5vW"
 
-class WahlapServerBoomedError(Exception):
+class SDGBApiError(Exception):
     pass
     
-class Request500Error(Exception):
+class SDGBMaxRetriesError(SDGBApiError):
+    pass
+
+class SDGBRequestError(SDGBApiError):
+    pass
+
+class SDGBResponseError(SDGBApiError):
     pass
 
 class AES_PKCS7(object):
@@ -56,93 +59,84 @@ class AES_PKCS7(object):
         padding_text = chr(padding) * padding
         return text + padding_text
 
-def SDGBApiHash(api):
+def getSDGBApiHash(api):
     return hashlib.md5((api+"MaimaiChn"+ObfuscateParam).encode()).hexdigest()
 
-def apiSDGB(data:str, useApi, agentExtraData, noLog=False):
+def apiSDGB(data:str, targetApi:str, userAgentExtraData:str, noLog:bool=False, timeout:int=5):
     """
     舞萌DX 2024 API 通讯用函数
     :param data: 请求数据
-    :param useApi: 使用的 API
-    :param agentExtraData: UA 附加信息，机台相关则为狗号（如A63E01E9564），用户相关则为 UID
+    :param targetApi: 使用的 API
+    :param userAgentExtraData: UA 附加信息，机台相关则为狗号（如A63E01E9564），用户相关则为 UID
     :param noLog: 是否不记录日志
     """
     maxRetries = 3
-
-    # 历史遗留代码有时候会传入 int，故先全部转 str
-    agentExtra = str(agentExtraData)
-
-    # 编码好请求，准备发送
+    agentExtra = str(userAgentExtraData)
     aes = AES_PKCS7(AesKey, AesIV)
-    data = data
-    data_enc = aes.encrypt(data)
-    data_def = zlib.compress(data_enc)
-    requests.packages.urllib3.disable_warnings()
+    reqData_encrypted = aes.encrypt(data)
+    reqData_deflated = zlib.compress(reqData_encrypted)
     endpoint = "https://maimai-gm.wahlap.com:42081/Maimai2Servlet/"
-
     if not noLog:
-        logger.debug("TitleServer Request Start: "+ str(useApi)+" , Data: "+str(data))
+        logger.debug(f"开始请求 {targetApi}，以 {data}")
 
     retries = 0
     while retries < maxRetries:
         try:
-            # 发送请求
-            responseRaw = requests.post(endpoint + SDGBApiHash(useApi), headers={
-                "User-Agent": f"{SDGBApiHash(useApi)}#{agentExtra}", 
-                "Content-Type": "application/json",
-                "Mai-Encoding": "1.40",
-                "Accept-Encoding": "",
-                "Charset": "UTF-8", 
-                "Content-Encoding": "deflate", 
-                "Expect": "100-continue"
-            }, data=data_def, verify=False)
-            logger.debug("TitleServer Request Sent.")
-            
-            logger.debug("TitleServer Response Code: " + str(responseRaw.status_code))
-            # 如果是 404 或 500，直接抛出异常，不再继续重试
-            match responseRaw.status_code:
-                case 200:
-                    logger.debug("Request 200 OK!")
-                case 404:
-                    logger.error(f"Request 404! ")
-                    raise NotImplementedError
-                case 500:
-                    logger.error(f"Request Failed! 500!!!! ")
-                    raise Request500Error
-                case _:
-                    logger.error(f"Request Failed! {responseRaw.status_code}")
-                    raise NotImplementedError
-            responseContent = responseRaw.content
-            # 尝试解压请求
+            response = httpx.post(
+                url=endpoint + getSDGBApiHash(targetApi),
+                headers={
+                    "User-Agent": f"{getSDGBApiHash(targetApi)}#{agentExtra}",
+                    "Content-Type": "application/json",
+                    "Mai-Encoding": "1.40",
+                    "Accept-Encoding": "",
+                    "Charset": "UTF-8",
+                    "Content-Encoding": "deflate",
+                    "Expect": "100-continue"
+                },
+                content=reqData_deflated,
+                verify=False,
+                timeout=timeout
+            )
+
+            logger.info(f"{targetApi} 请求结果: {response.status_code}")
+
+            if response.status_code == 200:
+                logger.debug("200 OK!")
+            else:
+                errorMessage = f"请求失败: {response.status_code}"
+                logger.error(errorMessage)
+                raise SDGBRequestError(errorMessage)
+
+            responseRAWContent = response.content
+
             try:
-                responseDecompressed = zlib.decompress(responseContent)
-                logger.debug("Successfully decompressed response.")
-            except zlib.error as e:
-                logger.warning(f"RAW Response: {responseContent}")
-                logger.warning(f"Wahlap Server Boomed! Will now retry.{e}")
-                retries += 1
-                time.sleep(4) # 休眠4秒后重试
-                continue
-            # 解压成功，解密请求并返回
+                responseDecompressed = zlib.decompress(responseRAWContent)
+                logger.debug("成功解压响应！")
+            except zlib.error:
+                logger.warning(f"无法解压，得到的原始响应: {responseRAWContent}")
+                raise SDGBResponseError("Decompression failed")
+
             resultResponse = unpad(aes.decrypt(responseDecompressed), 16).decode()
-            logger.info("TitleServer:" + useApi + " Response: " + str(responseRaw.status_code))
             if not noLog:
-                logger.debug("TitleServer Response: " + str(resultResponse))
+                logger.debug(f"响应: {resultResponse}")
             return resultResponse
         
-        # 除了 404 和 500 之外的错误重试
-        except Request500Error:
-            raise Request500Error("500，请求格式错误")
-        except NotImplementedError:
-            raise NotImplementedError("请求未知错误")
+        # 异常处理
+        except SDGBRequestError as e:
+            # 请求格式错误，不需要重试
+            raise SDGBRequestError("请求格式错误")
+        except SDGBResponseError as e:
+            # 响应解析错误，重试但是只一次
+            logger.warning(f"Will now retry. {e}")
+            retries += 2
+            time.sleep(2)
         except Exception as e:
-            logger.warning(f"Request Failed! Will now retry.. {e}")
+            # 其他错误，重试
+            logger.warning(f"Will now retry. {e}")
             retries += 1
             time.sleep(3)
-    else:
-        # 重试次数用尽，WahlapServerBoomedError
-        raise WahlapServerBoomedError("重试多次仍然不能成功请求")
-    
+
+    raise SDGBApiError("Multiple retries failed to make a successful request")
 
 def calcSpecialNumber():
     """使用 c_int32 实现的 SpecialNumber 算法"""
@@ -157,7 +151,6 @@ def calcSpecialNumber():
         num2 >>= 1
     return c_int32(result.value).value
 
-
 def calcSpecialNumber2():
     """实验性替代 SpecialNumber 算法"""
     max = 1037933