"device_type": "ubiquiti_edge", } net_connect = Netmiko(**edge1) switch = Switch(interpreter.getDeviceName(net_connect.find_prompt())) switch.addVlanList(net_connect.send_command("show vlan")) switch.addPortToPorts(net_connect.send_command("show vlan port all")) while True: mainMenu.printMenu() choice = input("Choose: ") if choice == "1": for vlan in switch.getVlans(): print(vlan.getDescription() + " " + vlan.getNumber()) elif choice == "2": for port in switch.getPorts(): print(port.getPortNr() + " " + port.getPortVlan()) elif choice == "3": switch = Switch(interpreter.getDeviceName(net_connect.find_prompt())) switch.addVlanList(net_connect.send_command("show vlan")) switch.addPortToPorts(net_connect.send_command("show vlan port all")) elif choice == "9": net_connect.disconnect() break else: print("error") # print() # output = net_connect.send_command(command) # print(output) # print()