refactor: module imports

ChargeTicket.py

@@ -1,106 +1,127 @@
+from datetime import datetime, timedelta
+
 # 倍票相关 API 的实现
 import rapidjson as json
 import pytz
-from datetime import datetime, timedelta
+from loguru import logger
 
-from Config import *
 from API_TitleServer import apiSDGB
 from HelperGetUserThing import implGetUser_
 
-from loguru import logger
+from Config import (
+    clientId,
+    placeId,
+    regionId,
+)
+from MyConfig import testUid2
+
 from HelperLogInOut import apiLogin, apiLogout, generateTimestamp
 from HelperFullPlay import implFullPlayAction, generateMusicData
 from HelperGetUserThing import implGetUser_
 
-def implWipeTickets(userId: int, currentLoginTimestamp:int, currentLoginResult) -> str:
-    '''清空用户所有票的 API 请求器，返回 Json String。'''
+
+def implWipeTickets(userId: int, currentLoginTimestamp: int, currentLoginResult) -> str:
+    """清空用户所有票的 API 请求器，返回 Json String。"""
     # 先得到当前用户的 Charge 数据
     currentUserCharge = implGetUser_("Charge", userId)
     # 取得 List
-    currentUserChargeList = currentUserCharge['userChargeList']
+    currentUserChargeList = currentUserCharge["userChargeList"]
     # 所有 stock 都置为 0
     for charge in currentUserChargeList:
-        charge['stock'] = 0
+        charge["stock"] = 0
 
     musicData = generateMusicData()
     userAllPatches = {
-    "upsertUserAll": {
-        "userChargeList": currentUserChargeList,
-        "userMusicDetailList": [musicData],
-        "isNewMusicDetailList": "1"  #1避免覆盖
-    }}
+        "upsertUserAll": {
+            "userChargeList": currentUserChargeList,
+            "userMusicDetailList": [musicData],
+            "isNewMusicDetailList": "1",  # 1避免覆盖
+        }
+    }
 
-    result = implFullPlayAction(userId, currentLoginTimestamp, currentLoginResult, musicData, userAllPatches)
+    result = implFullPlayAction(
+        userId, currentLoginTimestamp, currentLoginResult, musicData, userAllPatches
+    )
     return result
 
-def apiQueryTicket(userId:int) -> str:
-    '''查询已有票的 API 请求器，返回 Json String。'''
+
+def apiQueryTicket(userId: int) -> str:
+    """查询已有票的 API 请求器，返回 Json String。"""
     # 构建 Payload
-    data = json.dumps({
-        "userId": userId
-    })
+    data = json.dumps({"userId": userId})
     # 发送请求
     userdata_result = apiSDGB(data, "GetUserChargeApi", userId)
     # 返回响应
     return userdata_result
 
-def apiBuyTicket(userId:int, ticketType:int, price:int, playerRating:int, playCount:int) -> str:
-    '''倍票购买 API 的请求器'''
 
-    nowTime = datetime.now(pytz.timezone('Asia/Shanghai'))
+def apiBuyTicket(
+    userId: int, ticketType: int, price: int, playerRating: int, playCount: int
+) -> str:
+    """倍票购买 API 的请求器"""
+
+    nowTime = datetime.now(pytz.timezone("Asia/Shanghai"))
 
     # 构造请求数据 Payload
-    data = json.dumps({
-        "userId": userId,
-        "userChargelog": {
-            "chargeId": ticketType,
-            "price": price,
-            "purchaseDate": nowTime.strftime("%Y-%m-%d %H:%M:%S.0"),
-            "playCount": playCount,
-            "playerRating": playerRating,
-            "placeId": placeId,
-            "regionId": regionId,
-            "clientId": clientId
-        },
-        "userCharge": {
-            "chargeId": ticketType,
-            "stock": 1,
-            "purchaseDate": nowTime.strftime("%Y-%m-%d %H:%M:%S.0"),
-            "validDate": (nowTime + timedelta(days=90)).replace(hour=4, minute=0, second=0).strftime("%Y-%m-%d %H:%M:%S")
+    data = json.dumps(
+        {
+            "userId": userId,
+            "userChargelog": {
+                "chargeId": ticketType,
+                "price": price,
+                "purchaseDate": nowTime.strftime("%Y-%m-%d %H:%M:%S.0"),
+                "playCount": playCount,
+                "playerRating": playerRating,
+                "placeId": placeId,
+                "regionId": regionId,
+                "clientId": clientId,
+            },
+            "userCharge": {
+                "chargeId": ticketType,
+                "stock": 1,
+                "purchaseDate": nowTime.strftime("%Y-%m-%d %H:%M:%S.0"),
+                "validDate": (nowTime + timedelta(days=90))
+                .replace(hour=4, minute=0, second=0)
+                .strftime("%Y-%m-%d %H:%M:%S"),
+            },
         }
-    })
+    )
     # 发送请求，返回最终得到的 Json String 回执
     return apiSDGB(data, "UpsertUserChargelogApi", userId)
 
-def implBuyTicket(userId:int, ticketType:int):
-    '''
+
+def implBuyTicket(userId: int, ticketType: int):
+    """
     购买倍票 API 的参考实现。
     需要事先登录.
     返回服务器响应的 Json string。
-    '''
+    """
     # 先使用 GetUserData API 请求器，取得 rating 和 pc 数
     currentUserData = implGetUser_("Data", userId)
     if currentUserData:
-        playerRating = currentUserData['userData']['playerRating'] 
-        playCount = currentUserData['userData'].get('playCount', 0)
+        playerRating = currentUserData["userData"]["playerRating"]
+        playCount = currentUserData["userData"].get("playCount", 0)
     else:
         return False
     # 正式买票
-    getTicketResponseStr = apiBuyTicket(userId, ticketType, ticketType-1, playerRating, playCount)
+    getTicketResponseStr = apiBuyTicket(
+        userId, ticketType, ticketType - 1, playerRating, playCount
+    )
     # 返回结果
     return getTicketResponseStr
 
+
 if __name__ == "__main__":
     userId = testUid2
     currentLoginTimestamp = generateTimestamp()
     loginResult = apiLogin(currentLoginTimestamp, userId)
 
-    if loginResult['returnCode'] != 1:
+    if loginResult["returnCode"] != 1:
         logger.info("登录失败")
         exit()
     try:
         logger.info(implBuyTicket(userId, 2))  # 购买倍票
-        #logger.info(apiQueryTicket(userId))
+        # logger.info(apiQueryTicket(userId))
     finally:
         logger.info(apiLogout(currentLoginTimestamp, userId))
-        #logger.warning("Error")
+        # logger.warning("Error")