Fix export_chat_invite_link broken because of Layer update

delivrance · delivrance · commit 2aad59856d27 · 2019-04-30T11:04:37.000+02:00
Fixes #244
diff --git a/pyrogram/client/methods/chats/export_chat_invite_link.py b/pyrogram/client/methods/chats/export_chat_invite_link.py
@@ -44,15 +44,11 @@ def export_chat_invite_link(
         """
         peer = self.resolve_peer(chat_id)
 
-        if isinstance(peer, types.InputPeerChat):
+        if isinstance(peer, (types.InputPeerChat, types.InputPeerChannel)):
             return self.send(
                 functions.messages.ExportChatInvite(
-                    peer=peer.chat_id
-                )
-            ).link
-        elif isinstance(peer, types.InputPeerChannel):
-            return self.send(
-                functions.channels.ExportInvite(
-                    channel=peer
+                    peer=peer
                 )
             ).link
+        else:
+            raise ValueError("The chat_id \"{}\" belongs to a user".format(chat_id))
