<filename>pocorexp.py
#!/usr/bin/env python3

import time
import module.args

from module.other.utils.DataUtil import database
from api.screens import screen
# 设计不完美，导入screen会导入param，param自动初始化，导致reload时没有及时更新param数据


def start(mode):
    # 进入控制台
    if mode == "console":
        from module.console.mainconsole import main_console
        main_console.showConsole()

    # 进入批量利用模式
    if mode == "command":
        from module.command.CommandMode import CommandMode
        start_t = time.time()

        commandMode = CommandMode(inputArg.target, inputArg.command, inputArg.exp, inputArg.thread)
        commandMode.run()

        # print("")
        screen.info(f"Done. using {round(time.time() - start_t)}s")

    # 进入后门模式
    if mode == "webshell":
        from module.webshellmode.webshells import webshellController
        webshellController.showPanel()


def ready(debug: bool, reload: bool):
    if debug:
        screen.setDebug()
        screen.debug("Debug Mode Is On")

    if reload:
        database.reloadPocData()

        from api.parameters import param
        param.reloadPocList()

        screen.success("reload payload list success")


if __name__ == '__main__':
    inputArg = module.args.getArgs()

    ready(inputArg.debug, inputArg.reload)

    start(inputArg.sub_mode)

<filename>pocorexp.py
#!/usr/bin/env python3

import time
import module.args

from module.other.utils.DataUtil import database
from api.screens import screen
# 设计不完美，导入screen会导入param，param自动初始化，导致reload时没有及时更新param数据


def start(mode):
    # 进入控制台
    if mode == "console":
        from module.console.mainconsole import main_console
        main_console.showConsole()

    # 进入批量利用模式
    if mode == "command":
        from module.command.CommandMode import CommandMode
        start_t = time.time()

        commandMode = CommandMode(inputArg.target, inputArg.command, inputArg.exp, inputArg.thread)
        commandMode.run()

        # print("")
        screen.info(f"Done. using {round(time.time() - start_t)}s")

    # 进入后门模式
    if mode == "webshell":
        from module.webshellmode.webshells import webshellController
        webshellController.showPanel()


def ready(debug: bool, reload: bool):
    if debug:
        screen.setDebug()
        screen.debug("Debug Mode Is On")

    if reload:
        database.reloadPocData()

        from api.parameters import param
        param.reloadPocList()

        screen.success("reload payload list success")


if __name__ == '__main__':
    inputArg = module.args.getArgs()

    ready(inputArg.debug, inputArg.reload)

    start(inputArg.sub_mode)

#!/usr/bin/env python3 # 设计不完美，导入screen会导入param，param自动初始化，导致reload时没有及时更新param数据 # 进入控制台 # 进入批量利用模式 # print("") # 进入后门模式