Fix export_chat_invite_link not working correctly (channels/supergroups)

pyrogram · Jun 30, 2019 · 580c684 · 580c684
1 parent 0d6f8cd
commit 580c684
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/pyrogram/client/methods/chats/export_chat_invite_link.py b/pyrogram/client/methods/chats/export_chat_invite_link.py
@@ -48,18 +48,15 @@ def export_chat_invite_link(
 
         Raises:
             RPCError: In case of a Telegram RPC error.
+            ValueError: In case the chat_id belongs to a user.
         """
         peer = self.resolve_peer(chat_id)
 
-        if isinstance(peer, types.InputPeerChat):
+        if isinstance(peer, (types.InputPeerChannel, types.InputPeerChat)):
             return self.send(
                 functions.messages.ExportChatInvite(
                     peer=peer
                 )
             ).link
-        elif isinstance(peer, types.InputPeerChannel):
-            return self.send(
-                functions.channels.ExportInvite(
-                    channel=peer
-                )
-            ).link
+        else:
+            raise ValueError('The chat_id "{}" belongs to a user'.format(chat_id))