Fix query bug

vanhbakaa · Nov 16, 2024 · 8ded2eb · 8ded2eb
1 parent 8830ec1
commit 8ded2eb
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/bot/utils/launcher.py b/bot/utils/launcher.py
@@ -44,8 +44,13 @@ def fetch_username(query):
         json_data = json.loads(fetch_data)
         return json_data['username']
     except:
-        logger.warning(f"Invaild query: {query}")
-        sys.exit()
+        try:
+            fetch_data = unquote(query).split("user=")[1].split("&auth_date=")[0]
+            json_data = json.loads(fetch_data)
+            return json_data['username']
+        except:
+            logger.warning(f"Invaild query: {query}")
+            sys.exit()
 
 async def get_token(session_name):
     async with AIOFile('token.json', 'r') as file:
@@ -233,4 +238,4 @@ async def run_tasks(tg_clients: list[Client]):
         for tg_client in tg_clients
     ]
 
-    await asyncio.gather(*tasks)
+    await asyncio.gather(*tasks)