Update types to layer 214

KurimuzonAkuma · KurimuzonAkuma · commit 917af0b49874 · 2025-09-19T21:14:07.000+03:00
Also fixed some bugs
diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py
@@ -374,6 +374,7 @@ def get_title_list(s: str) -> list:
             transfer_chat_ownership
             get_suitable_discussion_chats
             set_chat_discussion_group
+            set_main_profile_tab
         """,
         users="""
         Users
@@ -1135,6 +1136,7 @@ def get_title_list(s: str) -> list:
             PollType
             PrivacyKey
             ProfileColor
+            ProfileTab
             ReplyColor
             SentCodeType
             StoriesPrivacyRules
diff --git a/pyrogram/enums/__init__.py b/pyrogram/enums/__init__.py
@@ -39,6 +39,7 @@
 from .privacy_key import PrivacyKey
 from .privacy_rule_type import PrivacyRuleType
 from .profile_color import ProfileColor
+from .profile_tab import ProfileTab
 from .reply_color import ReplyColor
 from .sent_code_type import SentCodeType
 from .sticker_type import StickerType
@@ -75,6 +76,7 @@
     'PrivacyKey',
     'PrivacyRuleType',
     'ProfileColor',
+    'ProfileTab',
     'ReplyColor',
     'SentCodeType',
     'StickerType',
diff --git a/pyrogram/enums/profile_tab.py b/pyrogram/enums/profile_tab.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 pyrogram import raw
+from .auto_name import AutoName
+
+
+class ProfileTab(AutoName):
+    """Profile tab enumeration used in :obj:`~pyrogram.types.Chat`."""
+
+    POSTS = raw.types.ProfileTabPosts
+    "A tab with stories posted by the user or the channel chat and saved to profile."
+
+    GIFTS = raw.types.ProfileTabGifts
+    "A tab with gifts received by the user or the channel chat."
+
+    MEDIA = raw.types.ProfileTabMedia
+    "A tab with photos and videos posted by the channel."
+
+    FILES = raw.types.ProfileTabFiles
+    "A tab with documents posted by the channel."
+
+    LINKS = raw.types.ProfileTabLinks
+    "A tab with messages posted by the channel and containing links."
+
+    MUSIC = raw.types.ProfileTabMusic
+    "A tab with audio messages posted by the channel."
+
+    VOICE = raw.types.ProfileTabVoice
+    "A tab with voice notes posted by the channel."
+
+    GIFS = raw.types.ProfileTabGifs
+    "A tab with animations posted by the channel."
diff --git a/pyrogram/methods/advanced/resolve_peer.py b/pyrogram/methods/advanced/resolve_peer.py
@@ -116,10 +116,7 @@ async def resolve_peer(
                         )
                     )
 
-                    try:
-                        return await self.storage.get_peer_by_id(utils.get_peer_id(r.peer))
-                    except KeyError:
-                        return await self.storage.get_peer_by_phone_number(phone)
+                    return await self.storage.get_peer_by_id(utils.get_peer_id(r.peer))
             else:
                 username = None
                 channel_id = None
@@ -136,23 +133,20 @@ async def resolve_peer(
 
                 if channel_id:
                     try:
-                        return await self.storage.get_peer_by_id(peer_id)
+                        return await self.storage.get_peer_by_id(channel_id)
                     except KeyError as e:
                         raise PeerIdInvalid from e
                 elif username:
                     try:
-                        return await self.storage.get_peer_by_username(peer_id)
+                        return await self.storage.get_peer_by_username(username)
                     except KeyError:
                         r = await self.invoke(
                             raw.functions.contacts.ResolveUsername(
-                                username=peer_id
+                                username=username
                             )
                         )
 
-                        try:
-                            return await self.storage.get_peer_by_id(utils.get_peer_id(r.peer))
-                        except KeyError:
-                            return await self.storage.get_peer_by_username(peer_id)
+                        return await self.storage.get_peer_by_id(utils.get_peer_id(r.peer))
                 else:
                     raise PeerIdInvalid
         else:
diff --git a/pyrogram/methods/auth/send_code.py b/pyrogram/methods/auth/send_code.py
@@ -103,17 +103,18 @@ async def send_code(
             except (PhoneMigrate, NetworkMigrate) as e:
                 dc_option = await self.get_dc_option(e.value, ipv6=self.ipv6)
                 await self.session.stop()
-                self.session = None
+
+                self.session = await self.get_session(
+                    dc_id=e.value,
+                    server_address=dc_option.ip_address,
+                    port=dc_option.port,
+                    export_authorization=False,
+                    temporary=True
+                )
 
                 await self.storage.dc_id(e.value)
                 await self.storage.server_address(dc_option.ip_address)
                 await self.storage.port(dc_option.port)
-
-                self.session = await self.get_session(
-                     server_address=await self.storage.server_address(),
-                     port=await self.storage.port(),
-                     export_authorization=False,
-                     temporary=True
-                )
+                await self.storage.auth_key(self.session.auth_key)
             else:
                 return types.SentCode._parse(r)
diff --git a/pyrogram/methods/auth/sign_in_bot.py b/pyrogram/methods/auth/sign_in_bot.py
@@ -58,19 +58,19 @@ async def sign_in_bot(
             except UserMigrate as e:
                 dc_option = await self.get_dc_option(e.value, ipv6=self.ipv6)
                 await self.session.stop()
-                self.session = None
-
-                await self.storage.dc_id(e.value)
-                await self.storage.server_address(dc_option.ip_address)
-                await self.storage.port(dc_option.port)
 
                 self.session = await self.get_session(
-                     server_address=await self.storage.server_address(),
-                     port=await self.storage.port(),
-                     export_authorization=False,
-                     temporary=True
+                    dc_id=e.value,
+                    server_address=dc_option.ip_address,
+                    port=dc_option.port,
+                    export_authorization=False,
+                    temporary=True
                 )
 
+                await self.storage.dc_id(e.value)
+                await self.storage.server_address(dc_option.ip_address)
+                await self.storage.port(dc_option.port)
+                await self.storage.auth_key(self.session.auth_key)
             else:
                 await self.storage.user_id(r.user.id)
                 await self.storage.is_bot(True)
diff --git a/pyrogram/methods/chats/__init__.py b/pyrogram/methods/chats/__init__.py
@@ -68,6 +68,7 @@
 from .set_chat_direct_messages_group import SetChatDirectMessagesGroup
 from .set_chat_permissions import SetChatPermissions
 from .set_chat_discussion_group import SetChatDiscussionGroup
+from .set_main_profile_tab import SetMainProfileTab
 from .set_chat_photo import SetChatPhoto
 from .set_chat_protected_content import SetChatProtectedContent
 from .set_chat_title import SetChatTitle
@@ -124,6 +125,7 @@ class Chats(
     SetChatUsername,
     SetChatPermissions,
     SetChatDiscussionGroup,
+    SetMainProfileTab,
     GetDialogsCount,
     GetFolders,
     GetForumTopics,
diff --git a/pyrogram/methods/chats/set_main_profile_tab.py b/pyrogram/methods/chats/set_main_profile_tab.py
@@ -0,0 +1,71 @@
+#  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 Union
+
+import pyrogram
+from pyrogram import enums, raw
+
+
+class SetMainProfileTab:
+    async def set_main_profile_tab(
+        self: "pyrogram.Client",
+        chat_id: Union[int, str],
+        main_profile_tab: "enums.ProfileTab"
+    ) -> bool:
+        """Changes the main profile tab of the user or channel.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            chat_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target chat.
+                For your personal cloud (Saved Messages) you can simply use "me" or "self".
+
+            main_profile_tab (:obj:`~pyrogram.enums.ProfileTab`):
+                The new value of the main profile tab.
+
+        Returns:
+            ``bool``: On success, True is returned.
+
+        Example:
+            .. code-block:: python
+
+                # Set user main profile tab to "Posts"
+                await app.set_main_profile_tab("me", main_profile_tab=enums.ProfileTab.POSTS)
+
+                # Set channel main profile tab to "Gifts"
+                await app.set_main_profile_tab("kurigram_news", main_profile_tab=enums.ProfileTab.GIFTS)
+        """
+        peer = await self.resolve_peer(chat_id)
+
+        if isinstance(peer, raw.types.InputPeerSelf):
+            r = await self.invoke(
+                raw.functions.account.SetMainProfileTab(
+                    tab=main_profile_tab.value(),
+                )
+            )
+        else:
+            r = await self.invoke(
+                raw.functions.channels.SetMainProfileTab(
+                    channel=peer,
+                    tab=main_profile_tab.value()
+                )
+            )
+
+        return r
diff --git a/pyrogram/types/messages_and_media/gift.py b/pyrogram/types/messages_and_media/gift.py
@@ -184,6 +184,16 @@ class Gift(Object):
         is_pinned (``bool``, *optional*):
             True, if the gift is pinned.
 
+        is_upgrade_separate (``bool``, *optional*):
+            True, if the upgrade was bought after the gift was sent.
+            In this case, prepaid upgrade cost must not be added to the gift cost.
+
+        is_theme_available (``bool``, *optional*):
+            True, if the gift can be used to set a theme in a chat.
+
+        used_theme_chat_id (``int``, *optional*):
+            Identifier of the chat for which the gift is used to set a theme.
+
         raw (:obj:`~pyrogram.raw.base.StarGift`, *optional*):
             The raw object as received from the server.
 
@@ -246,6 +256,9 @@ def __init__(
         is_for_birthday: Optional[bool] = None,
         is_premium: Optional[bool] = None,
         is_pinned: Optional[bool] = None,
+        is_upgrade_separate: Optional[bool] = None,
+        is_theme_available: Optional[bool] = None,
+        used_theme_chat_id: Optional[int] = None,
         raw: Optional["raw.base.StarGift"] = None
     ):
         super().__init__(client)
@@ -300,6 +313,9 @@ def __init__(
         self.is_for_birthday = is_for_birthday
         self.is_premium = is_premium
         self.is_pinned = is_pinned
+        self.is_upgrade_separate = is_upgrade_separate
+        self.is_theme_available = is_theme_available
+        self.used_theme_chat_id = used_theme_chat_id
         self.raw = raw
 
     @staticmethod
@@ -372,6 +388,8 @@ async def _parse_unique(
             total_upgraded_count=star_gift.availability_total,
             max_upgraded_count=star_gift.availability_issued,
             is_premium=star_gift.require_premium,
+            is_theme_available=star_gift.theme_available,
+            used_theme_chat_id=utils.get_peer_id(star_gift.theme_peer) if star_gift.theme_peer else None,
             owner=types.Chat._parse_chat(client, users.get(owner_id) or chats.get(owner_id)),
             owner_name=star_gift.owner_name,
             owner_address=star_gift.owner_address,
@@ -411,6 +429,7 @@ async def _parse_saved(
         parsed_gift.is_saved = not saved_gift.unsaved
         parsed_gift.is_refunded = saved_gift.refunded
         parsed_gift.is_pinned = saved_gift.pinned_to_top
+        parsed_gift.is_upgrade_separate = saved_gift.upgrade_separate
         parsed_gift.can_upgrade = saved_gift.can_upgrade
         parsed_gift.from_user = types.User._parse(client, users.get(utils.get_raw_peer_id(saved_gift.from_id)))
         parsed_gift.caption = caption
@@ -447,6 +466,7 @@ async def _parse_action(
             parsed_gift.is_converted = action.converted
             parsed_gift.is_upgraded = action.upgraded
             parsed_gift.is_refunded = action.refunded
+            parsed_gift.is_upgrade_separate = action.upgrade_separate
             parsed_gift.can_upgrade = action.can_upgrade
             parsed_gift.caption = caption
             parsed_gift.caption_entities = caption_entities
diff --git a/pyrogram/types/messages_and_media/gift_resale_price.py b/pyrogram/types/messages_and_media/gift_resale_price.py
@@ -38,7 +38,7 @@ def write(self) -> "raw.base.StarsAmount":
         raise NotImplementedError
 
 
-class GiftResalePriceStar(Object):
+class GiftResalePriceStar(GiftResalePrice):
     """Describes price of a resold gift in Telegram Stars.
 
     Parameters:
@@ -61,7 +61,7 @@ def write(self) -> "raw.types.StarsAmount":
         )
 
 
-class GiftResalePriceTon(Object):
+class GiftResalePriceTon(GiftResalePrice):
     """Describes price of a resold gift in Toncoins.
 
     Parameters:
diff --git a/pyrogram/types/user_and_chats/chat.py b/pyrogram/types/user_and_chats/chat.py
diff --git a/pyrogram/types/user_and_chats/user.py b/pyrogram/types/user_and_chats/user.py
