From b65279046ec64c2913fce856d07ef2ef2f613296 Mon Sep 17 00:00:00 2001 From: KurimuzonAkuma Date: Thu, 8 May 2025 16:21:42 +0300 Subject: [PATCH] Update upgraded gift regex Signed-off-by: wulan17 --- pyrogram/client.py | 1 + pyrogram/methods/payments/get_upgraded_gift.py | 2 +- pyrogram/methods/payments/set_pinned_gifts.py | 4 ++-- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/pyrogram/client.py b/pyrogram/client.py index 1e8fef35..52b8aba0 100644 --- a/pyrogram/client.py +++ b/pyrogram/client.py @@ -231,6 +231,7 @@ class Client(Methods): PARENT_DIR = Path(sys.argv[0]).parent INVITE_LINK_RE = re.compile(r"^(?:https?://)?(?:www\.)?(?:t(?:elegram)?\.(?:org|me|dog)/(?:joinchat/|\+))([\w-]+)$") + UPGRADED_GIFT_RE = re.compile(r"^(?:https?://)?(?:www\.)?(?:t(?:elegram)?\.(?:org|me|dog)/(?:nft/|\+))([\w-]+)$") WORKERS = min(32, (os.cpu_count() or 0) + 4) # os.cpu_count() can be None WORKDIR = PARENT_DIR diff --git a/pyrogram/methods/payments/get_upgraded_gift.py b/pyrogram/methods/payments/get_upgraded_gift.py index 613be8dd..b91dc65d 100644 --- a/pyrogram/methods/payments/get_upgraded_gift.py +++ b/pyrogram/methods/payments/get_upgraded_gift.py @@ -47,7 +47,7 @@ class GetUpgradedGift: # Get information about upgraded gift by slug gift = await client.get_upgraded_gift("SignetRing-903") """ - match = re.match(r"^(?:https?://)?(?:www\.)?(?:t(?:elegram)?\.(?:org|me|dog)/(?:nft/|\+))([\w-]+)$", link) + match = self.UPGRADED_GIFT_RE.match(link) if match: slug = match.group(1) diff --git a/pyrogram/methods/payments/set_pinned_gifts.py b/pyrogram/methods/payments/set_pinned_gifts.py index 56352d43..ce7318b9 100644 --- a/pyrogram/methods/payments/set_pinned_gifts.py +++ b/pyrogram/methods/payments/set_pinned_gifts.py @@ -62,8 +62,8 @@ class SetPinnedGifts: 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) + saved_gift_match = re.match(r"^(\d+)_(\d+)$", str(gift)) + slug_match = self.UPGRADED_GIFT_RE.match(str(gift)) if saved_gift_match: stargifts.append(