Added Filters.argument_equals

JosXa · JosXa · commit 9002fa8f4439 · 2018-02-01T16:58:41.000+01:00
diff --git a/telegram/ext/filters.py b/telegram/ext/filters.py
@@ -171,6 +171,34 @@ def filter(self, message):
     command = _Command()
     """:obj:`Filter`: Messages starting with ``/``."""
 
+
+    class argument_equals(BaseFilter):
+        """
+        Filters /commands with a specific argument value. Useful for deep-linking.
+
+        Examples:
+            Example ``CommandHandler("start", deep_linked_callback,
+            Filters.argument_equals('parameter'))``
+
+        Args:
+            argument(:obj:`str`): String to compare with command arguments
+
+        """
+        """:obj:`Filter`: 
+        """
+
+        def __init__(self, argument):
+            self.argument = argument
+            self.name = 'Filters.argument_equals({})'.format(self.argument)
+
+        def filter(self, message):
+            if ' ' in message.text:
+                return bool(message.text and message.text.startswith('/') and
+                            message.text.split(' ', 1)[1] == self.argument)
+            else:
+                return False
+
+
     class _Reply(BaseFilter):
         name = 'Filters.reply'
 
diff --git a/tests/test_filters.py b/tests/test_filters.py
@@ -51,6 +51,22 @@ def test_filters_command(self, message):
         message.text = '/test'
         assert Filters.command(message)
 
+    def test_filters_argument_equals(self, message):
+        message.text = 'test'
+        assert not Filters.argument_equals('abc')(message)
+        message.text = '/test'
+        assert not Filters.argument_equals('abc')(message)
+        message.text = '/test-abc'
+        assert not Filters.argument_equals('abc')(message)
+        message.text = '/test abc'
+        assert Filters.argument_equals('abc')(message)
+        message.text = '/test abc def'
+        assert Filters.argument_equals('abc def')(message)
+        message.text = '/test abc-def'
+        assert Filters.argument_equals('abc-def')(message)
+        message.text = '/test abc/def'
+        assert Filters.argument_equals('abc/def')(message)
+
     def test_filters_reply(self, message):
         another_message = Message(1, User(1, 'TestOther', False), datetime.datetime.now(),
                                   Chat(0, 'private'))
