New bound method for Gift type (#188)

Author: z44d

pyrogram/methods/payments/get_chat_gifts.py

@@ -16,7 +16,7 @@
 #  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 Optional, Union
+from typing import Optional, Union, AsyncGenerator
 
 import pyrogram
 from pyrogram import raw, types
@@ -34,7 +34,7 @@ async def get_chat_gifts(
         sort_by_price: Optional[bool] = None,
         limit: int = 0,
         offset: str = ""
-    ):
+    ) -> AsyncGenerator["types.Gift", None]:
         """Get all gifts owned by specified chat.
 
         .. include:: /_includes/usable-by/users.rst

pyrogram/methods/payments/get_gift_upgrade_preview.py

@@ -15,9 +15,6 @@
 #
 #  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 re
-
 import pyrogram
 from pyrogram import raw, types
 
@@ -26,7 +23,7 @@ class GetGiftUpgradePreview:
     async def get_gift_upgrade_preview(
         self: "pyrogram.Client",
         gift_id: int
-    ):
+    ) -> "types.GiftUpgradePreview":
         """Return examples of possible upgraded gifts for a regular gift.
 
         .. include:: /_includes/usable-by/users.rst

pyrogram/methods/payments/get_upgraded_gift.py

@@ -24,7 +24,7 @@ class GetUpgradedGift:
     async def get_upgraded_gift(
         self: "pyrogram.Client",
         link: str
-    ):
+    ) -> "types.Gift":
         """Get information about upgraded gift.
 
         .. include:: /_includes/usable-by/users.rst

pyrogram/methods/payments/search_gifts_for_resale.py

@@ -16,7 +16,7 @@
 #  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 List, Optional
+from typing import List, Optional, AsyncGenerator
 
 import pyrogram
 from pyrogram import enums, raw, types
@@ -30,7 +30,7 @@ async def search_gifts_for_resale(
         attributes: Optional[List["types.UpgradedGiftAttributeId"]] = None,
         limit: int = 0,
         offset: str = ""
-    ):
+    ) -> AsyncGenerator["types.Gift", None]:
         """Get upgraded gifts that can be bought from other owners.
 
         .. include:: /_includes/usable-by/users.rst

pyrogram/types/messages_and_media/gift.py

@@ -20,7 +20,7 @@
 from typing import Dict, List, Optional, Union
 
 import pyrogram
-from pyrogram import raw, types, utils
+from pyrogram import raw, types, utils, enums
 
 from ..object import Object
 
@@ -600,3 +600,41 @@ async def buy(self) -> Optional["types.Message"]:
             gift_link=self.link,
             new_owner_chat_id="me"
         )
+
+    async def send(
+        self,
+        chat_id: Union[int, str],
+        text: Optional[str] = None,
+        parse_mode: Optional["enums.ParseMode"] = None,
+        entities: Optional[List["types.MessageEntity"]] = None,
+        is_private: Optional[bool] = None,
+        pay_for_upgrade: Optional[bool] = None,
+    ) -> Optional["types.Message"]:
+        """Bound method *send* of :obj:`~pyrogram.types.Gift`.
+
+        Use as a shortcut for:
+
+        .. code-block:: python
+
+            await client.send_gift(
+                chat_id="me",
+                gift_id=gift.id
+            )
+
+        Example:
+            .. code-block:: python
+
+                await gift.send("me")
+
+        Returns:
+            :obj:`~pyrogram.types.Message`: On success, the sent message is returned.
+        """
+        return await self._client.send_gift(
+            chat_id=chat_id,
+            gift_id=self.id,
+            text=text,
+            parse_mode=parse_mode,
+            entities=entities,
+            is_private=is_private,
+            pay_for_upgrade=pay_for_upgrade
+        )