Fix duplicated commands in Message.command

delivrance · delivrance · commit 636ff776d61a · 2021-05-06T19:21:25.000+02:00
Also add more test cases
Related to #676
diff --git a/pyrogram/filters.py b/pyrogram/filters.py
@@ -789,7 +789,7 @@ async def func(flt, client: pyrogram.Client, message: Message):
                                 flags=re.IGNORECASE if not flt.case_sensitive else 0):
                     continue
 
-                without_command = re.sub(rf"{cmd}(?:@?{username})?\s", "", without_prefix, count=1)
+                without_command = re.sub(rf"{cmd}(?:@?{username})?\s?", "", without_prefix, count=1)
 
                 # match.groups are 1-indexed, group(1) is the quote, group(2) is the text
                 # between the quotes, group(3) is unquoted, whitespace-split text
diff --git a/tests/filters/test_command.py b/tests/filters/test_command.py
@@ -103,6 +103,14 @@ async def test_with_mention():
 async def test_with_args():
     f = filters.command("start")
 
+    m = Message("/start")
+    await f(c, m)
+    assert m.command == ["start"]
+
+    m = Message("/start@username")
+    await f(c, m)
+    assert m.command == ["start"]
+
     m = Message("/start a b c")
     await f(c, m)
     assert m.command == ["start"] + list("abc")
