From 34abd54f93852a6e7eec87ff7580d65a69d6133f Mon Sep 17 00:00:00 2001 From: 简律纯 Date: Fri, 21 Apr 2023 13:59:59 +0800 Subject: 😎迁移 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- HydroRoll/Hydro/__init__.py | 2 + HydroRoll/Hydro/iamai_plugin_dice/__init__.py | 51 ++++++++++++++++++++++ HydroRoll/Hydro/iamai_plugin_dice/config.py | 13 ++++++ HydroRoll/Hydro/roll.py | 11 +++++ HydroRoll/__init__.py | 6 +++ HydroRoll/main.py | 6 --- HydroRoll/plugins/Hydro.py | 1 + HydroRoll/plugins/iamai_plugin_dice/__init__.py | 50 --------------------- HydroRoll/plugins/iamai_plugin_dice/config.py | 13 ------ .../plugins/iamai_plugin_system_info/__init__.py | 41 +++++++++++++++++ .../plugins/iamai_plugin_system_info/config.py | 11 +++++ 11 files changed, 136 insertions(+), 69 deletions(-) create mode 100644 HydroRoll/Hydro/__init__.py create mode 100644 HydroRoll/Hydro/iamai_plugin_dice/__init__.py create mode 100644 HydroRoll/Hydro/iamai_plugin_dice/config.py create mode 100644 HydroRoll/Hydro/roll.py create mode 100644 HydroRoll/__init__.py delete mode 100644 HydroRoll/main.py create mode 100644 HydroRoll/plugins/Hydro.py delete mode 100644 HydroRoll/plugins/iamai_plugin_dice/__init__.py delete mode 100644 HydroRoll/plugins/iamai_plugin_dice/config.py create mode 100644 HydroRoll/plugins/iamai_plugin_system_info/__init__.py create mode 100644 HydroRoll/plugins/iamai_plugin_system_info/config.py diff --git a/HydroRoll/Hydro/__init__.py b/HydroRoll/Hydro/__init__.py new file mode 100644 index 0000000..4171cac --- /dev/null +++ b/HydroRoll/Hydro/__init__.py @@ -0,0 +1,2 @@ +from .iamai_plugin_dice import Dice +from .roll import Roll \ No newline at end of file diff --git a/HydroRoll/Hydro/iamai_plugin_dice/__init__.py b/HydroRoll/Hydro/iamai_plugin_dice/__init__.py new file mode 100644 index 0000000..396278a --- /dev/null +++ b/HydroRoll/Hydro/iamai_plugin_dice/__init__.py @@ -0,0 +1,51 @@ +import re +import random + +from iamai.log import logger + +from plugins.iamai_plugin_base import CommandPluginBase + +from .config import Config + +__all__ = ["Dice"] + +class Dice(CommandPluginBase[None, Config]): + Config = Config + + def __post_init__(self): + self.re_pattern = re.compile( + r"\s*(?P\d+)d(?P\d+)([*x](?P\d+))?", + flags=re.I, + ) + + async def handle(self) -> None: + dice_times = int(self.msg_match.group("dice_times")) + dice_faces = int(self.msg_match.group("dice_faces")) + if self.msg_match.group("dice_multiply") is None: + dice_multiply = None + else: + dice_multiply = int(self.msg_match.group("dice_multiply")) + + if dice_times > self.config.max_dice_times: + await self.event.reply( + self.format_str(self.config.exceed_max_dice_times_str) + ) + return + + dice = [random.randint(1, dice_faces) for _ in range(dice_times)] + dice_sum = sum(dice) + if dice_multiply is None: + result_str = f"{dice_times}D{dice_faces}=" + if dice_times != 1: + result_str += f"{'+'.join(map(lambda x: str(x), dice))}=" + result_str += str(dice_sum) + else: + result_str = f"{dice_times}D{dice_faces}x{dice_multiply}=" + if dice_times != 1: + result_str += ( + f"({'+'.join(map(lambda x: str(x), dice))})x{dice_multiply}=" + ) + result_str += f"{dice_sum}x{dice_multiply}={dice_sum * dice_multiply}" + + logger.info(f"Dice Plugin: {result_str}") + await self.event.reply(self.format_str(self.config.message_str, result_str)) diff --git a/HydroRoll/Hydro/iamai_plugin_dice/config.py b/HydroRoll/Hydro/iamai_plugin_dice/config.py new file mode 100644 index 0000000..a8c6700 --- /dev/null +++ b/HydroRoll/Hydro/iamai_plugin_dice/config.py @@ -0,0 +1,13 @@ +from typing import Set + +from plugins.iamai_plugin_base import CommandPluginConfig + + +class Config(CommandPluginConfig): + __config_name__ = "plugin_dice" + command: Set[str] = {"r", "roll", "dice"} + """命令文本。""" + max_dice_times: int = 1000 + """最大单次投掷次数。""" + exceed_max_dice_times_str: str = "错误：超过最大投掷次数。" + """超过最大单次投掷次数时的提示语。""" diff --git a/HydroRoll/Hydro/roll.py b/HydroRoll/Hydro/roll.py new file mode 100644 index 0000000..8ccf0cd --- /dev/null +++ b/HydroRoll/Hydro/roll.py @@ -0,0 +1,11 @@ +from iamai import Plugin + +class Roll(Plugin): + async def handle(self) -> None: + await self.event.reply("1") + + async def rule(self) -> bool: + return( + self.event.adapter.name == "cqhttp" and + self.event.get_plain_text() == "1" + ) \ No newline at end of file diff --git a/HydroRoll/__init__.py b/HydroRoll/__init__.py new file mode 100644 index 0000000..c290288 --- /dev/null +++ b/HydroRoll/__init__.py @@ -0,0 +1,6 @@ +from iamai import Bot +from .Hydro import Dice +bot = Bot(hot_reload=True) + +if __name__ == "__main__": + bot.run() diff --git a/HydroRoll/main.py b/HydroRoll/main.py deleted file mode 100644 index bb95e79..0000000 --- a/HydroRoll/main.py +++ /dev/null @@ -1,6 +0,0 @@ -from iamai import Bot - -bot = Bot(hot_reload=True) - -if __name__ == "__main__": - bot.run() diff --git a/HydroRoll/plugins/Hydro.py b/HydroRoll/plugins/Hydro.py new file mode 100644 index 0000000..52895ea --- /dev/null +++ b/HydroRoll/plugins/Hydro.py @@ -0,0 +1 @@ +from .. import Hydro \ No newline at end of file diff --git a/HydroRoll/plugins/iamai_plugin_dice/__init__.py b/HydroRoll/plugins/iamai_plugin_dice/__init__.py deleted file mode 100644 index 6005406..0000000 --- a/HydroRoll/plugins/iamai_plugin_dice/__init__.py +++ /dev/null @@ -1,50 +0,0 @@ -import re -import random - -from iamai.log import logger - -from plugins.iamai_plugin_base import CommandPluginBase - -from .config import Config - - -class Dice(CommandPluginBase[None, Config]): - Config = Config - - def __post_init__(self): - self.re_pattern = re.compile( - r"\s*(?P\d+)d(?P\d+)([*x](?P\d+))?", - flags=re.I, - ) - - async def handle(self) -> None: - dice_times = int(self.msg_match.group("dice_times")) - dice_faces = int(self.msg_match.group("dice_faces")) - if self.msg_match.group("dice_multiply") is None: - dice_multiply = None - else: - dice_multiply = int(self.msg_match.group("dice_multiply")) - - if dice_times > self.config.max_dice_times: - await self.event.reply( - self.format_str(self.config.exceed_max_dice_times_str) - ) - return - - dice = [random.randint(1, dice_faces) for _ in range(dice_times)] - dice_sum = sum(dice) - if dice_multiply is None: - result_str = f"{dice_times}D{dice_faces}=" - if dice_times != 1: - result_str += f"{'+'.join(map(lambda x: str(x), dice))}=" - result_str += str(dice_sum) - else: - result_str = f"{dice_times}D{dice_faces}x{dice_multiply}=" - if dice_times != 1: - result_str += ( - f"({'+'.join(map(lambda x: str(x), dice))})x{dice_multiply}=" - ) - result_str += f"{dice_sum}x{dice_multiply}={dice_sum * dice_multiply}" - - logger.info(f"Dice Plugin: {result_str}") - await self.event.reply(self.format_str(self.config.message_str, result_str)) diff --git a/HydroRoll/plugins/iamai_plugin_dice/config.py b/HydroRoll/plugins/iamai_plugin_dice/config.py deleted file mode 100644 index a8c6700..0000000 --- a/HydroRoll/plugins/iamai_plugin_dice/config.py +++ /dev/null @@ -1,13 +0,0 @@ -from typing import Set - -from plugins.iamai_plugin_base import CommandPluginConfig - - -class Config(CommandPluginConfig): - __config_name__ = "plugin_dice" - command: Set[str] = {"r", "roll", "dice"} - """命令文本。""" - max_dice_times: int = 1000 - """最大单次投掷次数。""" - exceed_max_dice_times_str: str = "错误：超过最大投掷次数。" - """超过最大单次投掷次数时的提示语。""" diff --git a/HydroRoll/plugins/iamai_plugin_system_info/__init__.py b/HydroRoll/plugins/iamai_plugin_system_info/__init__.py new file mode 100644 index 0000000..31fdf63 --- /dev/null +++ b/HydroRoll/plugins/iamai_plugin_system_info/__init__.py @@ -0,0 +1,41 @@ +import re +import sys +from plugins.iamai_plugin_base import CommandPluginBase +from .config import Config +import psutil +import time + + +class SystemInfo(CommandPluginBase[None, Config]): + priority: int = 0 + Config = Config + + def __post_init__(self): + self.re_pattern = re.compile(r"(?P.*)", flags=re.I) + + def get_system_info(self) -> str: + cpu_usage = psutil.cpu_percent() + memory_usage = psutil.virtual_memory().percent + disk_usage = psutil.disk_usage('/').percent + + current_time = time.time() + start_time = psutil.Process().create_time() + + uptime_seconds = int(current_time - start_time) + uptime_str = time.strftime("%H:%M:%S", time.gmtime(uptime_seconds)) + + info_str = f"HydroRoll!{self.bot.config.Chien['version']}({self.bot.config.Chien['svn']}) built in Python {sys.version}\n" + info_str += f"本地时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())}\n" + info_str += f"已启动时间：{uptime_str}\n" + info_str += f"CPU使用率：{cpu_usage}%\n" + info_str += f"内存占用率：{memory_usage}%\n" + info_str += f"磁盘使用率：{disk_usage}%" + + return info_str + + async def handle(self) -> None: + system_info = self.get_system_info() + message_str = f"{system_info}" + await self.event.reply( + self.format_str(self.config.message_str,message_str) + ) \ No newline at end of file diff --git a/HydroRoll/plugins/iamai_plugin_system_info/config.py b/HydroRoll/plugins/iamai_plugin_system_info/config.py new file mode 100644 index 0000000..8571c38 --- /dev/null +++ b/HydroRoll/plugins/iamai_plugin_system_info/config.py @@ -0,0 +1,11 @@ +from typing import Set + +from plugins.iamai_plugin_base import CommandPluginConfig + + +class Config(CommandPluginConfig): + __config_name__ = "plugin_system_info" + command: Set[str] = {"system"} + """命令文本。""" + message_str: str = "{message}" + """最终发送消息的格式。""" -- cgit v1.2.3-70-g09d2