diff options
Diffstat (limited to 'ATRI/plugins/console/driver/__init__.py')
-rw-r--r-- | ATRI/plugins/console/driver/__init__.py | 48 |
1 files changed, 16 insertions, 32 deletions
diff --git a/ATRI/plugins/console/driver/__init__.py b/ATRI/plugins/console/driver/__init__.py index e5dfdc6..b6b9f89 100644 --- a/ATRI/plugins/console/driver/__init__.py +++ b/ATRI/plugins/console/driver/__init__.py @@ -5,32 +5,14 @@ from fastapi.middleware.cors import CORSMiddleware from ATRI import conf from ATRI.log import log -from ATRI.plugins.console.data_source import FRONTEND_DIR -from .view import ( - handle_auther, - handle_base_uri, - handle_control_service, - handle_edit_block, - handle_get_block_list, - handle_get_service_list, - handle_runtime_info, - handle_message_deal_info, -) +from .path import * +from .api import * -CONSOLE_API_URI = "/capi" # base point -CONSOLE_API_AUTH_URI = "/capi/auth" # 验证后台许可 -CONSOLE_API_RUNTIME_URI = "/capi/runtime" # 获取运行占用信息 -CONSOLE_API_MESSAGE_URI = "/capi/message" # 获取信息处理信息 +from ..data_source import FRONTEND_DIR -CONSOLE_API_SERVICE_LIST_URI = "/capi/service/list" # 获取服务列表 -CONSOLE_API_SERVICE_CONTROL_URI = "/capi/service/control" # 对服务作出修改 -CONSOLE_API_BLOCK_LIST_URI = "/capi/block/list" # 获取封禁列表 -CONSOLE_API_BLOCK_EDIT_URI = "/capi/block/edit" # 编辑封禁列表 - - -def register_routes(driver: Driver): +def register_driver(driver: Driver): app = driver.server_app origins = ["*"] @@ -42,16 +24,18 @@ def register_routes(driver: Driver): allow_headers=["Content-Type"], ) - app.get(CONSOLE_API_URI)(handle_base_uri) - app.get(CONSOLE_API_AUTH_URI)(handle_auther) - app.get(CONSOLE_API_RUNTIME_URI)(handle_runtime_info) - app.get(CONSOLE_API_MESSAGE_URI)(handle_message_deal_info) + app.get(CONSOLE_BASE_URL)(base_url) + + app.get(CONSOLE_AUTH_URL)(auth_info) + + app.websocket(CONSOLE_RUNTIME_INFO_URL)(runtime_info) + app.websocket(CONSOLE_MESSAGE_INFO_URL)(message_info) - app.get(CONSOLE_API_SERVICE_LIST_URI)(handle_get_service_list) - app.get(CONSOLE_API_SERVICE_CONTROL_URI)(handle_control_service) + app.get(CONSOLE_SERVICE_LIST_URL)(service_list) + app.get(CONSOLE_SERVICE_EDIT_URL)(edit_service) - app.get(CONSOLE_API_BLOCK_LIST_URI)(handle_get_block_list) - app.get(CONSOLE_API_BLOCK_EDIT_URI)(handle_edit_block) + app.get(CONSOLE_BLOCK_LIST_URL)(block_list_info) + app.get(CONSOLE_BLOCK_EDIT_URL)(edit_block_list) static_path = str(FRONTEND_DIR) app.mount( @@ -64,6 +48,6 @@ def register_routes(driver: Driver): def init_driver(): from ATRI import driver + register_driver(driver()) # type: ignore c_url = f"{conf.BotConfig.host}:{conf.BotConfig.port}" - log.info(f"控制台将运行于: http://{c_url} 对应API节点为: /capi") - register_routes(driver()) # type: ignore + log.success(f"控制台将运行于: http://{c_url} 对应API节点为: /capi") |