summaryrefslogtreecommitdiff
path: root/ATRI/config.py
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2021-02-06 00:32:26 +0800
committerKyomotoi <[email protected]>2021-02-06 00:32:26 +0800
commitf5ceb8927f2e7f2a9e29d62c8e4cef876f917249 (patch)
tree40b9dcd6b7d3db486054e3aa9b5a04d25fa2284e /ATRI/config.py
parenteb52fab79ada7efe6191e3a5f90179766feaded0 (diff)
downloadATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.tar.gz
ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.tar.bz2
ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.zip
๐Ÿ— ๐Ÿ’ฉ ๆ›ดๆ”น้กน็›ฎ็ป“ๆž„๏ผŒไฟฎๅคๅ•ฅb BUG
Diffstat (limited to 'ATRI/config.py')
-rw-r--r--ATRI/config.py86
1 files changed, 23 insertions, 63 deletions
diff --git a/ATRI/config.py b/ATRI/config.py
index ea9bbff..4a8ed10 100644
--- a/ATRI/config.py
+++ b/ATRI/config.py
@@ -1,43 +1,14 @@
-import sys
-import time
from pathlib import Path
from datetime import timedelta
from ipaddress import IPv4Address
-from rich.progress import Progress
-from .log import logger
-from .utils import load_yaml
+from .utils.yaml import load_yml
-CONFIG_PATH = Path('.') / 'ATRI' / 'config' / 'main.config.yml'
-config = load_yaml(CONFIG_PATH)
+CONFIG_PATH = Path('.') / 'config.yml'
+config = load_yml(CONFIG_PATH)
+nonebot_config = config['BotSelfConfig']
-def check_config() -> None:
- logger.info('Please stand by, now in checking type.')
-
- len_config = len(config) + len(config['bot'])
-
- with Progress() as progress:
- task_c = progress.add_task("[cyan]Checking config...",
- total=len_config)
-
- while not progress.finished:
- bot = config['bot']
- for key in bot:
- if key == 'debug':
- if bot['debug'] != 0:
- print('DEBUG now is open.')
- progress.update(task_c, advance=1)
- time.sleep(0.1)
- else:
- if not bot[key]:
- print(f"Can't load [{key}] from config.yml")
- time.sleep(5)
- sys.exit(0)
- else:
- progress.update(task_c, advance=1)
- time.sleep(0.1)
-
COPYRIGHT = """
โ–ˆโ–ˆโ–ˆโ–ˆโ–ˆ โ–ˆโ–ˆโ–ˆโ–ˆโ–ˆโ–ˆโ–ˆโ–ˆ โ–ˆโ–ˆโ–ˆโ–ˆโ–ˆโ–ˆ โ–ˆโ–ˆ
@@ -50,33 +21,22 @@ Copyright ยฉ 2021 Kyomotoi, All Rights Reserved.
Project: https://github.com/Kyomotoi/ATRI
"""
-VERSION = config['version']
-
-RUNTIME_CONFIG: dict = {
- 'host': IPv4Address(config['bot'].get('host', '127.0.0.1')),
- 'port': int(config['bot'].get('port', 8080)),
- 'debug': bool(config['bot'].get('debug', False)),
- 'superusers': config['bot'].get('superusers', [1234567890]),
- 'nickname': set(
- config['bot'].get(
- 'nickname', ['ATRI', 'Atri', 'atri', 'ไบšๆ‰˜่Ž‰', 'ใ‚ขใƒˆใƒช'])),
- 'command_start': set(config['bot'].get('command_start', ['', '/'])),
- 'command_sep': set(config['bot'].get('command_sep', ['.'])),
- 'session_expire_timeout': timedelta(
- config['bot'].get('session_expire_timeout', 2))
- }
-
-PLUGIN_BOT_CONFIG = Path('.') / 'ATRI' / 'config' / 'character.plugin.yml'
-BOT_CONFIG: dict = load_yaml(PLUGIN_BOT_CONFIG)
-
-PLUGIN_HITOKOTO_CONFIG = Path('.') / 'ATRI' / 'config' / 'hitokoto.plugin.yml'
-HITOKOTO_CONFIG: dict = load_yaml(PLUGIN_HITOKOTO_CONFIG)
-
-PLUGIN_UTILS_CONFIG = Path('.') / 'ATRI' / 'config' / 'utils.plugin.yml'
-UTILS_CONFIG: dict = load_yaml(PLUGIN_UTILS_CONFIG)
-
-PLUGIN_CURSE_CONFIG = Path('.') / 'ATRI' / 'config' / 'curse.plugin.yml'
-CURSE_CONFIG: dict = load_yaml(PLUGIN_CURSE_CONFIG)
-
-PLUGIN_SETU_CONFIG = Path('.') / 'ATRI' / 'config' / 'setu.plguin.yml'
-SETU_CONFIG: dict = load_yaml(PLUGIN_SETU_CONFIG)
+VERSION = "YHN-001-A01"
+
+RUNTIME_CONFIG = {
+ "host": IPv4Address(nonebot_config.get('host', '127.0.0.1')),
+ "port": int(nonebot_config.get('port', '8080')),
+ "debug": bool(nonebot_config.get('debug', False)),
+ "superusers": set(nonebot_config.get('superusers', ["1234567890"])),
+ "nickname": set(
+ nonebot_config.get(
+ 'nickname',
+ ['ATRI', 'Atri', 'atri', 'ไบšๆ‰˜่Ž‰', 'ใ‚ขใƒˆใƒช']
+ )
+ ),
+ "command_start": set(nonebot_config.get('command_start', ['', '/'])),
+ "command_sep": set(nonebot_config.get('command_sep', ['.'])),
+ "session_expire_timeout": timedelta(
+ nonebot_config.get('session_expire_timeout', 2)
+ )
+}