Fix Message.command when case is different (pyrogram#757)

alissonlauffer · delivrance · web-flow · commit 02a396910172 · 2021-09-14T18:33:54.000+02:00
* Fix Message.command when case is different

* Update test_command.py

Co-authored-by: Dan &lt;14043624+delivrance@users.noreply.github.com&gt;
diff --git a/pyrogram/filters.py b/pyrogram/filters.py
@@ -794,7 +794,8 @@ 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,
+                                         flags=re.IGNORECASE if not flt.case_sensitive else 0)
 
                 # 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
@@ -107,6 +107,10 @@ async def test_with_args():
     await f(c, m)
     assert m.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"]
