diff options
author | Kyomotoi <[email protected]> | 2021-02-06 00:32:26 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2021-02-06 00:32:26 +0800 |
commit | f5ceb8927f2e7f2a9e29d62c8e4cef876f917249 (patch) | |
tree | 40b9dcd6b7d3db486054e3aa9b5a04d25fa2284e /ATRI/config.py | |
parent | eb52fab79ada7efe6191e3a5f90179766feaded0 (diff) | |
download | ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.tar.gz ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.tar.bz2 ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.zip |
๐ ๐ฉ ๆดๆน้กน็ฎ็ปๆ๏ผไฟฎๅคๅฅb BUG
Diffstat (limited to 'ATRI/config.py')
-rw-r--r-- | ATRI/config.py | 86 |
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) + ) +} |