Skip to content

Commit

Permalink
pyrofork: Add subscription_expired, subscription_period, and subscrip…
Browse files Browse the repository at this point in the history
…tion_price field to class ChatInviteLink

Signed-off-by: wulan17 <wulan17@nusantararom.org>
  • Loading branch information
wulan17 committed Aug 18, 2024
1 parent f32c304 commit 488356a
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 3 deletions.
2 changes: 1 addition & 1 deletion pyrogram/methods/invite_links/export_chat_invite_link.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,4 +77,4 @@ async def export_chat_invite_link(
)
)

return r.link
return types.ChatInviteLink._parse(self, r.invite)
23 changes: 21 additions & 2 deletions pyrogram/types/user_and_chats/chat_invite_link.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,15 @@ class ChatInviteLink(Object):
pending_join_request_count (``int``, *optional*):
Number of pending join requests created using this link
subscription_expired (``int``, *optional*):
Number of subscription which already expired.
subscription_period (``int``, *optional*):
The period of Subscription.
subscription_price (``int``, *optional*):
The price of Subscription (stars).
"""

def __init__(
Expand All @@ -83,7 +92,10 @@ def __init__(
expire_date: datetime = None,
member_limit: int = None,
member_count: int = None,
pending_join_request_count: int = None
pending_join_request_count: int = None,
subscription_expired: int = None,
subscription_period: int = None,
subscription_price: int = None
):
super().__init__()

Expand All @@ -99,6 +111,9 @@ def __init__(
self.member_limit = member_limit
self.member_count = member_count
self.pending_join_request_count = pending_join_request_count
self.subscription_expired = subscription_expired
self.subscription_period = subscription_period
self.subscription_price = subscription_price

@staticmethod
def _parse(
Expand All @@ -114,6 +129,7 @@ def _parse(
if users is not None
else None
)
subscription_pricing = getattr(invite, "subscription_pricing", None)

return ChatInviteLink(
invite_link=invite.link,
Expand All @@ -127,5 +143,8 @@ def _parse(
expire_date=utils.timestamp_to_datetime(invite.expire_date),
member_limit=invite.usage_limit,
member_count=invite.usage,
pending_join_request_count=invite.requested
pending_join_request_count=invite.requested,
subscription_expired=invite.subscription_expired,
subscription_period=subscription_pricing.period if subscription_pricing is not None else None,
subscription_price=subscription_pricing.amount if subscription_pricing is not None else None
)

0 comments on commit 488356a

Please sign in to comment.