def show_online_devices(self):
print_normal('Show the ' +
UseStyle('ONLINE', fore='green', mode='underline') +
' devices.')
devices_config_obj = DeviceConfigs.initialize(
mars_config=self.mars_config)
for device_config in devices_config_obj.get_data():
print_normal(device_to_line_string(device_config))
def show_snap_devices(self, snap_time):
snap_time_str = format_time_string_2_number(snap_time)
print_normal('Show the ' +
UseStyle(snap_time, fore='green', mode='underline') +
' devices.')
devices_config_obj = DeviceConfigs.initialize_with(
self.mars_config,
get_devices_configs(self.mars_config, snap_time_str))
for device_config in devices_config_obj.get_data():
print_normal(device_to_line_string(device_config))
def show_online_links(self):
print_normal('Show the ' +
UseStyle('ONLINE', fore='green', mode='underline') +
' links.')
link_obj = Links.initialize(mars_config=self.mars_config)
devices_config_obj = DeviceConfigs.initialize(
mars_config=self.mars_config)
for device_config in devices_config_obj.get_data():
print_normal_start('Device Name : ' + device_config['name'],
color='yellow')
for link in link_obj.get_data():
if link['src']['device'] == device_config['id']:
print_normal_center(
link_to_line_string(link, devices_config_obj))
print_normal_end('')
print_normal('')
def show_online_hosts(self):
print_normal('Show the ' +
UseStyle('ONLINE', fore='green', mode='underline') +
' hosts.')
host_obj = Hosts.initialize(mars_config=self.mars_config)
devices_config_obj = DeviceConfigs.initialize(
mars_config=self.mars_config)
for device_config in devices_config_obj.get_data():
print_normal_start('Device Name : ' + device_config['name'],
color='yellow')
for host in host_obj.get_data():
for location in host['locations']:
if location['elementId'] == device_config['id']:
print_normal_center(
host_to_line_string(host, devices_config_obj))
print_normal_end('')
print_normal('')
def show_snap_links(self, snap_time_str):
snap_time = format_time_string_2_number(snap_time_str)
print_normal('Show the ' +
UseStyle(snap_time_str, fore='green', mode='underline') +
' links.')
link_obj = Links.initialize_with(self.mars_config,
get_link(self.mars_config, snap_time))
devices_config_obj = DeviceConfigs.initialize_with(
self.mars_config, get_devices_configs(self.mars_config, snap_time))
for device_config in devices_config_obj.get_data():
print_normal_start('Device Name : ' + device_config['name'],
color='yellow')
for link in link_obj.get_data():
if link['src']['device'] == device_config['id']:
print_normal_center(
link_to_line_string(link, devices_config_obj))
print_normal_end('')
print_normal('')
def show_snap_hosts(self, snap_time_str):
print_normal('Show the ' +
UseStyle(snap_time_str, fore='green', mode='underline') +
' hosts.')
snap_time = format_time_string_2_number(snap_time_str)
host_obj = Hosts.initialize_with(self.mars_config,
get_host(self.mars_config, snap_time))
devices_config_obj = DeviceConfigs.initialize_with(
self.mars_config, get_devices_configs(self.mars_config, snap_time))
for device_config in devices_config_obj.get_data():
print_normal_start('Device Name : ' + device_config['name'],
color='yellow')
for host in host_obj.get_data():
for location in host['locations']:
if location['elementId'] == device_config['id']:
print_normal_center(
host_to_line_string(host, devices_config_obj))
print_normal_end('')
print_normal('')
def snap_all_data(self):
print_normal('Going to snap all the data from ' +
self.mars_config.get_url())
device_object = Devices.initialize(self.mars_config)
device_config_object = DeviceConfigs.initialize(self.mars_config)
host_object = Hosts.initialize(self.mars_config)
flow_object = Flows.initialize(self.mars_config)
group_object = Groups.initialize(self.mars_config)
link_object = Links.initialize(self.mars_config)
t = int(time.time())
data_path = self.mars_config.get_base_path() + '/' + str(t)
os.mkdir(data_path)
device_object.save(data_path)
device_config_object.save(data_path)
host_object.save(data_path)
flow_object.save(data_path)
group_object.save(data_path)
link_object.save(data_path)
print_normal('Success to snap the data')