Update create/edit_chat_invite_link

delivrance · delivrance · commit 29b461584859 · 2021-12-22T14:08:24.000+01:00
diff --git a/pyrogram/methods/invite_links/create_chat_invite_link.py b/pyrogram/methods/invite_links/create_chat_invite_link.py
@@ -27,8 +27,10 @@ class CreateChatInviteLink(Scaffold):
     async def create_chat_invite_link(
         self,
         chat_id: Union[int, str],
+        name: str = None,
         expire_date: int = None,
         member_limit: int = None,
+        creates_join_request: bool = None
     ) -> "types.ChatInviteLink":
         """Create an additional invite link for a chat.
 
@@ -41,6 +43,9 @@ async def create_chat_invite_link(
                 Unique identifier for the target chat or username of the target channel/supergroup
                 (in the format @username).
 
+            name (``str``, *optional*):
+                Invite link name.
+
             expire_date (``int``, *optional*):
                 Point in time (Unix timestamp) when the link will expire.
                 Defaults to None (no expiration date).
@@ -50,6 +55,10 @@ async def create_chat_invite_link(
                 this invite link; 1-99999.
                 Defaults to None (no member limit).
 
+            creates_join_request (``bool``, *optional*):
+                True, if users joining the chat via the link need to be approved by chat administrators.
+                If True, member_limit can't be specified.
+
         Returns:
             :obj:`~pyrogram.types.ChatInviteLink`: On success, the new invite link is returned.
 
@@ -67,6 +76,8 @@ async def create_chat_invite_link(
                 peer=await self.resolve_peer(chat_id),
                 expire_date=expire_date,
                 usage_limit=member_limit,
+                title=name,
+                request_needed=creates_join_request
             )
         )
 
diff --git a/pyrogram/methods/invite_links/edit_chat_invite_link.py b/pyrogram/methods/invite_links/edit_chat_invite_link.py
@@ -28,8 +28,10 @@ async def edit_chat_invite_link(
         self,
         chat_id: Union[int, str],
         invite_link: str,
+        name: str = None,
         expire_date: int = None,
         member_limit: int = None,
+        creates_join_request: bool = None
     ) -> "types.ChatInviteLink":
         """Edit a non-primary invite link.
 
@@ -43,6 +45,9 @@ async def edit_chat_invite_link(
             invite_link (``str``):
                 The invite link to edit
 
+            name (``str``, *optional*):
+                Invite link name.
+
             expire_date (``int``, *optional*):
                 Point in time (Unix timestamp) when the link will expire.
                 Defaults to None (no change), pass 0 to set no expiration date.
@@ -52,6 +57,10 @@ async def edit_chat_invite_link(
                 invite link; 1-99999.
                 Defaults to None (no change), pass 0 to set no member limit.
 
+            creates_join_request (``bool``, *optional*):
+                True, if users joining the chat via the link need to be approved by chat administrators.
+                If True, member_limit can't be specified.
+
         Returns:
             :obj:`~pyrogram.types.ChatInviteLink`: On success, the new invite link is returned
 
@@ -70,6 +79,8 @@ async def edit_chat_invite_link(
                 link=invite_link,
                 expire_date=expire_date,
                 usage_limit=member_limit,
+                title=name,
+                request_needed=creates_join_request
             )
         )
 
