From a5739a8dd2d65bd11e80ff86f7220171df197c27 Mon Sep 17 00:00:00 2001 From: Oleg A Date: Tue, 7 Dec 2021 17:45:13 +0300 Subject: [PATCH] feat: add shortcuts Added Chat.ban_sender_chat() and Chat.unban_sender_chat() shortcuts. --- aiogram/types/chat.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/aiogram/types/chat.py b/aiogram/types/chat.py index 7d808a4d..00b7a656 100644 --- a/aiogram/types/chat.py +++ b/aiogram/types/chat.py @@ -623,6 +623,30 @@ class Chat(base.TelegramObject): message_id=message_id, ) + async def ban_sender_chat( + self, + sender_chat_id: base.Integer, + until_date: typing.Union[ + base.Integer, datetime.datetime, datetime.timedelta, None + ] = None, + ): + """Shortcut for banChatSenderChat method.""" + return await self.bot.ban_chat_sender_chat( + chat_id=self.id, + sender_chat_id=sender_chat_id, + until_date=until_date, + ) + + async def unban_sender_chat( + self, + sender_chat_id: base.Integer, + ): + """Shortcut for unbanChatSenderChat method.""" + return await self.bot.unban_chat_sender_chat( + chat_id=self.id, + sender_chat_id=sender_chat_id, + ) + def __int__(self): return self.id