suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-04 02:26:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ (admin) 添加验证Cookie有效性接口

Browse Source
This commit is contained in:
suyiiyii 2024-10-21 09:54:01 +08:00
parent c3289671d8
commit a78bb73281
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nonebot_bison/admin_page/api.py
View File

@ -17,7 +17,7 @@ from ..utils.get_bot import get_groups
from .token_manager import token_manager from .token_manager import token_manager
from ..config.db_config import SubscribeDupException from ..config.db_config import SubscribeDupException
from ..platform import site_manager, platform_manager from ..platform import site_manager, platform_manager
from ..utils.site import CookieClientManager, is_cookie_client_manager from ..utils.site import CookieSite, CookieClientManager, is_cookie_client_manager
from ..config import NoSuchUserException, NoSuchTargetException, NoSuchSubscribeException, config from ..config import NoSuchUserException, NoSuchTargetException, NoSuchSubscribeException, config
from .types import ( from .types import (
Cookie, Cookie,
@ -267,3 +267,12 @@ async def add_cookie_target(platform_name: str, target: str, cookie_id: int) ->
async def del_cookie_target(platform_name: str, target: str, cookie_id: int) -> StatusResp: async def del_cookie_target(platform_name: str, target: str, cookie_id: int) -> StatusResp:
await config.delete_cookie_target(target, platform_name, cookie_id) await config.delete_cookie_target(target, platform_name, cookie_id)
return StatusResp(ok=True, msg="") return StatusResp(ok=True, msg="")
@router.post("/cookie/validate", dependencies=[Depends(check_is_superuser)])
async def get_cookie_valid(site_name: str, content: str) -> StatusResp:
site = cast(CookieSite, site_manager[site_name])
if await site.validate_cookie(content):
return StatusResp(ok=True, msg="")
else:
return StatusResp(ok=False, msg="")