From b7f99a0466834d5b9c45052d94ac57918efc584a Mon Sep 17 00:00:00 2001 From: 简律纯 Date: Sat, 18 Nov 2023 04:54:23 +0800 Subject: chore: 我想睡觉，但我失眠了。 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- example/plugins/HydroRoll/__init__.py | 31 ++++++++++++++++++------------- 1 file changed, 18 insertions(+), 13 deletions(-) diff --git a/example/plugins/HydroRoll/__init__.py b/example/plugins/HydroRoll/__init__.py index d0b4c72..0ff17fa 100644 --- a/example/plugins/HydroRoll/__init__.py +++ b/example/plugins/HydroRoll/__init__.py @@ -73,23 +73,27 @@ class Dice(Plugin[MessageEvent, Annotated[dict, {}], RegexPluginConfig]): flag = True in [cmd.startswith(current_cmd) for cmd in command_list] logger.info(f"Command {current_cmd} not found with flag {flag}") if args[0] in [".root", ".roots"]: - import requests - - data = requests.get("https://vercel-hitokoto.vercel.app/api/roots").json() - await self.event.reply(data["line"]) - else: - if args[0] == ".core": - await self.event.reply(f"{self.state}") + try: + import aiohttp + + async with aiohttp.ClientSession() as session: + async with session.get("https://api.hydroroll.team/api/roots") as response: + data = await response.json() + await self.event.reply(data["line"]) + except Exception as e: + await self.event.reply(f"{e!r}") + elif args[0] == ".core": + await self.event.reply(f"{self.state}") # if args[0].startswith(".set"): # resolve = Set(args[1:]) # TODO: handle multiple sets # elif args[0].startswith(".get"): # resolve = Get(args[1:]) # TODO: handle multiple gets - elif args[0].startswith(".test"): - try: - result = eval(self.event.message.get_plain_text()[5:]) - await self.event.reply(str(result)) - except Exception as error: - await self.event.reply(f"{error!r}") + elif args[0].startswith(".test"): + try: + result = eval(self.event.message.get_plain_text()[5:]) + await self.event.reply(str(result)) + except Exception as error: + await self.event.reply(f"{error!r}") async def rule(self) -> bool: """ @@ -128,4 +132,5 @@ class Dice(Plugin[MessageEvent, Annotated[dict, {}], RegexPluginConfig]): return models def load_models(self): + """我想睡觉, 但我失眠了。""" self.models = self._load_models(self.model_path_list, self.model_dict) -- cgit v1.2.3-70-g09d2