From b99c9688e27b0d2387a6137ccdfdfd0da6ce5258 Mon Sep 17 00:00:00 2001
From: yanyongyu <yanyongyu_1@126.com>
Date: Fri, 21 Aug 2020 15:42:33 +0800
Subject: [PATCH] fix ws reverse

---
 nonebot/drivers/fastapi.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/nonebot/drivers/fastapi.py b/nonebot/drivers/fastapi.py
index 7e51f92a..89b2af8e 100644
--- a/nonebot/drivers/fastapi.py
+++ b/nonebot/drivers/fastapi.py
@@ -157,12 +157,13 @@ class Driver(BaseDriver):
         websocket = WebSocket(websocket)
 
         # Create Bot Object
-        if adapter == "coolq":
-            bot = CQBot(self,
-                        "websocket",
-                        self.config,
-                        x_self_id,
-                        websocket=websocket)
+        if adapter in self._adapters:
+            BotClass = self._adapters[adapter]
+            bot = BotClass(self,
+                           "websocket",
+                           self.config,
+                           x_self_id,
+                           websocket=websocket)
         else:
             await websocket.close(code=status.WS_1003_UNSUPPORTED_DATA)
             return