chore: move payload scripts to action/

action/UpsertMusic.py

@@ -0,0 +1,83 @@
+import json
+import asyncio
+import logging
+
+import httpx
+
+from sdgb import MaimaiClient
+from sdgb.settings import userId, musicData
+from sdgb.payload import (
+    requestData_UserLogin,
+    requestData_UserLogout,
+    requestData_UserPreview,
+    requestData_UserData,
+    UserAll_payload,
+)
+
+maimai = MaimaiClient()
+
+
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
+logger = logging.getLogger(__name__)
+
+
+async def run_workflow(self: MaimaiClient):
+
+    async with httpx.AsyncClient(verify=False) as client:
+        # Preview 探测
+
+        PreviewResponse = json.loads(
+            await self.call_api(
+                client, "GetUserPreviewApi", requestData_UserPreview, userId
+            )
+        )
+        if PreviewResponse["isLogin"]:
+            logger.error("已在他处登录。")
+            return
+
+        # UserLogin
+
+        LoginResponse = json.loads(
+            await self.call_api(client, "UserLoginApi", requestData_UserLogin, userId)
+        )
+        if LoginResponse["returnCode"] != 1:
+            logger.error("login failed.")
+            return
+
+        loginId = LoginResponse["loginId"]
+        loginDate = LoginResponse["lastLoginDate"]
+
+        # UserData 等
+
+        tasks = [
+            self.call_api(client, "GetUserDataApi", requestData_UserData, userId),
+            self.call_api(client, "GetUserExtendApi", requestData_UserData, userId),
+            self.call_api(client, "GetUserOptionApi", requestData_UserData, userId),
+            self.call_api(client, "GetUserRatingApi", requestData_UserData, userId),
+            self.call_api(client, "GetUserChargeApi", requestData_UserData, userId),
+            self.call_api(client, "GetUserActivityApi", requestData_UserData, userId),
+            self.call_api(
+                client, "GetUserMissionDataApi", requestData_UserData, userId
+            ),
+        ]
+
+        GeneralUserInfo = await asyncio.gather(*tasks)
+        await asyncio.sleep(60)  # 模拟游戏时间
+
+        # UserAll
+
+        requestData_UserAll = UserAll_payload(
+            loginId, loginDate, musicData, GeneralUserInfo
+        )
+
+        await self.call_api(client, "UpsertUserAllApi", requestData_UserAll, userId)
+
+        # UserLogout
+
+        await self.call_api(client, "UserLogoutApi", requestData_UserLogout, userId)
+
+
+if __name__ == "__main__":
+    asyncio.run(run_workflow(maimai))