fix: 工作目录必须在 main.py 目录

This commit is contained in:
RockChinQ 2024-03-30 21:34:22 +08:00
parent 5a3e7fe8ee
commit 2e9229a6ad

16
main.py
View File

@ -1,5 +1,6 @@
# QChatGPT 终端启动入口 # QChatGPT 终端启动入口
# 在此层级解决依赖项检查。 # 在此层级解决依赖项检查。
# QChatGPT/main.py
asciiart = r""" asciiart = r"""
___ ___ _ _ ___ ___ _____ ___ ___ _ _ ___ ___ _____
@ -62,6 +63,21 @@ if __name__ == '__main__':
input("按任意键退出...") input("按任意键退出...")
exit(0) exit(0)
# 检查本目录是否有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 import asyncio
asyncio.run(main_entry()) asyncio.run(main_entry())