From a8afa113cfef80237031916e20df44bc11df2270 Mon Sep 17 00:00:00 2001 From: Kyomotoi <0w0@imki.moe> Date: Mon, 17 Oct 2022 23:46:00 +0800 Subject: =?UTF-8?q?=E2=99=BB=EF=B8=8F=20=E8=B7=9F=E8=BF=9B=20service=20?= =?UTF-8?q?=E7=9A=84=E5=8F=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/polaroid/__init__.py | 8 +++++++- ATRI/plugins/polaroid/data_source.py | 8 ++------ 2 files changed, 9 insertions(+), 7 deletions(-) (limited to 'ATRI/plugins/polaroid') diff --git a/ATRI/plugins/polaroid/__init__.py b/ATRI/plugins/polaroid/__init__.py index a3d2d61..18795e7 100644 --- a/ATRI/plugins/polaroid/__init__.py +++ b/ATRI/plugins/polaroid/__init__.py @@ -4,13 +4,19 @@ from nonebot.adapters.onebot.v11 import MessageEvent, MessageSegment, Message from nonebot.adapters.onebot.v11.helpers import Cooldown from ATRI.rule import to_bot +from ATRI.service import Service + from .data_source import Polaroid, TEMP_PATH +plugin = Service("拍立得").document("根据头像生成拍立得风格照片！") +pol = Polaroid() + + _flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~"]) -polaroid = Polaroid().on_command("拍立得", "获取一张以自己头像的拍立得图片！需at", rule=to_bot()) +polaroid = plugin.on_command("拍立得", "获取一张以自己头像的拍立得图片! 需at", rule=to_bot()) @polaroid.handle([Cooldown(15, prompt=_flmt_notice)]) diff --git a/ATRI/plugins/polaroid/data_source.py b/ATRI/plugins/polaroid/data_source.py index a48b5eb..e994c5e 100644 --- a/ATRI/plugins/polaroid/data_source.py +++ b/ATRI/plugins/polaroid/data_source.py @@ -1,6 +1,5 @@ from ATRI import driver -from ATRI.service import Service, ServiceTools -from ATRI.rule import is_in_service +from ATRI.service import ServiceTools from ATRI.utils import request from ATRI.log import log from ATRI.exceptions import RequestError @@ -12,10 +11,7 @@ TENCENT_AVATER_URL = "https://q1.qlogo.cn/g?b=qq&nk={user_id}&s=640" SOURCE_URL = "https://jsd.imki.moe/gh/Kyomotoi/CDN@master/project/ATRI/" -class Polaroid(Service): - def __init__(self): - Service.__init__(self, "拍立得", "根据头像生成拍立得风格照片！", rule=is_in_service("拍立得")) - +class Polaroid: @classmethod async def _request(cls, user_id: str) -> bytes: try: -- cgit v1.2.3