QChatGPT/main.py

# QChatGPT 终端启动入口
# 在此层级解决依赖项检查。
# QChatGPT/main.py

asciiart = r"""
  ___   ___ _         _    ___ ___ _____ 
 / _ \ / __| |_  __ _| |_ / __| _ \_   _|
| (_) | (__| ' \/ _` |  _| (_ |  _/ | |  
 \__\_\\___|_||_\__,_|\__|\___|_|   |_|  

⭐️开源地址: https://github.com/RockChinQ/QChatGPT
📖文档地址: https://q.rkcn.top
"""


async def main_entry():
    print(asciiart)

    import sys

    # 检查依赖

    from pkg.core.bootutils import deps

    missing_deps = await deps.check_deps()

    if missing_deps:
        print("以下依赖包未安装，将自动安装，请完成后重启程序：")
        for dep in missing_deps:
            print("-", dep)
        await deps.install_deps(missing_deps)
        print("已自动安装缺失的依赖包，请重启程序。")
        sys.exit(0)

    # 检查命令行
    import os

    if os.name == 'nt':
        import psutil
        allowed_parent_process = ['cmd.exe', 'powershell.exe', 'wsl.exe']

        parent_process = psutil.Process(os.getppid()).name()

        if parent_process not in allowed_parent_process:
            print("请在命令行中运行此程序。")
            input("按任意键退出...")
            exit(0)

    # 检查配置文件

    from pkg.core.bootutils import files

    generated_files = await files.generate_files()

    if generated_files:
        print("以下文件不存在，已自动生成，请按需修改配置文件后重启：")
        for file in generated_files:
            print("-", file)

        sys.exit(0)

    from pkg.core import boot
    await boot.main()


if __name__ == '__main__':
    import os

    # 检查本目录是否有main.py，且包含QChatGPT字符串
    invalid_pwd = False

    if not os.path.exists('main.py'):
        invalid_pwd = True
    else:
        with open('main.py', 'r', encoding='utf-8') as f:
            content = f.read()
            if "QChatGPT/main.py" not in content:
                invalid_pwd = True
    if invalid_pwd:
        print("请在QChatGPT项目根目录下运行此程序。")
        input("按任意键退出...")
        exit(0)

    import asyncio

    asyncio.run(main_entry())