Add edit_user_star_subscription and edit_star_subscription method

Closes #212

Author: KurimuzonAkuma

compiler/docs/compiler.py

@@ -449,6 +449,7 @@ def get_title_list(s: str) -> list:
             create_gift_collection
             delete_gift_collection
             drop_gift_original_details
+            edit_star_subscription
             get_gift_collections
             remove_collection_gifts
             reorder_collection_gifts
@@ -497,6 +498,7 @@ def get_title_list(s: str) -> list:
             set_bot_commands
             get_bot_commands
             delete_bot_commands
+            edit_user_star_subscription
             set_bot_default_privileges
             get_bot_default_privileges
             set_chat_menu_button

pyrogram/methods/bots/__init__.py

@@ -23,6 +23,7 @@
 from .answer_web_app_query import AnswerWebAppQuery
 from .create_invoice_link import CreateInvoiceLink
 from .delete_bot_commands import DeleteBotCommands
+from .edit_user_star_subscription import EditUserStarSubscription
 from .get_bot_commands import GetBotCommands
 from .get_bot_default_privileges import GetBotDefaultPrivileges
 from .get_bot_info_description import GetBotInfoDescription
@@ -64,6 +65,7 @@ class Bots(
     SetBotCommands,
     GetBotCommands,
     DeleteBotCommands,
+    EditUserStarSubscription,
     SetBotDefaultPrivileges,
     SetBotInfoDescription,
     SetBotInfoShortDescription,

pyrogram/methods/bots/edit_user_star_subscription.py

@@ -0,0 +1,56 @@
+#  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 raw
+
+
+class EditUserStarSubscription:
+    async def edit_user_star_subscription(
+        self: "pyrogram.Client",
+        user_id: Union[int, str],
+        telegram_payment_charge_id: str,
+        is_canceled: bool,
+    ) -> bool:
+        """Cancels or re-enables Telegram Star subscription for a user.
+
+        .. include:: /_includes/usable-by/bots.rst
+
+        Parameters:
+            user_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target user.
+
+            telegram_payment_charge_id (``str``):
+                Telegram payment identifier of the subscription.
+
+            is_canceled (``bool``):
+                Pass True to cancel the subscription.
+                Pass False to allow the user to enable it.
+
+        Returns:
+            ``bool``: On success, True is returned.
+        """
+        return await self.invoke(
+            raw.functions.payments.BotCancelStarsSubscription(
+                user_id=await self.resolve_peer(user_id),
+                charge_id=telegram_payment_charge_id,
+                restore=is_canceled,
+            )
+        )

pyrogram/methods/payments/__init__.py

@@ -24,6 +24,7 @@
 from .create_gift_collection import CreateGiftCollection
 from .delete_gift_collection import DeleteGiftCollection
 from .drop_gift_original_details import DropGiftOriginalDetails
+from .edit_star_subscription import EditStarSubscription
 from .get_available_gifts import GetAvailableGifts
 from .get_chat_gifts import GetChatGifts
 from .get_chat_gifts_count import GetChatGiftsCount
@@ -61,6 +62,7 @@ class Payments(
     CreateGiftCollection,
     DeleteGiftCollection,
     DropGiftOriginalDetails,
+    EditStarSubscription,
     GetAvailableGifts,
     GetChatGifts,
     GetChatGiftsCount,

pyrogram/methods/payments/edit_star_subscription.py

@@ -0,0 +1,47 @@
+#  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/>.
+
+import pyrogram
+from pyrogram import raw
+
+
+class EditStarSubscription:
+    async def edit_star_subscription(
+        self: "pyrogram.Client", subscription_id: str, is_canceled: bool
+    ) -> bool:
+        """Cancels or re-enables Telegram Star subscription.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            subscription_id (``str``):
+                Identifier of the subscription to change.
+
+            is_canceled (``bool``):
+                New value of is_canceled.
+
+        Returns:
+            ``bool``: On success, True is returned.
+        """
+        return await self.invoke(
+            raw.functions.payments.ChangeStarsSubscription(
+                peer=raw.types.InputPeerSelf(),
+                subscription_id=subscription_id,
+                canceled=is_canceled,
+            )
+        )