Fix export_chat_invite_link not working correctly (channels/supergroups)

delivrance · delivrance · commit 580c684cb3a6 · 2019-06-30T16:13:10.000+02:00
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))
