Update console_main.py

2022-03-31 12:25:29 +08:00
parent 96e023c0b0
commit b73643419e
1 changed files with 4 additions and 1 deletions
--- a/Framework/console_main.py
+++ b/Framework/console_main.py
@@ -46,10 +46,13 @@ def Console():
    if args.fofa:
        try:
            if FOFA_API_KEY:
+                print(args.num)
                if args.num and int(args.num) > 10000:
                    print("Num Don't > 10000 PLS~")
+                elif args.num and int(args.num) <= 10000:
+                    fofaapi.run(args.fofa, args.num)
                else:
-                    fofaapi.run(args.fofa, 1000)
+                    fofaapi.run(args.fofa,1000)
            else:
                print("如需使用FofaAPI，请在Config/config_api下完成相关配置")
        except: