切换虎牙宝典活动任务到WSS会话

core/huya/elite_session.py

@@ -28,6 +28,7 @@ class HuyaEliteWssSession:
     ):
         self.uid = int(uid or 0)
         self.cookie = cookie or ""
+        self.guid = HuyaHttpClient._resolve_cookie_guid(self.cookie)
         self.kind = kind
         self.logger = logger or (lambda _message: None)
         self.loop = asyncio.new_event_loop()
@@ -35,9 +36,8 @@ class HuyaEliteWssSession:
         self._closed = False
 
         if kind == "activity":
-            guid = HuyaHttpClient._resolve_cookie_guid(self.cookie)
             baseinfo = urllib.parse.unquote(
-                generate_http_baseinfo(self.uid, guid, "")
+                generate_http_baseinfo(self.uid, self.guid, "")
             )
             connect_host = host or ACTIVITY_WS_HOST
             origin = "https://zt.huya.com"
@@ -52,7 +52,7 @@ class HuyaEliteWssSession:
                 self.client.connect(host=connect_host, origin=origin, cookie=self.cookie)
             )
             initialized = self.loop.run_until_complete(
-                self.client.initialize_activity(self.uid, "", self.cookie)
+                self.client.initialize_activity(self.uid, self.guid, self.cookie)
                 if kind == "activity"
                 else self.client.initialize(self.uid, "", self.cookie)
             )
@@ -96,6 +96,9 @@ class HuyaEliteWssSession:
     def get_user_score(self, uid: int, cookie: str, sid: int):
         return self._run(self.client.get_user_score(uid, cookie, sid))
 
+    def query_user_score(self, uid: int, cookie: str, sid: int):
+        return self.get_user_score(uid, cookie, sid)
+
     def get_act_prize_list(self, uid: int, cookie: str, sid: int):
         return self._run(self.client.get_act_prize_list(uid, cookie, sid))