From f5ceb8927f2e7f2a9e29d62c8e4cef876f917249 Mon Sep 17 00:00:00 2001 From: Kyomotoi <1172294279@qq.com> Date: Sat, 6 Feb 2021 00:32:26 +0800 Subject: =?UTF-8?q?=F0=9F=8F=97=20=F0=9F=92=A9=20=E6=9B=B4=E6=94=B9?= =?UTF-8?q?=E9=A1=B9=E7=9B=AE=E7=BB=93=E6=9E=84=EF=BC=8C=E4=BF=AE=E5=A4=8D?= =?UTF-8?q?=E5=95=A5b=20BUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/setu/__init__.py | 77 ---------------------------------------- ATRI/plugins/setu/data_source.py | 27 -------------- 2 files changed, 104 deletions(-) delete mode 100644 ATRI/plugins/setu/__init__.py delete mode 100644 ATRI/plugins/setu/data_source.py (limited to 'ATRI/plugins/setu') diff --git a/ATRI/plugins/setu/__init__.py b/ATRI/plugins/setu/__init__.py deleted file mode 100644 index 2d630de..0000000 --- a/ATRI/plugins/setu/__init__.py +++ /dev/null @@ -1,77 +0,0 @@ -import random - -from nonebot.plugin import on_regex, on_command -from nonebot.adapters.cqhttp.message import MessageSegment, Message -from nonebot.adapters.cqhttp import Bot -from nonebot.permission import SUPERUSER -from nonebot.adapters.cqhttp.event import MessageEvent - -from ATRI.service.plugin import Plugin -from ATRI.config import SETU_CONFIG -from ATRI.utils import compress_image -from ATRI.request import Request -from ATRI.rule import is_in_service, is_in_ban_list, is_in_dormant - -from .data_source import setu_port - -# ===========================[Begin Command Processing]=========================== - -resolution = 1 - - -__plugin_name__ = 'setu' -__doc__ = """ -涩图,开冲! -使用正则匹配 -""" -Plugin.register(plugin_name=__plugin_name__, _type="func", doc=__doc__, command=SETU_CONFIG['setu']['command']) - -setu = on_regex('|'.join(SETU_CONFIG['setu']['command']), - rule=is_in_service(__plugin_name__) & is_in_ban_list() - & is_in_dormant()) - -@setu.handle() -async def _(bot: Bot, event: MessageEvent) -> None: - await bot.send(event, SETU_CONFIG['setu']['repo']['waiting']) - rd = random.randint(1, 2) - - if rd == 1: - data = await setu_port() - else: - data = await setu_port() - - if resolution == 1: - img = compress_image(await Request.get_image(data['data'][0]['url'])) - else: - img = await Request.get_image(data['data'][0]['url']) - - msg0 = ( - f"{data['data'][0]['title']}\n" - f"pid: {data['data'][0]['pid']}\n" - f"{MessageSegment.image(file=f'file:///{img}')}" - ) - - await setu.finish(Message(msg0)) - - -setu_resolution = on_command(SETU_CONFIG['admin']['command'][0], - aliases=set(SETU_CONFIG['admin']['command']), - permission=SUPERUSER) - -@setu_resolution.handle() -async def _(bot, event: MessageEvent, state: dict) -> None: - msg = str(event.get_message()).strip() - if msg: - state['msg'] = msg - -@setu_resolution.got('msg', prompt='请键入正确参数奥') -async def _(bot, event: MessageEvent, state: dict) -> None: - global resolution - resolution = int(state['msg']) - - if resolution == 1: - await setu_resolution.finish('完成~!已启用涩图压缩') - else: - await setu_resolution.finish('完成~!已关闭涩图压缩') - -# ===========================[End Command Processing]============================= diff --git a/ATRI/plugins/setu/data_source.py b/ATRI/plugins/setu/data_source.py deleted file mode 100644 index 521b716..0000000 --- a/ATRI/plugins/setu/data_source.py +++ /dev/null @@ -1,27 +0,0 @@ -import json -from pathlib import Path - -from ATRI.config import SETU_CONFIG -from ATRI.request import Request -from ATRI.exceptions import InvalidRequest - - -DATA_PATH = Path('.') / 'ATRI' / 'data' / 'database' - -async def setu_port() -> dict: - url = SETU_CONFIG['setu']['link']['url'] - params = { - "apikey": SETU_CONFIG['setu']['link']['api_key'], - "r18": 0, - "num": 1 - } - data = {} - try: - data = json.loads(await Request.post_bytes(url, params)) - except InvalidRequest: - raise InvalidRequest('Request failed!') - return data - - -async def setu_local() -> str: - ... -- cgit v1.2.3