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/auto_tietie.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'ATRI/plugins/auto_tietie.py')

diff --git a/ATRI/plugins/auto_tietie.py b/ATRI/plugins/auto_tietie.py
index 61f1370..177c2a9 100644
--- a/ATRI/plugins/auto_tietie.py
+++ b/ATRI/plugins/auto_tietie.py
@@ -1,12 +1,11 @@
 from random import choice
 
-from nonebot.permission import SUPERUSER
 from nonebot.adapters.onebot.v11 import MessageEvent
 from nonebot.adapters.onebot.v11 import MessageSegment
 from nonebot.adapters.onebot.v11.helpers import Cooldown
 
 from ATRI.service import Service
-from ATRI.rule import is_in_service
+from ATRI.permission import MASTER
 
 
 _is_tietie = True
@@ -29,7 +28,12 @@ _tietie_wd = choice(
     ]
 )
 
-tt = Service("贴贴", "全自动贴贴机", True, is_in_service("贴贴"), permission=SUPERUSER)
+tt = (
+    Service("贴贴")
+    .document("全自动贴贴机")
+    .only_admin(True)
+    .permission(MASTER)
+)
 
 
 auto_tietie = tt.on_message(
@@ -44,7 +48,7 @@ async def _(event: MessageEvent):
 
     user_id = event.get_user_id()
     at = MessageSegment.at(user_id)
-    result = at + _tietie_wd
+    result = at + _tietie_wd  # type: ignore
     await auto_tietie.finish(result)
 
 
-- 
cgit v1.2.3