diff options
Diffstat (limited to 'ATRI/plugins/hitokoto.py')
-rw-r--r-- | ATRI/plugins/hitokoto.py | 32 |
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): |