feat: Bot API 10.2 (1/3 - Communities)

Author: KurimuzonAkuma

compiler/docs/compiler.py

@@ -717,6 +717,15 @@ def get_title_list(s: str) -> list:
             GlobalPrivacySettings
             HistoryCleared
             ChatFolderInviteLinkInfo
+            Community
+            CommunityAdministratorRights
+            CommunityMemberStatus
+            CommunityMemberStatusCreator
+            CommunityMemberStatusAdministrator
+            CommunityMemberStatusMember
+            CommunityMemberStatusLeft
+            CommunityMemberStatusBanned
+            CommunityPermissions
         """,
         messages_media="""
         Messages & Media
@@ -859,6 +868,8 @@ def get_title_list(s: str) -> list:
             ChecklistTask
             ChecklistTasksAdded
             ChecklistTasksDone
+            CommunityChatAdded
+            CommunityChatRemoved
             Checklist
             RefundedPayment
             ReplyParameters

pyrogram/client.py

@@ -757,7 +757,7 @@ def set_parse_mode(self, parse_mode: Optional["enums.ParseMode"]):
 
         self.parse_mode = parse_mode
 
-    async def fetch_peers(self, peers: List[Union[raw.types.User, raw.types.Chat, raw.types.Channel]]) -> bool:
+    async def fetch_peers(self, peers: List[Union["raw.base.User", "raw.base.Chat"]]) -> bool:
         is_min = False
         parsed_peers = []
         parsed_usernames = []
@@ -797,6 +797,10 @@ async def fetch_peers(self, peers: List[Union[raw.types.User, raw.types.Chat, ra
                 peer_id = utils.get_channel_id(peer.id)
                 access_hash = peer.access_hash
                 peer_type = "channel" if peer.broadcast else "supergroup"
+            elif isinstance(peer, (raw.types.Community, raw.types.CommunityForbidden)):
+                peer_id = utils.get_channel_id(peer.id)
+                access_hash = peer.access_hash
+                peer_type = "community"
             else:
                 continue
 

pyrogram/enums/message_service_type.py

@@ -222,6 +222,12 @@ class MessageServiceType(AutoName):
     CHECKLIST_TASKS_ADDED = auto()
     "Checklist tasks added"
 
+    COMMUNITY_CHAT_ADDED = auto()
+    "Community chat added"
+
+    COMMUNITY_CHAT_REMOVED = auto()
+    "Community chat removed"
+
     UPGRADED_GIFT_PURCHASE_OFFER = auto()
     "Upgraded gift purchase offer"
 

pyrogram/storage/sqlite_storage.py

@@ -141,7 +141,7 @@ def get_input_peer(peer_id: int, access_hash: int, peer_type: str):
             chat_id=-peer_id
         )
 
-    if peer_type in ["direct", "channel", "forum", "supergroup"]:
+    if peer_type in ["direct", "channel", "forum", "supergroup", "community"]:
         return raw.types.InputPeerChannel(
             channel_id=utils.get_channel_id(peer_id),
             access_hash=access_hash

pyrogram/types/messages_and_media/__init__.py

@@ -36,6 +36,8 @@
 from .checklist_task import ChecklistTask
 from .checklist_tasks_added import ChecklistTasksAdded
 from .checklist_tasks_done import ChecklistTasksDone
+from .community_chat_added import CommunityChatAdded
+from .community_chat_removed import CommunityChatRemoved
 from .contact import Contact
 from .contact_registered import ContactRegistered
 from .craft_gift_result import CraftGiftResult, CraftGiftResultFail, CraftGiftResultSuccess
@@ -235,6 +237,8 @@
     "ChecklistTask",
     "ChecklistTasksAdded",
     "ChecklistTasksDone",
+    "CommunityChatAdded",
+    "CommunityChatRemoved",
     "Contact",
     "ContactRegistered",
     "CraftGiftResult",

pyrogram/types/messages_and_media/community_chat_added.py

@@ -0,0 +1,48 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram 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.
+#
+#  Pyrogram 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 Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+from typing import Dict
+
+import pyrogram
+from pyrogram import raw, types
+
+from ..object import Object
+
+
+class CommunityChatAdded(Object):
+    """Describes a service message about a chat being added to a community.
+
+    Parameters:
+        community (:obj:`~pyrogram.types.Community`):
+            The new community to which the chat belongs.
+    """
+
+    def __init__(self, *, community: "types.ChecklistTask"):
+        super().__init__()
+
+        self.community = community
+
+    @staticmethod
+    def _parse(
+        client: "pyrogram.Client",
+        action: "raw.types.MessageActionChangeCommunity",
+        chats: Dict[int, "raw.base.Chat"],
+    ) -> "CommunityChatAdded":
+        return CommunityChatAdded(
+            community=types.Community._parse(client, chats.get(action.community_id)),
+        )

pyrogram/types/messages_and_media/community_chat_removed.py

@@ -0,0 +1,29 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram 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.
+#
+#  Pyrogram 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 Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+from ..object import Object
+
+
+class CommunityChatRemoved(Object):
+    """Describes a service message about a chat being removed from a community.
+
+    Currently holds no information.
+    """
+
+    def __init__(self):
+        super().__init__()

pyrogram/types/messages_and_media/message.py

@@ -438,6 +438,12 @@ class Message(Object, Update):
         checklist_tasks_added (:obj:`~pyrogram.types.ChecklistTasksAdded`, *optional*):
             Service message: checklist tasks added.
 
+        community_chat_added (:obj:`~pyrogram.types.CommunityChatAdded`, *optional*):
+            Service message: chat added to a Community.
+
+        community_chat_removed (:obj:`~pyrogram.types.CommunityChatRemoved`, *optional*):
+            Service message: chat removed from a Community.
+
         premium_gift_code (:obj:`~pyrogram.types.PremiumGiftCode`, *optional*):
             Service message: premium gift code information.
 
@@ -734,6 +740,8 @@ def __init__(
         direct_message_price_changed: Optional["types.DirectMessagePriceChanged"] = None,
         checklist_tasks_done: Optional[List["types.ChecklistTasksDone"]] = None,
         checklist_tasks_added: Optional[List["types.ChecklistTasksAdded"]] = None,
+        community_chat_added: Optional[List["types.CommunityChatAdded"]] = None,
+        community_chat_removed: Optional[List["types.CommunityChatRemoved"]] = None,
         premium_gift_code: Optional["types.PremiumGiftCode"] = None,
         gifted_premium: Optional["types.GiftedPremium"] = None,
         gifted_stars: Optional["types.GiftedStars"] = None,
@@ -916,6 +924,8 @@ def __init__(
         self.direct_message_price_changed = direct_message_price_changed
         self.checklist_tasks_done = checklist_tasks_done
         self.checklist_tasks_added = checklist_tasks_added
+        self.community_chat_added = community_chat_added
+        self.community_chat_removed = community_chat_removed
         self.premium_gift_code = premium_gift_code
         self.gifted_premium = gifted_premium
         self.gifted_stars = gifted_stars
@@ -1064,6 +1074,8 @@ async def _parse_service(
         direct_message_price_changed = None
         checklist_tasks_done = None
         checklist_tasks_added = None
+        community_chat_added = None
+        community_chat_removed = None
 
         service_type = enums.MessageServiceType.UNSUPPORTED
 
@@ -1335,6 +1347,14 @@ async def _parse_service(
         elif isinstance(action, raw.types.MessageActionTodoAppendTasks):
             service_type = enums.MessageServiceType.CHECKLIST_TASKS_ADDED
             checklist_tasks_added = types.ChecklistTasksAdded._parse(client, message, users, chats)
+        elif isinstance(action, raw.types.MessageActionChangeCommunity):
+            if action.community_id:
+                service_type = enums.MessageServiceType.COMMUNITY_CHAT_ADDED
+                community_chat_added = types.CommunityChatAdded._parse(client, action, chats)
+            else:
+                service_type = enums.MessageServiceType.COMMUNITY_CHAT_REMOVED
+                community_chat_removed = types.CommunityChatRemoved()
+
 
         parsed_message = Message(
             id=message.id,
@@ -1411,6 +1431,8 @@ async def _parse_service(
             direct_message_price_changed=direct_message_price_changed,
             checklist_tasks_done=checklist_tasks_done,
             checklist_tasks_added=checklist_tasks_added,
+            community_chat_added=community_chat_added,
+            community_chat_removed=community_chat_removed,
             reactions=types.MessageReactions._parse(client, message.reactions, users, chats),
             business_connection_id=business_connection_id,
             raw=message,

pyrogram/types/user_and_chats/__init__.py

@@ -34,14 +34,31 @@
 from .chat_folder_invite_link_info import ChatFolderInviteLinkInfo
 from .chat_invite_link import ChatInviteLink
 from .chat_join_request import ChatJoinRequest
-from .chat_join_result import ChatJoinResult, ChatJoinResultSuccess, ChatJoinResultRequestSent, ChatJoinResultGuardBotApprovalRequired, ChatJoinResultDeclined
+from .chat_join_result import (
+    ChatJoinResult,
+    ChatJoinResultDeclined,
+    ChatJoinResultGuardBotApprovalRequired,
+    ChatJoinResultRequestSent,
+    ChatJoinResultSuccess,
+)
 from .chat_joiner import ChatJoiner
 from .chat_member import ChatMember
 from .chat_member_updated import ChatMemberUpdated
 from .chat_permissions import ChatPermissions
 from .chat_photo import ChatPhoto
 from .chat_reactions import ChatReactions
 from .chat_settings import ChatSettings
+from .community import Community
+from .community_administrator_rights import CommunityAdministratorRights
+from .community_member_status import (
+    CommunityMemberStatus,
+    CommunityMemberStatusAdministrator,
+    CommunityMemberStatusBanned,
+    CommunityMemberStatusCreator,
+    CommunityMemberStatusLeft,
+    CommunityMemberStatusMember,
+)
+from .community_permissions import CommunityPermissions
 from .dialog import Dialog
 from .emoji_status import EmojiStatus
 from .failed_to_add_member import FailedToAddMember
@@ -86,10 +103,10 @@
     "ChatInviteLink",
     "ChatJoinRequest",
     "ChatJoinResult",
-    "ChatJoinResultSuccess",
-    "ChatJoinResultRequestSent",
-    "ChatJoinResultGuardBotApprovalRequired",
     "ChatJoinResultDeclined",
+    "ChatJoinResultGuardBotApprovalRequired",
+    "ChatJoinResultRequestSent",
+    "ChatJoinResultSuccess",
     "ChatJoiner",
     "ChatMember",
     "ChatMemberUpdated",
@@ -98,6 +115,15 @@
     "ChatPrivileges",
     "ChatReactions",
     "ChatSettings",
+    "Community",
+    "CommunityAdministratorRights",
+    "CommunityMemberStatus",
+    "CommunityMemberStatusAdministrator",
+    "CommunityMemberStatusBanned",
+    "CommunityMemberStatusCreator",
+    "CommunityMemberStatusLeft",
+    "CommunityMemberStatusMember",
+    "CommunityPermissions",
     "Dialog",
     "EmojiStatus",
     "FailedToAddMember",
@@ -120,5 +146,5 @@
     "VideoChatEnded",
     "VideoChatMembersInvited",
     "VideoChatScheduled",
-    "VideoChatStarted"
+    "VideoChatStarted",
 ]