fix: parse fc2 id as int

HoloArchivists · Apr 4, 2023 · 2389c50 · 2389c50
1 parent fe1099f
commit 2389c50
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/fc2_live_dl/fc2.py b/fc2_live_dl/fc2.py
@@ -247,10 +247,13 @@ async def get_websocket_url(self):
             control_token = json.loads(
                 base64.b64decode(jwt_body + "==").decode("utf-8")
             )
-            fc2id = control_token["fc2_id"]
-            if len(fc2id) > 0:
-                self._logger.debug("Logged in with ID", fc2id)
-            else:
+            try:
+                fc2id = control_token["fc2_id"]
+                if int(fc2id) > 0:
+                    self._logger.debug("Logged in with ID", fc2id)
+                else:
+                    self._logger.debug("Using anonymous account")
+            except:
                 self._logger.debug("Using anonymous account")
 
             return "%(url)s?control_token=%(control_token)s" % info