WeChatOpenDevTools-Python/main.py at main · stdexception/WeChatOpenDevTools-Python · GitHub

43 lines (32 loc) · 1.41 KB

from argparse import RawTextHelpFormatter

from utils.commons import Commons

from utils.banner import generate_banner

from utils.colors import Color

import argparse

def print_colored_message(message, color):

print(color + message + Color.END)

def main():

HELPALL = """

请选择要执行的方法：

[+] python main.py -h 查看帮助

[+] python main.py -x 开启小程序F12

[+] python main.py -c 开启内置浏览器F12

[+] python main.py -all 开启内置浏览器F12与小程序F12

"""

parser = argparse.ArgumentParser(description=HELPALL, formatter_class=RawTextHelpFormatter)

parser.add_argument('-x', action='store_true', help='开启小程序F12')

parser.add_argument('-c', action='store_true', help='开启内置浏览器F12')

parser.add_argument('-all', action='store_true', help='开启内置浏览器F12与小程序F12')

args = parser.parse_args()

if args.x:

commons.load_wechatEx_configs()

elif args.c:

commons.load_wechatEXE_configs()

elif args.all:

commons.load_wechatEXE_and_wechatEx()

else:

print_colored_message(HELPALL, Color.RED)

if __name__ == "__main__":

generate_banner()

commons = Commons()

main()

Expand file tree