Fix filters.command not working with multiple running bots

delivrance · delivrance · commit 41f16a17c9e5 · 2022-04-24T11:56:07.000+02:00
Closes #864
diff --git a/pyrogram/client.py b/pyrogram/client.py
@@ -286,6 +286,9 @@ def __init__(
 
         self.disconnect_handler = None
 
+        # Username used for mentioned bot commands, e.g.: /start@usernamebot
+        self.username = None
+
         self.loop = asyncio.get_event_loop()
 
     def __enter__(self):
diff --git a/pyrogram/filters.py b/pyrogram/filters.py
@@ -745,11 +745,6 @@ async def linked_channel_filter(_, __, m: Message):
 
 
 # region command_filter
-
-# Used by the command filter below
-username = None
-
-
 def command(commands: Union[str, List[str]], prefixes: Union[str, List[str]] = "/", case_sensitive: bool = False):
     """Filter commands, i.e.: text messages starting with "/" or any other custom prefix.
 
@@ -772,12 +767,7 @@ def command(commands: Union[str, List[str]], prefixes: Union[str, List[str]] = "
     command_re = re.compile(r"([\"'])(.*?)(?<!\\)\1|(\S+)")
 
     async def func(flt, client: pyrogram.Client, message: Message):
-        # Username shared among all commands; used for mention commands, e.g.: /start@username
-        global username
-
-        if username is None:
-            username = (await client.get_me()).username or ""
-
+        username = client.username or ""
         text = message.text or message.caption
         message.command = None
 
diff --git a/pyrogram/methods/auth/initialize.py b/pyrogram/methods/auth/initialize.py
@@ -48,4 +48,5 @@ async def initialize(
         await self.dispatcher.start()
         await Syncer.add(self)
 
+        self.username = (await self.get_me()).username
         self.is_initialized = True
