Soulter
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
|
||
import os
|
||
import sys
|
||
import warnings
|
||
import traceback
|
||
import threading
|
||
from SparkleLogging.utils.core import LogManager
|
||
from logging import Formatter, Logger
|
||
|
||
warnings.filterwarnings("ignore")
|
||
abs_path = os.path.dirname(os.path.realpath(sys.argv[0])) + '/'
|
||
|
||
logger: Logger = None
|
||
|
||
logo_tmpl = """
|
||
___ _______.___________..______ .______ ______ .___________.
|
||
/ \ / | || _ \ | _ \ / __ \ | |
|
||
/ ^ \ | (----`---| |----`| |_) | | |_) | | | | | `---| |----`
|
||
/ /_\ \ \ \ | | | / | _ < | | | | | |
|
||
/ _____ \ .----) | | | | |\ \----.| |_) | | `--' | | |
|
||
/__/ \__\ |_______/ |__| | _| `._____||______/ \______/ |__|
|
||
|
||
"""
|
||
|
||
def make_necessary_dirs():
|
||
os.makedirs("data/config", exist_ok=True)
|
||
os.makedirs("temp", exist_ok=True)
|
||
|
||
def main():
|
||
logger = LogManager.GetLogger(
|
||
log_name='astrbot-core',
|
||
out_to_console=True,
|
||
# HTTPpost_url='http://localhost:6185/api/log',
|
||
# http_mode = True,
|
||
custom_formatter=Formatter('[%(asctime)s| %(name)s - %(levelname)s|%(filename)s:%(lineno)d]: %(message)s', datefmt="%H:%M:%S")
|
||
)
|
||
logger.info(logo_tmpl)
|
||
# config.yaml 配置文件加载和环境确认
|
||
try:
|
||
import botpy, logging, yaml
|
||
import cores.astrbot.core as qqBot
|
||
# delete qqbotpy's logger
|
||
for handler in logging.root.handlers[:]:
|
||
logging.root.removeHandler(handler)
|
||
ymlfile = open(abs_path+"configs/config.yaml", 'r', encoding='utf-8')
|
||
cfg = yaml.safe_load(ymlfile)
|
||
except ImportError as import_error:
|
||
logger.error(import_error)
|
||
input("检测到一些依赖库没有安装。请先安装，然后重试。")
|
||
exit()
|
||
except FileNotFoundError as file_not_found:
|
||
logger.error(file_not_found)
|
||
input("配置文件不存在，请检查是否已经下载配置文件。")
|
||
exit()
|
||
except BaseException as e:
|
||
raise e
|
||
|
||
# 设置代理
|
||
if 'http_proxy' in cfg and cfg['http_proxy'] != '':
|
||
os.environ['HTTP_PROXY'] = cfg['http_proxy']
|
||
if 'https_proxy' in cfg and cfg['https_proxy'] != '':
|
||
os.environ['HTTPS_PROXY'] = cfg['https_proxy']
|
||
os.environ['NO_PROXY'] = 'https://api.sgroup.qq.com'
|
||
|
||
make_necessary_dirs()
|
||
|
||
# 启动主程序（cores/qqbot/core.py）
|
||
qqBot.init(cfg)
|
||
|
||
|
||
def check_env():
|
||
if not (sys.version_info.major == 3 and sys.version_info.minor >= 9):
|
||
logger.error("请使用 Python3.9+ 运行本项目。按任意键退出。")
|
||
input("")
|
||
exit()
|
||
|
||
if __name__ == "__main__":
|
||
check_env()
|
||
t = threading.Thread(target=main, daemon=True)
|
||
t.start()
|
||
try:
|
||
t.join()
|
||
except KeyboardInterrupt as e:
|
||
logger.info("退出 AstrBot。")
|
||
exit()
|