tools_main.py

# Import necessary modules
import subprocess
import tools


# Function for displaying menu
def Display_Menu():
    header = """
    ___________________________
    |                          |
    |  Basic Networking Tools  |
    |_____________   __________|
                  \  \                 
                    ^_ _^              
                    (o o)\_________    
                    (_ _)\         )/\/
                      U   ||----W||    
                          ||     ||    

    ### Available Tools:
        [1] Locate an IP address
        [2] Get the IP address of a domain
        [3] Ping a host or IP address
        [4] Port Scanner
        [5] Exit
    
    * Type <clear> to clear previous commands
    * Type <q> or to abort operation of a tool

    """
    print(header)


# Functionality of every option
# Each option operates the tool related to the option number.
def option_1():
    """
    Option 1 --> IP Locater
    """
    tools.locate_ip()
    Display_Menu()
    Home()


def option_2():
    """
    Option 2 --> IP Finder
    """
    tools.get_ip()
    Display_Menu()
    Home()


def option_3():
    """
    Option 3 --> Pinging
    """
    tools.ping()
    Display_Menu()
    Home()


def option_4():
    """
    Option 4 --> Port Scanner
    """
    tools.port_scanner()
    Display_Menu()
    Home()


def option_5():
    """
    Option 5 --> Exit
    """
    quit()


# Function for selecting options
def Home():
    """
    Home of the program is selecting an option from available options.
    """
    # Available options' numbers as shown in the menu
    available_options = (1, 2, 3, 4, 5)
    # Get the option's number.
    # If it is wrong, this while loop will run unless it gets
    # a proper number that is related with one of the available options.
    while True:
        try:
            selected_option = input("\nEnter your option\n>>> ")
            # In order to clear the mess created by the previous commands,
            # we define 'clear' command to clear up the screen from previous commands.
            if selected_option == 'clear':
                subprocess.call('cls', shell=True)
                Display_Menu()
                continue
            else:
                selected_option = int(selected_option)
        except ValueError:
            print("Please enter the number of the option")
            continue
        else:
            # If the given option number is not available, try to get an available option number.
            if selected_option not in available_options:
                print("The option is not available.\nTry another one.")
                continue
            else:
                break
    # Operate the option related to the given number
    if selected_option == 1:
        option_1()
    elif selected_option == 2:
        option_2()
    elif selected_option == 3:
        option_3()
    elif selected_option == 4:
        option_4()
    elif selected_option == 5:
        option_5()


if __name__ == "__main__":
    Display_Menu()
    Home()