♻️ 跟进 service 的变化

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: