1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from iamai import Bot as _Bot
from typing import Optional, Dict, List, Type, Any, Union
from pathlib import Path
import os
from hydroroll.config import GlobalConfig
current_dir = Path.cwd()
script_file = current_dir.resolve() / __file__
script_dir = script_file.parent
__all__ = ["Bot"]
class Bot:
def __init__(
self,
*,
config_file: Optional[str] = "config.toml",
config_dict: Optional[Dict] = None,
hot_reload: bool = False,
) -> None:
self.bot = _Bot(hot_reload=hot_reload,
config_file=config_file,
config_dict=config_dict
)
self.bot.load_plugins_from_dirs(Path(f"{script_dir}/plugins"))
self.create_folder_structure(GlobalConfig._folder_dict)
def run(self) -> None:
self.bot.run()
def restart(self) -> None:
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'))
|