X Tutup
Skip to content

Commit 27238c5

Browse files
committed
add successful_payment filter under message
1 parent 43f4128 commit 27238c5

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

telegram/ext/filters.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -238,6 +238,13 @@ def filter(self, message):
238238

239239
game = _Game()
240240

241+
class _ShippingQuery(BaseFilter):
242+
243+
def filter(self, message):
244+
return bool(message.shipping_query)
245+
246+
shipping_query = _ShippingQuery()
247+
241248
class entity(BaseFilter):
242249
"""Filters messages to only allow those which have a :class:`telegram.MessageEntity`
243250
where their `type` matches `entity_type`.

tests/test_filters.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,12 @@ def test_filters_game(self):
118118
self.message.game = None
119119
self.assertFalse(Filters.game(self.message))
120120

121+
def test_filters_successful_payment(self):
122+
self.message.successful_payment = 'test'
123+
self.asserTrue(Filters.successful_payment(self.message))
124+
self.message.successful_payment = None
125+
self.assertFalse(Filters.successful_payment(self.message))
126+
121127
def test_filters_status_update(self):
122128
self.assertFalse(Filters.status_update(self.message))
123129

0 commit comments

Comments
 (0)
X Tutup