summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ATRI/plugins/console/driver/api.py10
1 files changed, 7 insertions, 3 deletions
diff --git a/ATRI/plugins/console/driver/api.py b/ATRI/plugins/console/driver/api.py
index 4319954..668a308 100644
--- a/ATRI/plugins/console/driver/api.py
+++ b/ATRI/plugins/console/driver/api.py
@@ -2,7 +2,7 @@ import asyncio
from fastapi import Depends, status
from starlette.websockets import WebSocket, WebSocketState
-from websockets.exceptions import ConnectionClosedOK
+from websockets.exceptions import ConnectionClosedOK, ConnectionClosedError
from .depends import *
from .data_source import (
@@ -34,6 +34,8 @@ async def runtime_info(websocket: WebSocket, _pass=Depends(websocket_author)):
await asyncio.sleep(1)
except ConnectionClosedOK:
pass
+ except ConnectionClosedError:
+ pass
finally:
await websocket.close()
return
@@ -50,6 +52,8 @@ async def message_info(websocket: WebSocket, _pass=Depends(websocket_author)):
await asyncio.sleep(1)
except ConnectionClosedOK:
pass
+ except ConnectionClosedError:
+ pass
finally:
await websocket.close()
return
@@ -79,10 +83,10 @@ def block_list_info(_=Depends(http_author)):
return {"status": status.HTTP_200_OK, "data": get_block_list()}
-def edit_block_list(
+async def edit_block_list(
enabled: str, user_id: str = str(), group_id: str = str(), _=Depends(http_author)
):
return {
"status": status.HTTP_200_OK,
- "data": _edit_block_list(bool(int(enabled)), user_id, group_id),
+ "data": await _edit_block_list(bool(int(enabled)), user_id, group_id),
}