pyrofork/pyrogram/methods/chats/promote_chat_member.py
wulan17 48c7a5e2f9
Pyrofork: Add support for t.me/<username> link in resolve_peer method
Signed-off-by: wulan17 <wulan17@nusantararom.org>
2024-01-04 19:50:17 +07:00

108 lines
4.3 KiB
Python

# Pyrofork - Telegram MTProto API Client Library for Python
# Copyright (C) 2017-present Dan <https://github.com/delivrance>
# Copyright (C) 2022-present Mayuri-Chan <https://github.com/Mayuri-Chan>
#
# This file is part of Pyrofork.
#
# Pyrofork is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pyrofork is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Pyrofork. If not, see <http://www.gnu.org/licenses/>.
from typing import Union
import pyrogram
from pyrogram import raw, types, errors
class PromoteChatMember:
async def promote_chat_member(
self: "pyrogram.Client",
chat_id: Union[int, str],
user_id: Union[int, str],
privileges: "types.ChatPrivileges" = None,
) -> bool:
"""Promote or demote a user in a supergroup or a channel.
You must be an administrator in the chat for this to work and must have the appropriate admin rights.
Pass False for all boolean parameters to demote a user.
.. include:: /_includes/usable-by/users-bots.rst
Parameters:
chat_id (``int`` | ``str``):
Unique identifier (int) or username (str) of the target chat.
You can also use chat public link in form of *t.me/<username>* (str).
user_id (``int`` | ``str``):
Unique identifier (int) or username (str) of the target user.
For a contact that exists in your Telegram address book you can use his phone number (str).
You can also use user profile link in form of *t.me/<username>* (str).
privileges (:obj:`~pyrogram.types.ChatPrivileges`, *optional*):
New user privileges.
Returns:
``bool``: True on success.
Example:
.. code-block:: python
# Promote chat member to admin
await app.promote_chat_member(chat_id, user_id)
"""
chat_id = await self.resolve_peer(chat_id)
user_id = await self.resolve_peer(user_id)
# See Chat.promote_member for the reason of this (instead of setting types.ChatPrivileges() as default arg).
if privileges is None:
privileges = types.ChatPrivileges()
try:
raw_chat_member = (await self.invoke(
raw.functions.channels.GetParticipant(
channel=chat_id,
participant=user_id
)
)).participant
except errors.RPCError:
raw_chat_member = None
rank = None
if isinstance(raw_chat_member, raw.types.ChannelParticipantAdmin):
rank = raw_chat_member.rank
await self.invoke(
raw.functions.channels.EditAdmin(
channel=chat_id,
user_id=user_id,
admin_rights=raw.types.ChatAdminRights(
anonymous=privileges.is_anonymous,
change_info=privileges.can_change_info,
post_messages=privileges.can_post_messages,
edit_messages=privileges.can_edit_messages,
delete_messages=privileges.can_delete_messages,
ban_users=privileges.can_restrict_members,
invite_users=privileges.can_invite_users,
pin_messages=privileges.can_pin_messages,
add_admins=privileges.can_promote_members,
manage_call=privileges.can_manage_video_chats,
manage_topics=privileges.can_manage_topics,
post_stories=privileges.can_post_stories,
edit_stories=privileges.can_edit_stories,
delete_stories=privileges.can_delete_stories,
other=privileges.can_manage_chat
),
rank=rank or ""
)
)
return True