Support base Client (#40)

Ext-Creators · Jun 29, 2021 · abfe8e5 · abfe8e5
1 parent 1cb6105
commit abfe8e5
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/discord/ext/ipc/server.py b/discord/ext/ipc/server.py
@@ -156,13 +156,17 @@ async def handle_accept(self, request):
                     response = {"error": "Invalid or no endpoint given.", "code": 400}
                 else:
                     server_response = IpcServerResponse(request)
-                    attempted_cls = self.bot.cogs.get(
-                        self.endpoints[endpoint].__qualname__.split(".")[0]
-                    )
+                    try:
+                        attempted_cls = self.bot.cogs.get(
+                            self.endpoints[endpoint].__qualname__.split(".")[0]
+                        )
 
-                    if attempted_cls:
-                        arguments = (attempted_cls, server_response)
-                    else:
+                        if attempted_cls:
+                            arguments = (attempted_cls, server_response)
+                        else:
+                            arguments = (server_response,)
+                    except AttributeError:
+                        # Support base Client
                         arguments = (server_response,)
 
                     try: