Skip to content

Commit dbe7fd7

Browse files
Rename resale count parameters in Gift class
1 parent 21b2e20 commit dbe7fd7

1 file changed

Lines changed: 20 additions & 13 deletions

File tree

  • pyrogram/types/messages_and_media

pyrogram/types/messages_and_media/gift.py

Lines changed: 20 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,10 @@ class Gift(Object):
101101
resale_price (``int``, *optional*):
102102
Number of Telegram Stars that must be paid to buy the gift and send it to someone else.
103103
104-
last_resale_price (``int``, *optional*):
104+
last_resale_star_count (``int``, *optional*):
105105
Number of Telegram Stars that were paid by the sender for the gift; 0 if the gift was upgraded or transferred.
106106
107-
last_ton_resale_price (``int``, *optional*):
107+
last_resale_ton_count (``int``, *optional*):
108108
Number of TON that were paid by the sender for the gift.
109109
110110
number (``int``, *optional*):
@@ -195,8 +195,8 @@ def __init__(
195195
upgrade_price: Optional[int] = None,
196196
transfer_price: Optional[int] = None,
197197
resale_price: Optional[int] = None,
198-
last_resale_price: Optional[int] = None,
199-
last_ton_resale_price: Optional[int] = None,
198+
last_resale_star_count: Optional[int] = None,
199+
last_resale_ton_count: Optional[int] = None,
200200
upgrade_message_id: Optional[int] = None,
201201
name: Optional[str] = None,
202202
title: Optional[str] = None,
@@ -242,8 +242,8 @@ def __init__(
242242
self.upgrade_price = upgrade_price
243243
self.transfer_price = transfer_price
244244
self.resale_price = resale_price
245-
self.last_resale_price = last_resale_price
246-
self.last_ton_resale_price = last_ton_resale_price
245+
self.last_resale_star_count = last_resale_star_count
246+
self.last_resale_ton_count = last_resale_ton_count
247247
self.upgrade_message_id = upgrade_message_id
248248
self.name = name
249249
self.title = title
@@ -321,15 +321,15 @@ async def _parse_unique(
321321

322322
owner_id = utils.get_raw_peer_id(star_gift.owner_id)
323323

324-
last_resale_price = None
325-
last_ton_resale_price = None
324+
last_resale_star_count = None
325+
last_resale_ton_count = None
326326

327327
if star_gift.resell_amount:
328328
for currency in star_gift.resell_amount:
329329
if isinstance(currency, raw.types.StarsAmount):
330-
last_resale_price = currency.amount
330+
last_resale_star_count = currency.amount
331331
elif isinstance(currency, raw.types.StarsTonAmount):
332-
last_ton_resale_price = currency.amount
332+
last_resale_ton_count = currency.amount
333333

334334
return Gift(
335335
id=star_gift.id,
@@ -345,8 +345,8 @@ async def _parse_unique(
345345
owner_name=star_gift.owner_name,
346346
owner_address=star_gift.owner_address,
347347
gift_address=star_gift.gift_address,
348-
last_resale_price=last_resale_price,
349-
last_ton_resale_price=last_ton_resale_price,
348+
last_resale_star_count=last_resale_star_count,
349+
last_resale_ton_count=last_resale_ton_count,
350350
is_upgraded=True,
351351
raw=star_gift,
352352
client=client
@@ -430,7 +430,14 @@ async def _parse_action(
430430
parsed_gift.can_transfer_at = utils.timestamp_to_datetime(action.can_transfer_at)
431431
parsed_gift.can_resell_at = utils.timestamp_to_datetime(action.can_resell_at)
432432
parsed_gift.transfer_price = action.transfer_stars
433-
parsed_gift.last_resale_star_count = action.resale_stars
433+
434+
if action.resale_amount:
435+
for currency in action.resale_amount:
436+
if isinstance(currency, raw.types.StarsAmount):
437+
parsed_gift.last_resale_star_count = currency.amount
438+
elif isinstance(currency, raw.types.StarsTonAmount):
439+
parsed_gift.last_resale_ton_count = currency.amount
440+
434441
parsed_gift.upgrade_message_id = message.id
435442

436443

0 commit comments

Comments
 (0)