Skip to content

Commit

Permalink
Pyrofork: fix restrict_chat_member method
Browse files Browse the repository at this point in the history 
Signed-off-by: wulan17 <[email protected]>
(cherry picked from commit 68ca845)
  • Loading branch information
@wulan17 @eyMarv
wulan17 authored and eyMarv committed Feb 5, 2024
1 parent 93e2391 commit ad712e9
Showing 1 changed file with 106 additions and 19 deletions.
125 changes: 106 additions & 19 deletions pyrogram/methods/chats/restrict_chat_member.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,31 +78,118 @@ async def restrict_chat_member(
await app.restrict_chat_member(chat_id, user_id,
ChatPermissions(can_send_messages=True))
"""

if permissions.all_perms is not None:
send_audios=None
send_docs=None
send_games=None
send_gifs=None
send_photos=None
send_plain=None
send_roundvideos=None
send_stickers=None
send_videos=None
send_voices=None
if permissions.all_perms:
send_messages=False
send_media=False
send_polls=False
embed_links=False
change_info=False
invite_users=False
pin_messages=False
manage_topics=False
send_inline=False
else:
send_messages=True
send_media=True
send_polls=True
embed_links=True
change_info=True
invite_users=True
pin_messages=True
manage_topics=True
send_inline=True
else:
old_permissions = (await self.get_chat(chat_id)).permissions
send_messages = None
send_media = None
embed_links=not permissions.can_add_web_page_previews if permissions.can_add_web_page_previews is not None else not old_permissions.can_add_web_page_previews
send_polls=not permissions.can_send_polls if permissions.can_send_polls is not None else not old_permissions.can_send_polls
change_info=not permissions.can_change_info if permissions.can_change_info is not None else not old_permissions.can_change_info
invite_users=not permissions.can_invite_users if permissions.can_invite_users is not None else not old_permissions.can_invite_users
pin_messages=not permissions.can_pin_messages if permissions.can_pin_messages is not None else not old_permissions.can_pin_messages
manage_topics=not permissions.can_manage_topics if permissions.can_manage_topics is not None else not old_permissions.can_manage_topics
send_audios=not permissions.can_send_audios if permissions.can_send_audios is not None else not old_permissions.can_send_audios
send_docs=not permissions.can_send_docs if permissions.can_send_docs is not None else not old_permissions.can_send_docs
send_games=not permissions.can_send_games if permissions.can_send_games is not None else not old_permissions.can_send_games
send_gifs=not permissions.can_send_gifs if permissions.can_send_gifs is not None else not old_permissions.can_send_gifs
send_inline=not permissions.can_send_inline if permissions.can_send_inline is not None else not old_permissions.can_send_inline
send_photos=not permissions.can_send_photos if permissions.can_send_photos is not None else not old_permissions.can_send_photos
send_plain=not permissions.can_send_plain if permissions.can_send_plain is not None else not old_permissions.can_send_plain
send_roundvideos=not permissions.can_send_roundvideos if permissions.can_send_roundvideos is not None else not old_permissions.can_send_roundvideos
send_stickers=not permissions.can_send_stickers if permissions.can_send_stickers is not None else not old_permissions.can_send_stickers
send_videos=not permissions.can_send_videos if permissions.can_send_videos is not None else not old_permissions.can_send_videos
send_voices=not permissions.can_send_voices if permissions.can_send_voices is not None else not old_permissions.can_send_voices
if permissions.can_send_messages is not None:
if permissions.can_send_messages:
send_plain = False
else:
send_plain = True
if permissions.can_send_media_messages is None:
permissions.can_send_media_messages = old_permissions.can_send_media_messages
if permissions.can_send_media_messages is not None:
if permissions.can_send_media_messages:
embed_links = False
send_audios = False
send_docs = False
send_games = False
send_gifs = False
send_inline = False
send_photos = False
send_polls = False
send_roundvideos = False
send_stickers = False
send_videos = False
send_voices = False
else:
embed_links = True
send_audios = True
send_docs = True
send_games = True
send_gifs = True
send_inline = True
send_photos = True
send_polls = True
send_roundvideos = True
send_stickers = True
send_videos = True
send_voices = True
r = await self.invoke(
raw.functions.channels.EditBanned(
channel=await self.resolve_peer(chat_id),
participant=await self.resolve_peer(user_id),
banned_rights=raw.types.ChatBannedRights(
until_date=utils.datetime_to_timestamp(until_date),
send_messages=not permissions.can_send_messages,
send_media=not permissions.can_send_media_messages,
embed_links=not permissions.can_add_web_page_previews,
send_polls=not permissions.can_send_polls,
change_info=not permissions.can_change_info,
invite_users=not permissions.can_invite_users,
pin_messages=not permissions.can_pin_messages,
manage_topics=not permissions.can_manage_topics,
send_audios=not permissions.can_send_audios,
send_docs=not permissions.can_send_docs,
send_games=not permissions.can_send_games,
send_gifs=not permissions.can_send_gifs,
send_inline=not permissions.can_send_inline,
send_photos=not permissions.can_send_photos,
send_plain=not permissions.can_send_plain,
send_roundvideos=not permissions.can_send_roundvideos,
send_stickers=not permissions.can_send_stickers,
send_videos=not permissions.can_send_videos,
send_voices=not permissions.can_send_voices
send_messages=send_messages,
send_media=send_media,
embed_links=embed_links,
send_polls=send_polls,
change_info=change_info,
invite_users=invite_users,
pin_messages=pin_messages,
manage_topics=manage_topics,
send_audios=send_audios,
send_docs=send_docs,
send_games=send_games,
send_gifs=send_gifs,
send_inline=send_inline,
send_photos=send_photos,
send_plain=send_plain,
send_roundvideos=send_roundvideos,
send_stickers=send_stickers,
send_videos=send_videos,
send_voices=send_voices
)
)
)
Expand Down

0 comments on commit ad712e9

Please sign in to comment.