summaryrefslogtreecommitdiff
path: root/ATRI/plugins/hitokoto.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/hitokoto.py')
-rw-r--r--ATRI/plugins/hitokoto.py32
1 files changed, 15 insertions, 17 deletions
diff --git a/ATRI/plugins/hitokoto.py b/ATRI/plugins/hitokoto.py
index fdc58c0..94fec98 100644
--- a/ATRI/plugins/hitokoto.py
+++ b/ATRI/plugins/hitokoto.py
@@ -8,25 +8,23 @@ from ATRI.modules import response # type: ignore
url = 'https://api.imjad.cn/hitokoto/?cat=a&charset=utf-8&length=50&encode=json&fun=sync&source='
-HITOKOTO_REPLY = """{hitokoto}
- from {author}"""
-
-
@on_command('hitokoto', aliases = ['一言'], only_to_me = False)
async def hitokoto(session: CommandSession):
- rep = response.request_api(url)
-
- if not rep:
- session.finish('获取失败')
-
- dc = json.loads(response.request_api(url))
-
- await session.send(
- HITOKOTO_REPLY.format(
- hitokoto = dc["hitokoto"],
- author = dc["author"]
- )
- )
+ user = session.event.user_id
+ with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
+ data = json.load(f)
+
+ if data[f"{user}"] == str(user):
+ pass
+ else:
+ rep = response.request_api(url)
+
+ if not rep:
+ session.finish('获取失败')
+
+ dc = json.loads(response.request_api(url))
+
+ await session.send(dc["hitokoto"])
@on_natural_language('一言', only_to_me = False)
async def _(session: NLPSession):