Banner Updated

bhavsec · Apr 8, 2020 · 56461da · 56461da
1 parent 60915a1
commit 56461da
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 22 deletions.
diff --git a/core/repl_prompt.py b/core/repl_prompt.py
@@ -17,18 +17,20 @@
 from plugins.Username import user
 from core.updater import update
 from prompt_toolkit import prompt
+from reconspider import menu
 
 
 def repl():  # Read–eval–print loop
     while 1:
+        print(menu())
         user_input = prompt("\nReconspider >> ")
         if len(user_input)==0:
-            print("ENTER 1 - 13 TO SELECT OPTIONS")
+            print("\n")
             continue
         try:
             choice = int(user_input)
         except ValueError:
-            print("ENTER 1 - 13 TO SELECT OPTIONS")
+            print("\n")
             continue
 
         if choice == 1:
@@ -42,9 +44,17 @@ def repl():  # Read–eval–print loop
         elif choice == 2:
             while 1:
                 host = input("HOST (URL / IP) >> ")
-                port = int(input("PORT >> "))
+                port = input("PORT >> ")
+                try:
+                    if port == "":
+                        port=80
+                    elif port not in [80,443]:
+                        print("Invalid port - Available(80,443)")
+                        continue
+                except ValueError:
+                    port=80
                 break
-            domain(host,port)
+            domain(host,int(port))
             continue
 
         elif choice == 3:

diff --git a/reconspider.py b/reconspider.py
@@ -1,17 +1,17 @@
 import sys
 
 def banner():
-       return("""
+       return ("""
 __________                               _________       __     ___
 \______   \ ____   ____  ____   ____    /   _____/_____ |__| __| _/___________
  |       _// __ \_/ ___\/  _ \ /    \   \_____  \\\____ \|  |/ __ |/ __ \_  __ \\
  |    |   \  ___/\  \__(  <_> )   |  \  /        \  |_> >  / /_/ \  ___/|  | \/
  |____|_  /\___  >\___  >____/|___|  / /_______  /   __/|__\____ |\___  >__|
         \/     \/     \/           \/          \/|__|           \/    \/
 
-                      developer: https://bhavkaran.com
-
-
+                      developer: https://bhavkaran.com""")
+def menu():
+    return ("""
 ENTER 0 - 13 TO SELECT OPTIONS
 
 1.  IP                           Enumerate  information  from  IP Address
@@ -32,17 +32,17 @@ def banner():
 0. EXIT                         Exit from  ReconSpider  to your terminal
        """)
 
-if sys.version_info[0] > 2:
-       try:
-              print(banner())
-              from core import repl_prompt
-       except ModuleNotFoundError:
-              print('\nSeems like you haven\'t installed Requirements, Please install using: python setup.py install')
-              quit()
-else:
-       try:
-              print(banner())
-              from core import repl_prompt
-       except ImportError:
-              print('\nSeems like you haven\'t installed Requirements, Please install using: python setup.py install')
-              quit()
+if __name__ == '__main__':
+    if sys.version_info[0] > 2:
+           try:
+                  print(banner())
+                  from core import repl_prompt
+           except ModuleNotFoundError:
+                  print('\nSeems like you haven\'t installed Requirements, Please install using: python setup.py install')
+                  quit()
+    else:
+           try:
+                  from core import repl_prompt
+           except ImportError:
+                  print('\nSeems like you haven\'t installed Requirements, Please install using: python setup.py install')
+                  quit()