From 7c79376f40e1a23b2bc82b8b89957b6f42a73b14 Mon Sep 17 00:00:00 2001 From: Kyomotoi <0w0@imki.moe> Date: Fri, 10 Mar 2023 01:12:02 +0800 Subject: =?UTF-8?q?=E2=99=BB=EF=B8=8F=20=E9=87=8D=E6=9E=84=20console,=20?= =?UTF-8?q?=E4=BD=BF=E4=B9=8B=E4=BB=A3=E7=A0=81=E5=8F=AF=E8=AF=BB=E6=80=A7?= =?UTF-8?q?=E6=8F=90=E9=AB=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/console/driver/depends.py | 38 ---------------------------------- 1 file changed, 38 deletions(-) delete mode 100644 ATRI/plugins/console/driver/depends.py (limited to 'ATRI/plugins/console/driver/depends.py') diff --git a/ATRI/plugins/console/driver/depends.py b/ATRI/plugins/console/driver/depends.py deleted file mode 100644 index 8205f44..0000000 --- a/ATRI/plugins/console/driver/depends.py +++ /dev/null @@ -1,38 +0,0 @@ -from typing import Union -from datetime import datetime - -from fastapi import Query, HTTPException, status -from starlette.websockets import WebSocket - -from ..data_source import AuthDealer - - -def http_author(token: Union[str, None] = Query(default=None)): - data = AuthDealer.get() - if data is None: - raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="验证信息不存在") - - now_time = datetime.now().timestamp() - if token != data.token: - raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="密钥不匹配, 请检查") - elif now_time > data.dead_time: - raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="密钥已过期") - else: - return token - - -async def websocket_author( - websocket: WebSocket, token: Union[str, None] = Query(default=None) -): - data = AuthDealer.get() - if not data: - await websocket.close(code=status.WS_1008_POLICY_VIOLATION) - return - - now_time = datetime.now().timestamp() - if token != data.token: - await websocket.close(code=status.WS_1008_POLICY_VIOLATION) - elif now_time > data.dead_time: - await websocket.close(code=status.WS_1008_POLICY_VIOLATION) - else: - return token -- cgit v1.2.3