add aliases for edit/delete reply markup to Message (#662)

* add aliases for edit/delete reply markup to Message * add towncrier patch note * add missed towncrier patch note description
aiogram · Aug 16, 2021 · c1f605c · c1f605c
1 parent 04bbc82
commit c1f605c
Show file tree

Hide file tree

Showing 3 changed files with 78 additions and 0 deletions.
diff --git a/CHANGES/662.feature b/CHANGES/662.feature
@@ -0,0 +1 @@
+add aliases for edit/delete reply markup to Message
diff --git a/aiogram/types/message.py b/aiogram/types/message.py
@@ -14,6 +14,7 @@
         CopyMessage,
         DeleteMessage,
         EditMessageCaption,
+        EditMessageReplyMarkup,
         EditMessageText,
         SendAnimation,
         SendAudio,
@@ -1788,6 +1789,21 @@ def edit_text(
             reply_markup=reply_markup,
         )
 
+    def edit_reply_markup(
+        self,
+        reply_markup: Optional[InlineKeyboardMarkup] = None,
+    ) -> EditMessageReplyMarkup:
+        from ..methods import EditMessageReplyMarkup
+
+        return EditMessageReplyMarkup(
+            chat_id=self.chat.id,
+            message_id=self.message_id,
+            reply_markup=reply_markup,
+        )
+
+    def delete_reply_markup(self) -> EditMessageReplyMarkup:
+        return self.edit_reply_markup(reply_markup=None)
+
     def edit_caption(
         self,
         caption: str,

diff --git a/tests/test_api/test_types/test_message.py b/tests/test_api/test_types/test_message.py
@@ -7,6 +7,7 @@
     CopyMessage,
     DeleteMessage,
     EditMessageCaption,
+    EditMessageReplyMarkup,
     EditMessageText,
     SendAnimation,
     SendAudio,
@@ -36,6 +37,8 @@
     Document,
     EncryptedCredentials,
     Game,
+    InlineKeyboardButton,
+    InlineKeyboardMarkup,
     Invoice,
     Location,
     MessageAutoDeleteTimerChanged,
@@ -561,6 +564,64 @@ def test_edit_text(self):
         assert isinstance(method, EditMessageText)
         assert method.chat_id == message.chat.id
 
+    def test_edit_reply_markup(self):
+        reply_markup = InlineKeyboardMarkup(
+            inline_keyboard=[
+                [
+                    InlineKeyboardButton(
+                        text="test",
+                        callback_data="test",
+                    ),
+                ],
+            ]
+        )
+        reply_markup_new = InlineKeyboardMarkup(
+            inline_keyboard=[
+                [
+                    InlineKeyboardButton(
+                        text="test2",
+                        callback_data="test2",
+                    ),
+                ],
+            ]
+        )
+
+        message = Message(
+            message_id=42,
+            chat=Chat(id=42, type="private"),
+            date=datetime.datetime.now(),
+            reply_markup=reply_markup,
+        )
+        method = message.edit_reply_markup(
+            reply_markup=reply_markup_new,
+        )
+        assert isinstance(method, EditMessageReplyMarkup)
+        assert method.reply_markup == reply_markup_new
+        assert method.chat_id == message.chat.id
+
+    def test_delete_reply_markup(self):
+        reply_markup = InlineKeyboardMarkup(
+            inline_keyboard=[
+                [
+                    InlineKeyboardButton(
+                        text="test",
+                        callback_data="test",
+                    ),
+                ],
+            ]
+        )
+
+        message = Message(
+            message_id=42,
+            chat=Chat(id=42, type="private"),
+            date=datetime.datetime.now(),
+            reply_markup=reply_markup,
+        )
+        method = message.delete_reply_markup()
+        assert isinstance(method, EditMessageReplyMarkup)
+        assert method.reply_markup is None
+        assert method.chat_id == message.chat.id
+
     def test_edit_caption(self):
         message = Message(
             message_id=42, chat=Chat(id=42, type="private"), date=datetime.datetime.now()