本文整理汇总了Python中AutoNetkit.plot_paths方法的典型用法代码示例。如果您正苦于以下问题：Python AutoNetkit.plot_paths方法的具体用法？Python AutoNetkit.plot_paths怎么用？Python AutoNetkit.plot_paths使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类AutoNetkit的用法示例。
在下文中一共展示了AutoNetkit.plot_paths方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: traceroute
# 需要导入模块: import AutoNetkit [as 别名]
# 或者: from AutoNetkit import plot_paths [as 别名]
def traceroute(self, src_dst_list):
LOG.info("Performing Netkit traceroutes for %s" % self.server.host)
shell = self.server.get_shell()
shell.setecho(False)
resultant_paths = defaultdict(dict) #indexed format for presenting/analysis
paths = [] # list format for plotting
template_dir = resource_filename("AutoNetkit","lib/templates")
linux_traceroute_template = open(os.path.join(template_dir, "textfsm", "linux_traceroute"), "r")
# build reverse-IP lookup
#TODO: move this into seperate helper function
ip_mappings = {}
#TODO: see if netaddr has a better way other than casting to string
ip_mappings.update( dict( (str(device.lo_ip.ip), device)
for device in self.network.devices()))
for link in self.network.links():
ip_mappings[str(link.local_ip)] = link.local_host
ip_mappings[str(link.remote_ip)] = link.remote_host
#convert pairs list into 2-tuples, eg [1, 2, 3, 4, ...] -> (1, 2), (3, 4), ...
label_pairs = [ (src_dst_list[i], src_dst_list[i+1])
for i in range(0, len(src_dst_list), 2)]
for src_label, dst_label in label_pairs:
try:
src = self.network.find(src_label)
except ank.network.DeviceNotFoundException:
LOG.warn("Unable to find device %s" % src_label)
continue
try:
dst = self.network.find(dst_label)
except ank.network.DeviceNotFoundException:
LOG.warn("Unable to find device %s" % dst_label)
continue
LOG.info("Tracing from %s to %s" % (src_label, dst_label))
self.server.connect_vm(src.tap_ip, shell)
#TODO: make this handle appropriate ID for servers (no lo_ip)
# TODO: probably be st to integrate into the device namedtuple for consistency
shell.sendline("traceroute -n %s" % dst.lo_ip.ip)
shell.expect(self.server.NETKIT_PROMPT)
command_output = shell.before
self.server.disconnect_vm(shell)
shell.prompt()
re_table = textfsm.TextFSM(linux_traceroute_template)
route = re_table.ParseText(command_output)
# conver to just the returned IPs
route = [result[0] for result in route]
# try lookups
#TODO: put a try/except here, KeyError?
resolved_route = [ip_mappings[host] for host in route]
print "resolved route", resolved_route
resultant_paths[src_label][dst_label] = resolved_route
resolved_route.insert(0, src)
paths.append(resolved_route)
#pprint.pprint( resultant_paths)
# format for plotting
#TODO: enable/disable this from command line/function default
print "plotting paths", paths
ank.plot_paths(self.network, paths = paths)