106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
|
import click
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
from ndct.core.device import Device, devices
|
||
|
|
from ndct.core.log import log
|
||
|
|
from ndct.core.paths import DB_PATH
|
||
|
|
|
||
|
|
@click.command(short_help = 'Add a device')
|
||
|
|
@click.option('-n', '--name', help = 'Name', required = True)
|
||
|
|
@click.option('-i', '--ip', help = 'IP address', required = True)
|
||
|
|
@click.option('-u', '--username', help = 'Username to authenticate against', required = True)
|
||
|
|
@click.option('-p', '--password', help = 'Password to authenticate with', required = True)
|
||
|
|
@click.option('-o', '--os', type = click.Choice(['cisco_ios', 'vyos']), help = 'Operating system', required = True)
|
||
|
|
def add(name, ip, username, password, os):
|
||
|
|
'''
|
||
|
|
Summary:
|
||
|
|
Adds a device.
|
||
|
|
|
||
|
|
Takes:
|
||
|
|
name: Name of device
|
||
|
|
ip: Management IP address of device
|
||
|
|
username: Username to authenticate against
|
||
|
|
password: Password to authenticate with
|
||
|
|
os: Operating system of device cisco_ios|vyos
|
||
|
|
'''
|
||
|
|
Device.get_devices_from_file()
|
||
|
|
for device in devices:
|
||
|
|
if name in device:
|
||
|
|
log('[{}] Device already exists'.format(name), 'info')
|
||
|
|
sys.exit(1)
|
||
|
|
device_object = Device(name, ip, username, password, os)
|
||
|
|
devices.append({name: device_object})
|
||
|
|
log('[{}] Device added successfully'.format(name), 'info')
|
||
|
|
Device.save_devices_to_file()
|
||
|
|
|
||
|
|
@click.command(short_help = 'Add devices from a file')
|
||
|
|
@click.option('-f', '--filename', help = 'File to add devices from', required = True)
|
||
|
|
def add_from_file(filename):
|
||
|
|
Device.get_devices_from_file()
|
||
|
|
file_path = DB_PATH + filename
|
||
|
|
if os.path.isfile(file_path):
|
||
|
|
log("Adding devices from '{}'".format(file_path), 'info')
|
||
|
|
with open(file_path, 'r') as devices_file:
|
||
|
|
all_lines = [line.strip() for line in devices_file.readlines()]
|
||
|
|
for device_attribute in range(0, len(all_lines), 5):
|
||
|
|
device_exists = False
|
||
|
|
for device in devices:
|
||
|
|
if all_lines[device_attribute] in device:
|
||
|
|
log('[{}] Device already exists'.format(all_lines[device_attribute]), 'info')
|
||
|
|
device_exists = True
|
||
|
|
if device_exists == False:
|
||
|
|
device_object = Device(all_lines[device_attribute], all_lines[device_attribute+1], all_lines[device_attribute+2], all_lines[device_attribute+3], all_lines[device_attribute+4])
|
||
|
|
devices.append({all_lines[device_attribute]: device_object})
|
||
|
|
log('[{}] Device added successfully'.format(all_lines[device_attribute]), 'info')
|
||
|
|
Device.save_devices_to_file()
|
||
|
|
|
||
|
|
@click.command(short_help = 'Remove a device')
|
||
|
|
@click.option('-n', '--name', help = 'Name', required = True)
|
||
|
|
def remove(name):
|
||
|
|
'''
|
||
|
|
Summary:
|
||
|
|
Removes a device.
|
||
|
|
|
||
|
|
Takes:
|
||
|
|
name: Name of device to remove
|
||
|
|
'''
|
||
|
|
Device.get_devices_from_file()
|
||
|
|
for device in devices:
|
||
|
|
if name in device:
|
||
|
|
devices.remove(device)
|
||
|
|
log('[{}] Device removed successfully'.format(name), 'info')
|
||
|
|
Device.save_devices_to_file()
|
||
|
|
return
|
||
|
|
|
||
|
|
log('[{}] Device does not exist'.format(name), 'error')
|
||
|
|
|
||
|
|
@click.command(short_help = 'View a device')
|
||
|
|
@click.option('-n', '--name', help = 'Name', required = True)
|
||
|
|
def view(name):
|
||
|
|
'''
|
||
|
|
Summary:
|
||
|
|
Prints attributes of a Device instance.
|
||
|
|
|
||
|
|
Takes:
|
||
|
|
name: Name of device to view information about
|
||
|
|
'''
|
||
|
|
Device.get_devices_from_file()
|
||
|
|
device_information = Device.get_device_information(name)
|
||
|
|
if device_information != None:
|
||
|
|
log('Name: ' + str(device_information['name']), 'info')
|
||
|
|
log('IP: ' + str(device_information['ip']), 'info')
|
||
|
|
log('Username: ' + str(device_information['username']), 'info')
|
||
|
|
log('Password: ' + str(device_information['password']), 'info')
|
||
|
|
log('OS: ' + str(device_information['os']), 'info')
|
||
|
|
else:
|
||
|
|
log('[{}] Device does not exist'.format(name), 'error')
|
||
|
|
|
||
|
|
@click.group(short_help = 'Device commands')
|
||
|
|
def device():
|
||
|
|
pass
|
||
|
|
|
||
|
|
device.add_command(add)
|
||
|
|
device.add_command(add_from_file)
|
||
|
|
device.add_command(remove)
|
||
|
|
device.add_command(view)
|