TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make restrict_chat_member return Chat instead of a simple boolean

Browse Source
This commit is contained in:
Dan 2019-03-13 14:48:30 +01:00
parent b01caf10a9
commit 0c8b5f02fc
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pyrogram/client/methods/chats/restrict_chat_member.py
View File

@ -20,6 +20,7 @@ from typing import Union
from pyrogram.api import functions, types from pyrogram.api import functions, types
from ...ext import BaseClient from ...ext import BaseClient
from ...types.user_and_chats import Chat
class RestrictChatMember(BaseClient): class RestrictChatMember(BaseClient):
@ -35,8 +36,8 @@ class RestrictChatMember(BaseClient):
can_send_polls: bool = False, can_send_polls: bool = False,
can_change_info: bool = False, can_change_info: bool = False,
can_invite_users: bool = False, can_invite_users: bool = False,
can_pin_messages: bool = False, can_pin_messages: bool = False
) -> bool: ) -> Chat:
"""Use this method to restrict a user in a supergroup. The bot must be an administrator in the supergroup for """Use this method to restrict a user in a supergroup. The bot must be an administrator in the supergroup for
this to work and must have the appropriate admin rights. Pass True for all boolean parameters to lift this to work and must have the appropriate admin rights. Pass True for all boolean parameters to lift
restrictions from a user. restrictions from a user.
@ -79,8 +80,9 @@ class RestrictChatMember(BaseClient):
can_pin_messages (``bool``, *optional*): can_pin_messages (``bool``, *optional*):
Pass True, if the user can pin messages. Pass True, if the user can pin messages.
Returns: Returns:
True on success. On success, a :obj:`Chat <pyrogram.Chat>` object is returned.
Raises: Raises:
:class:`Error <pyrogram.Error>` in case of a Telegram RPC error. :class:`Error <pyrogram.Error>` in case of a Telegram RPC error.
@ -130,7 +132,7 @@ class RestrictChatMember(BaseClient):
if can_pin_messages: if can_pin_messages:
pin_messages = None pin_messages = None
self.send( r = self.send(
functions.channels.EditBanned( functions.channels.EditBanned(
channel=self.resolve_peer(chat_id), channel=self.resolve_peer(chat_id),
user_id=self.resolve_peer(user_id), user_id=self.resolve_peer(user_id),
@ -151,4 +153,4 @@ class RestrictChatMember(BaseClient):
) )
) )
return True return Chat._parse_chat(self, r.chats[0])