show bot username

SpringsFern · May 10, 2024 · f358110 · f358110
1 parent 8550bd8
commit f358110
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/WebStreamer/bot/clients.py b/WebStreamer/bot/clients.py
@@ -46,7 +46,9 @@ async def start_client(client_id, token):
                 session_string=session_string,
                 in_memory=True,
             ).start()
-            client.id = (await client.get_me()).id
+            me = await client.get_me()
+            client.id = me.id
+            client.username = me.username
             work_loads[client_id] = 0
             return client_id, client
         except Exception:

diff --git a/WebStreamer/server/stream_routes.py b/WebStreamer/server/stream_routes.py
@@ -24,8 +24,8 @@ async def root_route_handler(_):
             "telegram_bot": "@" + StreamBot.username,
             "connected_bots": len(multi_clients),
             "loads": dict(
-                ("bot" + str(c), l)
-                for c,  l in sorted(work_loads.items())
+                ("bot" + multi_clients[c].username, l)
+                for c,  l in work_loads.items()
                 # for c, (_, l) in enumerate(
             #         sorted(work_loads.items(), key=lambda x: x[1], reverse=True)
             #     )