import os
import subprocess

runPath = os.path.abspath(os.path.dirname(__file__))

# 该函数检测是否安装了必要的Python库，如果没有安装，则自动安装
def HaveLib():
    try:
        import wget
        import requests
    except:
        o = open(runPath+"/py_lib.txt")
        print(o.read())
        print(" [ERR] YOU MUST INSTALL THESE LIB.")
        print(" [INFO] 为你自动安装Python库，请等待...")
        # 调用pip安装库，并且检测命令是否执行成功
        if subprocess.call("pip install -r "+runPath+"/py_lib.txt", shell=True) == 0:
            print(" [INFO] 安装成功！")
        else:
            print(" [ERR] 安装失败！请手动安装。")
        o.close()
        exit(1)

if __name__ == '__main__':

    HaveLib()
    # 检测系统是否是windows，如果是windows则运行MainTerminal/main.py，否则运行MainTerminal/main.py
    # linwin-sploit 渗透测试工具2.0支持Windows和Linux系统
    if os.name == 'nt':
        os.system("cd "+runPath+"/MainTerminal/ && python main.py")
    else:
        os.system("cd "+runPath+"/MainTerminal/ && python3 main.py")