From 9bf222471d34cb756a4878b103ec82c6c4bfb191 Mon Sep 17 00:00:00 2001 From: Kyomotoi <1172294279@qq.com> Date: Sun, 20 Dec 2020 17:14:41 +0800 Subject: [Update] MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 创建项目结构 --- check.py | 111 --------------------------------------------------------------- 1 file changed, 111 deletions(-) delete mode 100644 check.py (limited to 'check.py') diff --git a/check.py b/check.py deleted file mode 100644 index cb98011..0000000 --- a/check.py +++ /dev/null @@ -1,111 +0,0 @@ -#!/usr/bin/env python3 -# -*- encoding: utf-8 -*- -''' -@File : check.py -@Time : 2020/11/07 14:30:34 -@Author : Kyomotoi -@Contact : kyomotoiowo@gmail.com -@Github : https://github.com/Kyomotoi -@License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved. -''' -__author__ = 'kyomotoi' - -import os -import sys -import time -import pkg_resources - -rely_list = ''.join(f'{rely}' for rely in pkg_resources.working_set) -need_rely_list = ['pathlib', 'pyyaml', 'rich'] -for rely in need_rely_list: - if rely not in rely_list: - os.system(f'pip3 install {rely}') - -from pathlib import Path -from rich.progress import Progress -from ATRI.utils.utils_yml import load_yaml - -CONFIG_PATH = Path('.') / 'config.yml' -config = load_yaml(CONFIG_PATH) - - -class CheckATRI(): - """运行前检查必要条件""" - def checkConfig(self) -> None: - """检查配置文件是否填写完整""" - len_config = len(config) + len(config['bot']) + len( - config['api']) + len(config['html']) - - with Progress() as progress: - task = 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, 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, advance=1) - time.sleep(0.1) - - # 检查接口配置 - api = config['api'] - for key in api: - if not api[key]: - print(f"Can't load [{key}] from config.yml") - time.sleep(5) - sys.exit(0) - else: - progress.update(task, advance=1) - time.sleep(0.1) - - # 检查网页配置 - html = config['html'] - for key in html: - if not html[key]: - print(f"Can't load [{key}] from config.yml") - time.sleep(5) - sys.exit(0) - else: - progress.update(task, advance=1) - time.sleep(0.1) - - def checkRely(self) -> None: - ''' - 检查依赖是否完整 - 如不完整自动安装 - 别吐槽 暴 力 - ''' - rely_list = [ - 'nonebot2', 'nonebot2[scheduler]', 'nonebot_plugin_apscheduler', 'nltk', - 'requests', 'pillow', 'psutil' - ] - rely_len = len(rely_list) - - with Progress() as progress: - task = progress.add_task("[cyan]Checking Rely...", total=rely_len) - - while not progress.finished: - for rely in rely_list: - if rely not in rely_list: - try: - os.system(f'pip3 install {rely}') - except: - print( - "Can't install package {rely}. Please use Google/Bing search error repo and fix it by yourself." - ) - time.sleep(5) - sys.exit(0) - progress.update(task, advance=1) - time.sleep(0.1) -- cgit v1.2.3