Fix export_chat_invite_link broken because of Layer update

Fixes #244
2019-04-30 11:04:37 +02:00 · 2019-04-30 11:04:37 +02:00 · 2aad59856d
commit 2aad59856d
parent 16d3b2d56e
1 changed files with 4 additions and 8 deletions
--- a/pyrogram/client/methods/chats/export_chat_invite_link.py
+++ b/pyrogram/client/methods/chats/export_chat_invite_link.py
@ -44,15 +44,11 @@ class ExportChatInviteLink(BaseClient):
        """
        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))