Skip to content

Add can_send_plain permission to ChatPermissions #196

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 11 commits into
base: dev
Choose a base branch
from
Open

Add can_send_plain permission to ChatPermissions #196

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
83f2111
FIX get_dialogs.py
WubbaLubbaDubDubDev Aug 19, 2025
6ebe425
FIX get_dialogs.py
WubbaLubbaDubDubDev Aug 19, 2025
7c58833
Update get_dialogs.py
SpEcHiDe Aug 20, 2025
1414e4c
Update get_dialogs.py
SpEcHiDe Aug 20, 2025
d54c5bd
Update get_dialogs.py
SpEcHiDe Aug 20, 2025
4fab224
Update chat_permissions.py
WubbaLubbaDubDubDev Aug 20, 2025
34847ab
Update chat_permissions.py docstring
WubbaLubbaDubDubDev Aug 20, 2025
6e06426
Merge branch 'TelegramPlayground:dev' into dev
WubbaLubbaDubDubDev Aug 20, 2025
161b347
Merge branch 'dev' into dev
SpEcHiDe Aug 28, 2025
e3171d5
Update set_chat_permissions.py
SpEcHiDe Aug 28, 2025
acb1ddf
Update chat_permissions.py
SpEcHiDe Aug 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 3 additions & 16 deletions pyrogram/methods/chats/set_chat_permissions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -54,33 +54,20 @@ async def set_chat_permissions(
# Completely restrict chat
await app.set_chat_permissions(chat_id, ChatPermissions())

# Chat members can only send text messages and media messages
# Chat members can only send text messages and documents
await app.set_chat_permissions(
chat_id,
ChatPermissions(
can_send_messages=True,
can_send_media_messages=True
can_send_documents=True
)
)
"""

r = await self.invoke(
raw.functions.messages.EditChatDefaultBannedRights(
peer=await self.resolve_peer(chat_id),
banned_rights=raw.types.ChatBannedRights(
until_date=0,
send_messages=not permissions.can_send_messages,
send_media=not permissions.can_send_media_messages,
send_stickers=not permissions.can_send_other_messages,
send_gifs=not permissions.can_send_other_messages,
send_games=not permissions.can_send_other_messages,
send_inline=not permissions.can_send_other_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,
)
banned_rights=permissions.write(False)
)
)

Expand Down
138 changes: 99 additions & 39 deletions pyrogram/types/user_and_chats/chat_permissions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,10 +16,13 @@
# You should have received a copy of the GNU Lesser General Public License
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.

import logging
from datetime import datetime
from pyrogram import raw, utils
from ..object import Object

log = logging.getLogger(__name__)


class ChatPermissions(Object):
"""Describes actions that a non-administrator user is allowed to take in a chat.
Expand Down Expand Up @@ -70,9 +73,6 @@ class ChatPermissions(Object):
True, if the user is allowed to create forum topics
If omitted defaults to the value of can_pin_messages

can_send_media_messages (``bool``, *optional*):
True, if the user is allowed to send audios, documents, photos, videos, video notes and voice notes.
Implies *can_send_messages*.
"""

def __init__(
Expand All @@ -92,7 +92,6 @@ def __init__(
can_invite_users: bool = None,
can_pin_messages: bool = None,
can_manage_topics: bool = None,
can_send_media_messages: bool = None, # Audio files, documents, photos, videos, video notes and voice notes
):
super().__init__(None)

Expand All @@ -110,7 +109,6 @@ def __init__(
self.can_invite_users = can_invite_users
self.can_pin_messages = can_pin_messages
self.can_manage_topics = can_manage_topics
self.can_send_media_messages = can_send_media_messages

@staticmethod
def _parse(denied_permissions: "raw.base.ChatBannedRights") -> "ChatPermissions":
Expand All @@ -128,7 +126,6 @@ def _parse(denied_permissions: "raw.base.ChatBannedRights") -> "ChatPermissions"
can_invite_users = False
can_pin_messages = False
can_manage_topics = False
can_send_media_messages = False

if isinstance(denied_permissions, raw.types.ChatBannedRights):
can_send_messages = not denied_permissions.send_messages
Expand Down Expand Up @@ -161,14 +158,27 @@ def _parse(denied_permissions: "raw.base.ChatBannedRights") -> "ChatPermissions"
can_send_videos = not denied_permissions.send_videos
can_send_video_notes = not denied_permissions.send_roundvideos
can_send_voice_notes = not denied_permissions.send_voices
send_aidem = any([
can_send_audios,
can_send_documents,
can_send_photos,
can_send_videos,
can_send_video_notes,
can_send_voice_notes,
])
if not send_aidem:
can_send_messages = not any([
denied_permissions.send_messages,
denied_permissions.send_plain
])
else:
can_send_media_messages = not denied_permissions.send_media
can_send_audios = can_send_media_messages
can_send_documents = can_send_media_messages
can_send_photos = can_send_media_messages
can_send_videos = can_send_media_messages
can_send_video_notes = can_send_media_messages
can_send_voice_notes = can_send_media_messages
send_media = not denied_permissions.send_media
can_send_audios = send_media
can_send_documents = send_media
can_send_photos = send_media
can_send_videos = send_media
can_send_video_notes = send_media
can_send_voice_notes = send_media

return ChatPermissions(
can_send_messages=can_send_messages,
Expand All @@ -185,37 +195,87 @@ def _parse(denied_permissions: "raw.base.ChatBannedRights") -> "ChatPermissions"
can_invite_users=can_invite_users,
can_pin_messages=can_pin_messages,
can_manage_topics=can_manage_topics,
can_send_media_messages=can_send_media_messages
)

def write(
permissions: "ChatPermissions",
use_independent_chat_permissions: bool,
until_date: datetime = utils.zero_datetime()
) -> "raw.base.ChatBannedRights":
return 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,
send_stickers=not permissions.can_send_other_messages,
send_gifs=not permissions.can_send_other_messages,
send_games=not permissions.can_send_other_messages,
send_inline=not permissions.can_send_other_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=(
permissions.can_manage_topics and
not permissions.can_manage_topics
) or not permissions.can_pin_messages,
# view_messages=# TODO
send_audios=not permissions.can_send_audios,# TODO
send_docs=not permissions.can_send_documents,# TODO
send_photos=not permissions.can_send_photos,# TODO
send_videos=not permissions.can_send_videos,# TODO
send_roundvideos=not permissions.can_send_video_notes,# TODO
send_voices=not permissions.can_send_voice_notes,# TODO
# send_plain=# TODO
if use_independent_chat_permissions:
return raw.types.ChatBannedRights(
until_date=utils.datetime_to_timestamp(until_date),
send_messages=not permissions.can_send_messages,
send_media=False,
send_stickers=not permissions.can_send_other_messages,
send_gifs=not permissions.can_send_other_messages,
send_games=not permissions.can_send_other_messages,
send_inline=not permissions.can_send_other_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=(
permissions.can_manage_topics and
not permissions.can_manage_topics
) or not permissions.can_pin_messages,
view_messages=False,
send_audios=not permissions.can_send_audios,
send_docs=not permissions.can_send_documents,
send_photos=not permissions.can_send_photos,
send_videos=not permissions.can_send_videos,
send_roundvideos=not permissions.can_send_video_notes,
send_voices=not permissions.can_send_voice_notes,
send_plain=not permissions.can_send_messages,
)
else:
send_aidem = any([
permissions.can_send_audios,
permissions.can_send_documents,
permissions.can_send_photos,
permissions.can_send_videos,
permissions.can_send_video_notes,
permissions.can_send_voice_notes,
])
return raw.types.ChatBannedRights(
until_date=utils.datetime_to_timestamp(until_date),
send_messages=False,
send_media=not send_aidem,
send_stickers=not permissions.can_send_other_messages,
send_gifs=not permissions.can_send_other_messages,
send_games=not permissions.can_send_other_messages,
send_inline=not permissions.can_send_other_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=(
permissions.can_manage_topics and
not permissions.can_manage_topics
) or not permissions.can_pin_messages,
view_messages=False,
send_audios=not permissions.can_send_audios,
send_docs=not permissions.can_send_documents,
send_photos=not permissions.can_send_photos,
send_videos=not permissions.can_send_videos,
send_roundvideos=not permissions.can_send_video_notes,
send_voices=not permissions.can_send_voice_notes,
send_plain=not permissions.can_send_messages,
)

@property
def can_send_media_messages(permissions: "ChatPermissions"):
log.warning(
"can_send_media_messages property is deprecated. "
"Please use any of can_send_audios, can_send_documents, can_send_photos, can_send_videos, can_send_video_notes or can_send_voice_notes."
)
return any([
permissions.can_send_audios,
permissions.can_send_documents,
permissions.can_send_photos,
permissions.can_send_videos,
permissions.can_send_video_notes,
permissions.can_send_voice_notes,
])