aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
author简律纯 <hsiangnianian@outlook.com>2023-04-28 03:02:00 +0800
committer简律纯 <hsiangnianian@outlook.com>2023-04-28 03:02:00 +0800
commit19c6e564af050cb68420d1a2249d0570248952af (patch)
tree4e6b85642309352fa365a33ba2601d5d175352aa
parentb278995fd05e0f660cd3bb7296f2f3a1ba770e3a (diff)
parent3d25e00eca9aac40743af903f08a8feb779b626c (diff)
downloadHydroRoll-19c6e564af050cb68420d1a2249d0570248952af.tar.gz
HydroRoll-19c6e564af050cb68420d1a2249d0570248952af.zip
Merge branch 'main' of https://github.com/retrofor/ChienDice
-rw-r--r--HydroRoll/__init__.py4
-rw-r--r--HydroRoll/config.py124
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_bot/__init__.py30
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_bot/config.py11
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_dice/__init__.py51
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_dice/config.py13
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_echo/__init__.py17
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_echo/config.py11
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_luck/__init__.py21
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_luck/config.py15
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_send/__init__.py28
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_send/config.py15
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_system/__init__.py75
-rw-r--r--HydroRoll/plugins/HydroRoll_plugin_system/config.py11
14 files changed, 0 insertions, 426 deletions
diff --git a/HydroRoll/__init__.py b/HydroRoll/__init__.py
deleted file mode 100644
index c72c991..0000000
--- a/HydroRoll/__init__.py
+++ /dev/null
@@ -1,4 +0,0 @@
-name = "hydroroll"
-
-from hydroroll.bot import Bot
-from hydroroll.config import GlobalConfig \ No newline at end of file
diff --git a/HydroRoll/config.py b/HydroRoll/config.py
deleted file mode 100644
index 6739249..0000000
--- a/HydroRoll/config.py
+++ /dev/null
@@ -1,124 +0,0 @@
-import argparse
-import sys
-import platform
-from importlib.metadata import version
-from iamai import Plugin
-import os
-import pickle
-import threading
-
-
-# 创建全局 ArgumentParser 对象
-global_parser = argparse.ArgumentParser(description='hydroroll[水系] 全局命令参数')
-
-# 定义全局配置类
-class GlobalConfig:
- _name = "hydroroll"
- _version = "0.1.0"
- _svn = "2"
- _author = "简律纯"
- _iamai_version = version('iamai')
- _python_ver = sys.version
- _python_ver_raw= '.'.join(map(str, platform.python_version_tuple()[:3]))
- current_path = os.path.dirname(os.path.abspath('__file__'))
-
- # 定义系统组件
- class HydroSystem:
- def __init__(self):
- self.parser = argparse.ArgumentParser(description='hydroroll[水系].system 系统命令参数')
- self.subparsers = self.parser.add_subparsers()
- self.status_parser = self.subparsers.add_parser('status', aliases=['s'], help='系统状态')
- self.reload_parser = self.subparsers.add_parser('reload', aliases=['rld'], help='重新加载系统')
- self.restart_parser = self.subparsers.add_parser('restart', aliases=['rst'], help='重启系统')
- self.help = '\n'.join(self.parser.format_help().replace('\r\n', '\n').replace('\r', '').split('\n')[2:-3])
- class HydroBot:
- def __init__(self) -> None:
- self.parser = argparse.ArgumentParser(description="Bot命令")
-
-
-class ConfigManager:
- def __init__(self, directory):
- self.directory = directory
- self.configs = {}
- self.locks = {}
-
- def get(self, filename, section, key, default=None):
- if not self._check_file_exists(filename):
- return default
- config = self._get_config(filename)
- if section not in config:
- return default
- return config[section].get(key, default)
-
- def set(self, filename, section, key, value):
- config = self._get_config(filename)
- if section not in config:
- config[section] = {}
- config[section][key] = value
- self._save_config(filename, config)
-
- def delete(self, filename, section, key):
- config = self._get_config(filename)
- if section not in config:
- return
- if key not in config[section]:
- return
- del config[section][key]
- self._save_config(filename, config)
-
- def sections(self, filename):
- if not self._check_file_exists(filename):
- return []
- config = self._get_config(filename)
- return list(config.keys())
-
- def section_items(self, filename, section):
- if not self._check_file_exists(filename):
- return []
- config = self._get_config(filename)
- if section not in config:
- return []
- return list(config[section].items())
-
- def files(self):
- return [filename for filename in os.listdir(self.directory) if filename.endswith('.dat')]
-
- def _get_lock(self, filename):
- if filename not in self.locks:
- self.locks[filename] = threading.Lock()
- return self.locks[filename]
-
- def _get_config(self, filename):
- with self._get_lock(filename):
- if filename not in self.configs:
- filepath = os.path.join(self.directory, filename)
- if os.path.exists(filepath):
- try:
- with open(filepath, 'rb') as f:
- self.configs[filename] = pickle.load(f)
- except:
- pass
- if filename not in self.configs:
- self.configs[filename] = {}
- return self.configs[filename]
-
- def _save_config(self, filename, config):
- with self._get_lock(filename):
- try:
- filepath = os.path.join(self.directory, filename)
- backuppath = os.path.join(self.directory, filename + '.bak')
- lockpath = os.path.join(self.directory, filename + '.lock')
- with open(lockpath, 'wb') as f:
- pass
- if os.path.exists(filepath):
- os.replace(filepath, backuppath)
- with open(filepath, 'wb') as f:
- pickle.dump(config, f)
- os.remove(backuppath)
- os.remove(lockpath)
- except:
- pass
-
- def _check_file_exists(self, filename):
- filepath = os.path.join(self.directory, filename)
- return os.path.exists(filepath) and filename.endswith('.dat')
diff --git a/HydroRoll/plugins/HydroRoll_plugin_bot/__init__.py b/HydroRoll/plugins/HydroRoll_plugin_bot/__init__.py
deleted file mode 100644
index 2b3eec8..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_bot/__init__.py
+++ /dev/null
@@ -1,30 +0,0 @@
-import re
-from importlib.metadata import version
-from plugins.hydroroll_plugin_base import CommandPluginBase
-from hydroroll.config import GlobalConfig
-
-from .config import Config
-
-
-class HydroBot(CommandPluginBase[None, Config]):
- Config = Config
- CurrentConfig = GlobalConfig
- priority = 0
-
- def __post_init__(self):
- self.re_pattern = re.compile(r"(?P<bot_info_str>.*)", flags=re.I)
-
- def bot_info(self):
- info_str = f'{self.CurrentConfig._name} '\
- f'{self.CurrentConfig._version}({self.CurrentConfig._svn}) '\
- f'by {self.CurrentConfig._author} '\
- f'on Python {self.CurrentConfig._python_ver_raw} '\
- f'with {" & ".join([adapter + "("+version("iamai-adapter-"+adapter) +")" for adapter in dict(self.bot.config.adapter)])} '\
- f'for iamai({self.CurrentConfig._iamai_version})'
-
- return info_str
-
- async def handle(self) -> None:
- await self.event.reply(
- self.format_str(self.config.message_str, self.bot_info())
- )
diff --git a/HydroRoll/plugins/HydroRoll_plugin_bot/config.py b/HydroRoll/plugins/HydroRoll_plugin_bot/config.py
deleted file mode 100644
index 717b7b2..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_bot/config.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from typing import Set
-
-from plugins.hydroroll_plugin_base import CommandPluginConfig
-
-
-class Config(CommandPluginConfig):
- __config_name__ = "plugin_bot_info"
- command: Set[str] = {"bot"}
- """命令文本。"""
- message_str: str = "{message}"
- """最终发送消息的格式。"""
diff --git a/HydroRoll/plugins/HydroRoll_plugin_dice/__init__.py b/HydroRoll/plugins/HydroRoll_plugin_dice/__init__.py
deleted file mode 100644
index 2c9d2b4..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_dice/__init__.py
+++ /dev/null
@@ -1,51 +0,0 @@
-import re
-import random
-
-from iamai.log import logger
-
-from plugins.hydroroll_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<dice_times>\d+)d(?P<dice_faces>\d+)([*x](?P<dice_multiply>\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/HydroRoll_plugin_dice/config.py b/HydroRoll/plugins/HydroRoll_plugin_dice/config.py
deleted file mode 100644
index 4846409..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_dice/config.py
+++ /dev/null
@@ -1,13 +0,0 @@
-from typing import Set
-
-from plugins.hydroroll_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/HydroRoll_plugin_echo/__init__.py b/HydroRoll/plugins/HydroRoll_plugin_echo/__init__.py
deleted file mode 100644
index e800384..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_echo/__init__.py
+++ /dev/null
@@ -1,17 +0,0 @@
-import re
-
-from plugins.hydroroll_plugin_base import CommandPluginBase
-
-from .config import Config
-
-
-class Echo(CommandPluginBase[None, Config]):
- Config = Config
-
- def __post_init__(self):
- self.re_pattern = re.compile(r"(?P<echo_str>.*)", flags=re.I)
-
- async def handle(self) -> None:
- await self.event.reply(
- self.format_str(self.config.message_str, self.msg_match.group("echo_str"))
- )
diff --git a/HydroRoll/plugins/HydroRoll_plugin_echo/config.py b/HydroRoll/plugins/HydroRoll_plugin_echo/config.py
deleted file mode 100644
index 030bbec..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_echo/config.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from typing import Set
-
-from plugins.hydroroll_plugin_base import CommandPluginConfig
-
-
-class Config(CommandPluginConfig):
- __config_name__ = "plugin_echo"
- command: Set[str] = {"echo"}
- """命令文本。"""
- message_str: str = "*{user_name} {message}"
- """最终发送消息的格式。"""
diff --git a/HydroRoll/plugins/HydroRoll_plugin_luck/__init__.py b/HydroRoll/plugins/HydroRoll_plugin_luck/__init__.py
deleted file mode 100644
index a30c964..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_luck/__init__.py
+++ /dev/null
@@ -1,21 +0,0 @@
-import re
-import time
-import random
-
-from plugins.hydroroll_plugin_base import CommandPluginBase
-
-from .config import Config
-
-
-class Luck(CommandPluginBase[None, Config]):
- Config = Config
-
- def __post_init__(self):
- self.re_pattern = re.compile(r".*", flags=re.I)
-
- async def handle(self) -> None:
- random.seed(
- time.strftime("%Y%j", time.localtime()) + self.format_str("{user_id}")
- )
- lucy = random.randint(self.config.min_int, self.config.max_int)
- await self.event.reply(self.format_str(self.config.message_str, str(lucy)))
diff --git a/HydroRoll/plugins/HydroRoll_plugin_luck/config.py b/HydroRoll/plugins/HydroRoll_plugin_luck/config.py
deleted file mode 100644
index eeef14b..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_luck/config.py
+++ /dev/null
@@ -1,15 +0,0 @@
-from typing import Set
-
-from plugins.hydroroll_plugin_base import CommandPluginConfig
-
-
-class Config(CommandPluginConfig):
- __config_name__ = "plugin_luck"
- command: Set[str] = {"luck"}
- """命令文本。"""
- min_int: int = 0
- """最小随机整数。"""
- max_int: int = 100
- """最大随机整数。"""
- message_str: str = "{user_name}今天的运气是: {message}"
- """最终发送消息的格式。"""
diff --git a/HydroRoll/plugins/HydroRoll_plugin_send/__init__.py b/HydroRoll/plugins/HydroRoll_plugin_send/__init__.py
deleted file mode 100644
index 234e5b3..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_send/__init__.py
+++ /dev/null
@@ -1,28 +0,0 @@
-import re
-
-from plugins.hydroroll_plugin_base import CommandPluginBase
-
-from .config import Config
-
-
-class Send(CommandPluginBase[None, Config]):
- Config = Config
-
- def __post_init__(self):
- self.re_pattern = re.compile(r"\s*(?P<message>.*)", flags=re.I)
-
- async def handle(self) -> None:
- try:
- await self.event.adapter.send(
- self.msg_match.group("message"),
- "private",
- self.config.send_user_id,
- )
- except Exception as e:
- if self.config.send_filed_msg is not None:
- await self.event.reply(
- self.format_str(self.config.send_filed_msg, repr(e))
- )
- else:
- if self.config.send_success_msg is not None:
- await self.event.reply(self.format_str(self.config.send_success_msg))
diff --git a/HydroRoll/plugins/HydroRoll_plugin_send/config.py b/HydroRoll/plugins/HydroRoll_plugin_send/config.py
deleted file mode 100644
index 6903c0f..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_send/config.py
+++ /dev/null
@@ -1,15 +0,0 @@
-from typing import Set, Optional
-
-from plugins.hydroroll_plugin_base import CommandPluginConfig
-
-
-class Config(CommandPluginConfig):
- __config_name__ = "plugin_send"
- command: Set[str] = {"send"}
- """命令文本。"""
- send_user_id: int = 2753364619
- """发送消息的对象的 QQ 号码。"""
- send_success_msg: Optional[str] = "已将消息送出√"
- """发送成功时回复的消息,设置为 None 表示不发送任何消息。"""
- send_filed_msg: Optional[str] = "发送失败:{message}"
- """发送失败时回复的消息。"""
diff --git a/HydroRoll/plugins/HydroRoll_plugin_system/__init__.py b/HydroRoll/plugins/HydroRoll_plugin_system/__init__.py
deleted file mode 100644
index 6e4f232..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_system/__init__.py
+++ /dev/null
@@ -1,75 +0,0 @@
-import re
-from plugins.hydroroll_plugin_base import CommandPluginBase
-from .config import Config
-import psutil
-import time
-from hydroroll.config import GlobalConfig
-from iamai.adapter.cqhttp.message import CQHTTPMessageSegment
-
-class System(CommandPluginBase[None, Config]):
- priority: int = 0
- block: bool = True
- Config = Config
- CurrentConfig = GlobalConfig
-
- def __post_init__(self):
- self.re_pattern = re.compile(r"(?P<system_info_str>.*)", flags=re.I)
-
- def eventReply(self, message: str):
- return self.event.reply(
- self.format_str(self.config.message_str, message)
- )
-
- def get_system_status(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"{self.CurrentConfig._name} Ver.{self.CurrentConfig._version}"
- info_str += f"({self.CurrentConfig._svn}) built in Python {self.CurrentConfig._python_ver}\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 = self.CurrentConfig.HydroSystem()
- try:
- sub_command = self.event.get_plain_text().split()[1]
- except IndexError:
- sub_command = ""
-
- if sub_command in ["status", "s"]:
- await self.event.reply(
- self.format_str(self.config.message_str,
- self.get_system_status())
- )
- elif sub_command in ["restart", "rst"]:
- await self.event.reply(
- self.format_str(self.config.message_str, "正在重启系统...")
- )
- self.bot.restart()
-
- elif sub_command in ["reload", "rld"]:
- await self.event.reply(
- self.format_str(self.config.message_str, "正在重载...")
- )
- self.bot.reload_plugins()
- await self.event.reply(
- self.format_str(self.config.message_str,
- f"已加载{len(self.bot.plugins)}枚插件")
- )
- else:
- await self.event.reply(
- CQHTTPMessageSegment.reply(self.event.message_id) +
- self.format_str(self.config.message_str, system.help)
- )
diff --git a/HydroRoll/plugins/HydroRoll_plugin_system/config.py b/HydroRoll/plugins/HydroRoll_plugin_system/config.py
deleted file mode 100644
index 416fcbe..0000000
--- a/HydroRoll/plugins/HydroRoll_plugin_system/config.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from typing import Set
-
-from plugins.hydroroll_plugin_base import CommandPluginConfig
-
-
-class Config(CommandPluginConfig):
- __config_name__ = "plugin_system_info"
- command: Set[str] = {"system"}
- """命令文本。"""
- message_str: str = "{message}"
- """最终发送消息的格式。"""