MihoyoBBSTools/main_multi.py

import os
import sys
import main
import time
import tools
import config
import random
import setting
from error import cookieError


# 搜索配置文件
def Fund_config() -> list:
    file_Name = []
    for files in os.listdir(config.path):
        if os.path.splitext(files)[1] == '.json':
            file_Name.append(files)
    return file_Name


def main_multi(autorun: bool):
    tools.log.info("AutoMihoyoBBS Multi User mode")
    tools.log.info("正在搜索配置文件！")
    config_List = Fund_config()
    if len(config_List) == 0:
        tools.log.warning("未检测到配置文件，请确认config文件夹存在.json后缀名的配置文件！")
        exit(1)
    if autorun:
        tools.log.info(f"已搜索到{len(config_List)}个配置文件，正在开始执行！")
    else:
        tools.log.info(f"已搜索到{len(config_List)}个配置文件，请确认是否无多余文件！\r\n{config_List}")
        try:
            input("请输入回车继续，需要重新搜索配置文件请Ctrl+C退出脚本")
        except:
            exit(0)
    results = {"ok":[],"error":[]}
    for i in iter(config_List):
        tools.log.info(f"正在执行{i}")
        setting.mihoyobbs_List_Use = []
        config.config_Path = f"{config.path}/{i}"
        try:
            main.main()
        except cookieError:
            results["error"].append(i)
        else:
            results["ok"].append(i)
        tools.log.info(f"{i}执行完毕")
        time.sleep(random.randint(3, 10))
    print(f'脚本执行完毕，共执行{len(config_List)}个配置文件，成功{len(results["ok"])}个，失败{len(results["error"])}个')


if __name__ == "__main__":
    if (len(sys.argv) >= 2 and sys.argv[1] == "autorun") or os.getenv("AutoMihoyoBBS_autorun") == "1":
        autorun = True
    else:
        autorun = False
    main_multi(autorun)
    exit(0)
pass