NetworkTroubleshootingSimulator/Main2.py at main · Kallnilious/NetworkTroubleshootingSimulator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
from Node import Node
from NetworkTechnician import NetworkTechnician
from NetworkTopology import NetworkTopology
from ProgramAdmin import ProgramAdmin

def main():
    # Initialize network topology
    network = NetworkTopology()

    # Add nodes
    network.add_node("192.168.1.1", "Router")
    network.add_node("192.168.1.2", "PC")
    network.add_node("192.168.1.3", "Server")

    # Connect nodes
    network.connect_nodes("192.168.1.1", "192.168.1.2")
    network.connect_nodes("192.168.1.1", "192.168.1.3")

    # Initialize program admin and technician
    program_admin = ProgramAdmin(network)
    network_technician = NetworkTechnician(network)

    while True:
        print("\nSelect mode:")
        print("1. Program Admin")
        print("2. Network Technician")
        print("3. Exit")
        choice = input("Enter choice: ")

        if choice == "1":
            command = input("Enter admin command (e.g., '192.168.1.2 down'): ")
            program_admin.simulate_problem(command)

        elif choice == "2":
            print("\nNetwork Technician Mode active")
            print("Type /help for help or /help <tool name> for help about a particular tool.")

            tool_choice = input("")
            print("command was: " + tool_choice)

            if tool_choice == "/ping"

          #  print("\nSelect tool:")
          #  print("1. Ping")
          #  print("2. Tracert")
          #  print("3. NSLookup")
          #  print("4. Check Service")
          #  tool_choice = input("Enter choice: ")
          #
          #  if tool_choice == "1":
          #      ip = input("Enter IP address to ping: ")
          #      network_technician.ping(ip)
          #  elif tool_choice == "2":
          #      ip = input("Enter IP address to tracert: ")
          #      network_technician.tracert(ip)
          #  elif tool_choice == "3":
          #      domain = input("Enter domain name for NSLookup: ")
          #      network_technician.nslookup(domain)
          #  elif tool_choice == "4":
          #      ip = input("Enter IP address to check service: ")
          #      service = input("Enter service name: ")
          #      network_technician.check_service(ip, service)

        elif choice == "3":
            print("Exiting program.")
            break

        else:
            print("Invalid choice, please try again.")


if __name__ == "__main__":
    main()