mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-02 09:26:12 +08:00
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from nonebot.typing import T_State
|
||
from nonebot.matcher import Matcher
|
||
from nonebot.params import EventPlainText
|
||
from nonebot_plugin_saa import MessageFactory
|
||
from nonebot.adapters.onebot.v11 import MessageEvent
|
||
|
||
from ..config import config
|
||
from ..utils import parse_text
|
||
from .utils import gen_handle_cancel, only_allow_private
|
||
|
||
|
||
def do_del_cookie(del_cookie: type[Matcher]):
|
||
handle_cancel = gen_handle_cancel(del_cookie, "删除中止")
|
||
|
||
@del_cookie.handle()
|
||
async def send_list(state: T_State, event: MessageEvent):
|
||
await only_allow_private(event, del_cookie)
|
||
cookies = await config.get_cookie(is_anonymous=False)
|
||
if not cookies:
|
||
await del_cookie.finish("暂无已添加的 Cookie\n请使用“添加cookie”命令添加")
|
||
res = "已添加的 Cookie 为：\n"
|
||
state["cookie_table"] = {}
|
||
for index, cookie in enumerate(cookies, 1):
|
||
state["cookie_table"][index] = cookie
|
||
res += f"{index} {cookie.site_name} {cookie.cookie_name} {len(cookie.targets)}个关联\n"
|
||
if res[-1] != "\n":
|
||
res += "\n"
|
||
res += "请输入要删除的 Cookie 的序号\n输入'取消'中止"
|
||
await MessageFactory(await parse_text(res)).send()
|
||
|
||
@del_cookie.receive(parameterless=[handle_cancel])
|
||
async def do_del(
|
||
state: T_State,
|
||
index_str: str = EventPlainText(),
|
||
):
|
||
try:
|
||
index = int(index_str)
|
||
cookie = state["cookie_table"][index]
|
||
if cookie.targets:
|
||
await del_cookie.reject("只能删除未关联的 Cookie，请使用“取消关联cookie”命令取消关联")
|
||
await config.delete_cookie_by_id(cookie.id)
|
||
except KeyError:
|
||
await del_cookie.reject("序号错误")
|
||
except Exception:
|
||
await del_cookie.reject("删除错误")
|
||
else:
|
||
await del_cookie.finish("删除成功")
|