diff options
Diffstat (limited to 'HydroRoll/bot.py')
| -rw-r--r-- | HydroRoll/bot.py | 34 |
1 files changed, 32 insertions, 2 deletions
diff --git a/HydroRoll/bot.py b/HydroRoll/bot.py index 9dcc623..c2b99e6 100644 --- a/HydroRoll/bot.py +++ b/HydroRoll/bot.py @@ -1,7 +1,7 @@ from iamai import Bot as _Bot from typing import Optional, Dict, List, Type, Any, Union from pathlib import Path - +import os # 获取当前目录路径 current_dir = Path.cwd() @@ -26,9 +26,39 @@ class Bot: config_dict=config_dict ) self.bot.load_plugins_from_dirs(Path(f"{script_dir}/plugins")) + self.create_folders() + self.init_data() + self.init_conf() + self.init_webui() def run(self) -> None: self.bot.run() def restart(self) -> None: - self.bot.restart()
\ No newline at end of file + self.bot.restart() + + def create_folders(self): + folder_path = os.path.dirname(os.path.abspath('__file__')) # 获取main.py所在文件夹路径 + if not os.path.isdir(os.path.join(folder_path, 'user')): + os.mkdir(os.path.join(folder_path, 'user')) + if not os.path.isdir(os.path.join(folder_path, 'data')): + os.mkdir(os.path.join(folder_path, 'data')) + if not os.path.isdir(os.path.join(folder_path, 'models')): + os.mkdir(os.path.join(folder_path, 'models')) + if not os.path.isdir(os.path.join(folder_path, 'web')): + os.mkdir(os.path.join(folder_path, 'web')) + if not os.path.isdir(os.path.join(folder_path, 'config')): + os.mkdir(os.path.join(folder_path, 'config')) + if not os.path.isdir(os.path.join(folder_path, 'logs')): + os.mkdir(os.path.join(folder_path, 'logs')) + if not os.path.isdir(os.path.join(folder_path, 'rules')): + os.mkdir(os.path.join(folder_path, 'rules')) + + def init_data(self): + ... + + def init_webui(self): + ... + + def init_conf(self): + ...
\ No newline at end of file |