Make sure owned_gift_id is str, instead of converting it (#182)

Author: ALiwoto

pyrogram/methods/payments/convert_gift_to_stars.py

@@ -53,6 +53,9 @@ async def convert_gift_to_stars(
                 # Convert gift
                 await app.convert_gift_to_stars(message_id=123)
         """
+        if not isinstance(owned_gift_id, str):
+            raise ValueError(f"owned_gift_id has to be str, but {type(owned_gift_id)} was provided")
+    
         saved_gift_match = re.match(r"^(-\d+)_(\d+)$", owned_gift_id)
         slug_match = self.UPGRADED_GIFT_RE.match(owned_gift_id)
 

pyrogram/methods/payments/get_upgraded_gift.py

@@ -16,8 +16,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
 

pyrogram/methods/payments/hide_gift.py

@@ -54,7 +54,8 @@ async def hide_gift(
                 # Hide gift in channel (owned_gift_id packed in format chatID_savedID)
                 await app.hide_gift(owned_gift_id="123_456")
         """
-        owned_gift_id = str(owned_gift_id)
+        if not isinstance(owned_gift_id, str):
+            raise ValueError(f"owned_gift_id has to be str, but {type(owned_gift_id)} was provided")
 
         saved_gift_match = re.match(r"^(-\d+)_(\d+)$", owned_gift_id)
         slug_match = self.UPGRADED_GIFT_RE.match(owned_gift_id)

pyrogram/methods/payments/set_gift_resale_price.py

@@ -51,7 +51,8 @@ async def set_gift_resale_price(
                 # Change resale price of a unique gift
                 await app.set_gift_resale_price(owned_gift_id="123456", resale_star_count=100)
         """
-        owned_gift_id = str(owned_gift_id)
+        if not isinstance(owned_gift_id, str):
+            raise ValueError(f"owned_gift_id has to be str, but {type(owned_gift_id)} was provided")
 
         saved_gift_match = re.match(r"^(-\d+)_(\d+)$", owned_gift_id)
         slug_match = self.UPGRADED_GIFT_RE.match(owned_gift_id)

pyrogram/methods/payments/set_pinned_gifts.py

@@ -58,8 +58,11 @@ async def set_pinned_gifts(
         stargifts = []
 
         for gift in owned_gift_ids:
-            saved_gift_match = re.match(r"^(-\d+)_(\d+)$", str(gift))
-            slug_match = self.UPGRADED_GIFT_RE.match(str(gift))
+            if not isinstance(gift, str):
+                raise ValueError(f"gift id has to be str, but {type(gift)} was provided")
+
+            saved_gift_match = re.match(r"^(-\d+)_(\d+)$", gift)
+            slug_match = self.UPGRADED_GIFT_RE.match(gift)
 
             if saved_gift_match:
                 stargifts.append(

pyrogram/methods/payments/show_gift.py

@@ -54,7 +54,8 @@ async def show_gift(
                 # Show gift in channel (owned_gift_id packed in format chatID_savedID)
                 await app.show_gift(owned_gift_id="123_456")
         """
-        owned_gift_id = str(owned_gift_id)
+        if not isinstance(owned_gift_id, str):
+            raise ValueError(f"owned_gift_id has to be str, but {type(owned_gift_id)} was provided")
 
         saved_gift_match = re.match(r"^(-\d+)_(\d+)$", owned_gift_id)
         slug_match = self.UPGRADED_GIFT_RE.match(owned_gift_id)

pyrogram/methods/payments/transfer_gift.py

@@ -65,7 +65,8 @@ async def transfer_gift(
                 # Transfer gift to another user
                 await app.transfer_gift(owned_gift_id="123", new_owner_chat_id=123)
         """
-        owned_gift_id = str(owned_gift_id)
+        if not isinstance(owned_gift_id, str):
+            raise ValueError(f"owned_gift_id has to be str, but {type(owned_gift_id)} was provided")
 
         saved_gift_match = re.match(r"^(-\d+)_(\d+)$", owned_gift_id)
         slug_match = self.UPGRADED_GIFT_RE.match(owned_gift_id)

pyrogram/methods/payments/upgrade_gift.py

@@ -66,7 +66,8 @@ async def upgrade_gift(
                 # Upgrade gift in channel (owned_gift_id packed in format chatID_savedID)
                 await app.upgrade_gift(owned_gift_id="123_456")
         """
-        owned_gift_id = str(owned_gift_id)
+        if not isinstance(owned_gift_id, str):
+            raise ValueError(f"owned_gift_id has to be str, but {type(owned_gift_id)} was provided")
 
         saved_gift_match = re.match(r"^(-\d+)_(\d+)$", owned_gift_id)
         slug_match = self.UPGRADED_GIFT_RE.match(owned_gift_id)