Merge pull request #5 from luizmlo/1.x

Install.sh and reopen() function

README.md

@@ -14,14 +14,11 @@ A Python tool used to automate the execution of the following tools : Nmap , Nik
 ## 🛠 Installation
 ### Linux & Unix
 ```
-$ sudo apt-get install gnome-terminal
-$ sudo apt-get install nmap
-$ sudo apt-get install nikto
-$ cd /opt && sudo git clone https://github.com/maurosoria/dirsearch.git
 $ git clone https://github.com/Anteste/WebMap.git
 $ cd WebMap
-$ sudo chmod +x webmap.py
-$ python3 webmap.py
+$ chmod +x install.sh
+$ sudo ./install.sh
+$ sudo webmap
 ```
 
 # ✨ Other 

install.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/sh
+
+sudo apt-get install gnome-terminal -y
+sudo apt-get install nmap -y
+sudo apt-get install nikto -y
+sudo chmod +x webmap.py
+sudo cp webmap.py /bin/webmap
+

webmap.py

@@ -3,8 +3,19 @@
 import os
 import sys
 import socket
+import string
 
 ans=True
+
+installed = True if os.path.exists('/bin/webmap') else False
+
+def reopen():
+    if installed:
+        os.system('sudo webapp')
+
+    else:
+        os.system('sudo python3 webmap.py')
+
 while ans:
     print
     print('====================================')
@@ -17,7 +28,7 @@
     print("G. Get The Tools")
     print("E. Exit")
     print('====================================')
-    ans=input("What would you like to do? Enter your selection: ")
+    ans = input("What would you like to do? Enter your selection: ").upper()
 
     if ans=="1":
       os.system('cls' if os.name == 'nt' else 'clear')
@@ -46,6 +57,7 @@
                 print("______________________________________________________________________")
                 os.system("nmap " + portipscan)
                 print("______________________________________________________________________")
+
           elif ans=="2":
                 os.system('cls' if os.name == 'nt' else 'clear')
                 print
@@ -68,6 +80,7 @@
                   print("______________________________________________________________________")
                   os.system("nmap -A " + aipscan)
                   print("______________________________________________________________________")
+
           elif ans=="4":
                 os.system('cls' if os.name == 'nt' else 'clear')
                 print
@@ -79,17 +92,19 @@
                 print("______________________________________________________________________")
                 os.system("nmap -sn " + snipscan)
                 print("______________________________________________________________________")
+
           elif ans=="M":
                   os.system('cls' if os.name == 'nt' else 'clear')
-                  os.system('python3 webmap.py')
+                  reopen()
 
           else:
                   os.system('cls' if os.name == 'nt' else 'clear')
                   print
                   print("Not Valid Choice Try again")
                   print
-                  os.system("python3 webmap.py")
+                  reopen()
                   ans = None
+
     elif ans=="2":
       os.system('cls' if os.name == 'nt' else 'clear')
       print
@@ -104,6 +119,7 @@
       print("______________________________________________________________________")
       if ans=="":
           os.system('cls' if os.name == 'nt' else 'clear')
+
     elif ans=="3":
       os.system('cls' if os.name == 'nt' else 'clear')
       print
@@ -174,14 +190,15 @@
 
           elif ans=="M":
               os.system('cls' if os.name == 'nt' else 'clear')
-              os.system('python3 webmap.py')
+              reopen()
           else:
                 os.system('cls' if os.name == 'nt' else 'clear')
                 print
                 print("Not Valid Choice Try again")
                 print
-                os.system("python3 webmap.py")
+                reopen()
                 ans = None
+
     elif ans=="E":
         os.system('cls' if os.name == 'nt' else 'clear')
         ans = None
@@ -190,4 +207,4 @@
     	print
     	print("Not Valid Choice Try again")
     	print
-    	os.system("python3 webmap.py")
+    	reopen()