标签:blog io os 使用 ar for div sp 代码
这两天实验室网络不给力,后来发现是有人占用了实验室太多的带宽,而登陆到实验室老的h3c s5500交换机上看端口流量情况很不方便,于是萌生写个小工具来统计端口流量情况,已求找到谁占用了大量带宽。
于是查了下,发现python 有个telnetlib的库,登陆交换机以及进行简单的操作相当简单,于是就写了这么个小工具:
*************************************工作原理********************************************
1、本程序采用telnet的方式登陆到交换机并通过不停的发送display interface [接口] 的方式请求交换机接口
信息并从中提取total input 和 total output的方式计算当前端口速率。
2、本程序仅统计状态为UP的端口的信息。使用display brief interface获取到状态为UP的接口。
3、本程序仅统计gigabytes端口。
*************************************其他**********************************************
1、第一次统计信息,由于没有时间,信息不准确
2、速度的单位都是MB
3、统计结果只能作为参考,并不能代表速率。
4、由于交换机本身更新接口信息速度并不快,而且telnet发送命令回传耗时也较大,本程序设置最小刷新时间,
当前为5s,若由于请求交换机接口所耗时超过5s,则刷新时间为请求耗时时间。
#!/usr/bin/python import re import telnetlib import time import platform import os Host = ‘.....‘ username = ‘‘ password = ‘‘ finish = ‘<....>‘ MIN_INTERVAL = 5.0 # use float PORT_COUNT = 52 # h3c s5500 has 52 gigabyte ports # return system type as a string def get_system_info(): sys_platform = platform.system() if sys_platform == ‘Linux‘ or sys_platform == ‘Darwin‘: return ‘UNIX‘ elif sys_platform == ‘Windows‘: return ‘WINDOWS‘ else: return ‘UNIX‘ def clear_screen(): sys_type = get_system_info() if sys_type == ‘UNIX‘: os.system("clear") elif sys_type == ‘WINDOWS‘: os.system(‘cls‘) else: os.system(‘clear‘) # login to the device and return the Telnet object def login(): # telnet to the device print ‘connect...‘ tn = telnetlib.Telnet(Host,timeout = 5) tn.read_until(‘Username:‘,timeout=5) tn.write(username + ‘\n‘) tn.read_until(‘Password:‘) tn.write(password + ‘\n‘) tn.read_until(finish) print ‘telnet success‘ return tn #‘‘‘using Telnet object to get port status and return a tuple filled with ‘UP‘ ports‘‘‘ def get_up_ports(tn): example = ‘‘‘ The brief information of interface(s) under bridge mode: Interface Link Speed Duplex Link-type PVID GE1/0/1 UP 100M(a) full(a) access 409 GE1/0/2 UP 100M(a) full(a) access 409 GE1/0/3 DOWN auto auto access 409‘‘‘ tn.write(‘display brief interface\n‘) tn.write(‘ ‘) tn.write(‘ ‘) ports_brief = tn.read_until(finish) up_ports = [] port_info = re.findall(r"GE1/0/(\d+)(\s+)(\w+)",ports_brief) for i in port_info: if i[-1] == ‘UP‘: up_ports.append(i[0]) print up_ports return tuple(up_ports) def extract_data(port_str): ‘‘‘ get the data from the result of command ‘display interface GigabitEthernet 1/0/i‘ ‘‘‘ # (VLAN_ID, total_input, total_output, max_input, max_output) if re.search(‘GigabitEthernet‘, port_str) == None: return None VLAN_ID_list = re.findall(r"PVID: (\d+)",port_str) input_total_list = re.findall(r"Input \(total\): (\d+) packets, (\d+) bytes", port_str) output_total_list = re.findall(r"Output \(total\): (\d+) packets, (\d+) bytes", port_str) peak_input_list = re.findall(r"Peak value of input: (\d+) bytes/sec,", port_str); peak_output_list = re.findall(r"Peak value of output: (\d+) bytes/sec,", port_str); state_list = re.findall(r"current state: (.+)",port_str) VLAN_ID = VLAN_ID_list[0] # string input_total = long((list(input_total_list[0]))[1]) # long output_total = long((list(output_total_list[0]))[1]) # long peak_input = long(peak_input_list[0]) # long peak_output = long(peak_output_list[0]) # long state = str(state_list[0]) # string return (VLAN_ID, input_total, output_total, peak_input, peak_output, state) def do_statistic(): last_input = [0] * PORT_COUNT last_output = [0] * PORT_COUNT last_update = time.time() tn = login() up_ports = get_up_ports(tn) clear_screen() print ‘connected, waiting...‘ while(True): ports_str = [] # h3c s5500 g1/0/1 - g1/0/52 # input command to get output for i in up_ports: tn.write(‘display interface GigabitEthernet 1/0/‘ + str(i) +‘\n‘) tn.write(‘ ‘) port_info = tn.read_until(finish) ports_str.append(port_info) # get interval interval = (time.time() - last_update) if interval < MIN_INTERVAL: time.sleep(MIN_INTERVAL - interval) interval = MIN_INTERVAL # get data and print clear_screen() print "the input/output is from the port view of the switch." print "From the user‘s view: input <-> download; output <-> upload." print "the VLAN 1000 is connected to the firewall. So, it‘s opposite\n\n" print ‘PORT_NO\tVLAN_ID\tINPUT\tOUTPUT\tMAX_IN\tMAX_OUT\tSTATE (MB)‘ port_index = 0 for _port_str in ports_str: # (VLAN_ID, total_input, total_output, max_input, max_output) data = extract_data(_port_str) if data == None: continue port_no = up_ports[port_index] vlan_id = data[0] speed_input = (data[1] - last_input[port_index]) / (interval * 1024 * 1024) speed_output = (data[2] - last_output[port_index]) / (interval * 1024 * 1024) max_input = data[3] / (1024 * 1024 ) max_output = data[4] / (1024 * 1024 ) state = data[5] last_input[port_index] = data[1] last_output[port_index] = data[2] port_index += 1 # show print port_no, ‘\t‘, vlan_id, ‘\t‘,float(‘%.2f‘ %speed_input), ‘\t‘, float(‘%.2f‘ %speed_output), ‘\t‘, print float(‘%.2f‘ %max_input), ‘\t‘ ,float(‘%.2f‘ %max_output), ‘\t‘, state last_update = time.time() if __name__ == "__main__": username = raw_input("please input username:") password = raw_input("please input password:") print username print password do_statistic() tn.close()
代码未做异常处理,仅实现了简单的流量统计功能。
标签:blog io os 使用 ar for div sp 代码
原文地址:http://www.cnblogs.com/ljxuan/p/3966732.html