diff options
Diffstat (limited to 'ATRI/plugins/polaroid')
| -rw-r--r-- | ATRI/plugins/polaroid/__init__.py | 8 | ||||
| -rw-r--r-- | ATRI/plugins/polaroid/data_source.py | 8 |
2 files changed, 9 insertions, 7 deletions
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: |