fix login

HoloArchivists · Feb 19, 2022 · 2657ad6 · 2657ad6
1 parent e663e22
commit 2657ad6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/twspace_dl/login.py b/twspace_dl/login.py
@@ -118,6 +118,8 @@ def login(self) -> str:
         request_flow = self.session.post(
             self.task_url, headers=self._headers, json=self._account_dup_check_data
         ).json()
+        if "auth_token" in request_flow.cookies.keys():
+            return str(request_flow.cookies["auth_token"])
         if "flow_token" in request_flow.keys():
             self.flow_token = request_flow["flow_token"]
             # Sometimes it doesn't check account duplication