#!/usr/bin/python
#coding=utf-8
import re,sys,csv
from string import replace
def Interfaceinfo(file_path,firms):
with open(file_path,‘r‘) as f:
filecontext = f.readlines()
if firms == ‘a10‘:
splitstring_keyword = ‘(^!)‘
firms_interface = ‘^interface (.+)‘
ip_keyword = ‘^ip address (.+)‘
desc_keyword = ‘^name (.+)‘
access_vlan_keyword = ‘^switchport access vlan (.+)‘
trunk_vlan_keyword = ‘^switchport trunk vlan (.+)‘
elif firms == ‘zxr10‘:
splitstring_keyword = ‘(^\$)‘
firms_interface = ‘^interface (.+)‘
ip_keyword = ‘^ip address (.+)‘
desc_keyword = ‘^description (.+)‘
access_vlan_keyword = ‘^switchport access vlan (.+)‘
trunk_vlan_keyword = ‘^switchport trunk vlan (.+)‘
elif firms == ‘zte‘ or firms == ‘ruijie‘:
splitstring_keyword = ‘(^!)‘
firms_interface = ‘^interface (.+)‘
ip_keyword = ‘^ip address (.+)‘
desc_keyword = ‘^description (.+)‘
access_vlan_keyword = ‘^switchport access vlan (.+)‘
trunk_vlan_keyword = ‘^switchport trunk vlan (.+)‘
elif firms == ‘hillstone‘:
splitstring_keyword = ‘(^exit)‘
firms_interface = ‘^interface (.+)‘
ip_keyword = ‘^ip address (.+)‘
desc_keyword = ‘^description (.+)‘
access_vlan_keyword = ‘^port access vlan (.+)‘
trunk_vlan_keyword = ‘^port trunk permit vlan (.+)‘
else:
splitstring_keyword = ‘(^#)‘
firms_interface = ‘^interface (.+)‘
ip_keyword = ‘^ip address (.+)‘
desc_keyword = ‘^description (.+)‘
access_vlan_keyword = ‘^port access vlan (.+)‘
trunk_vlan_keyword = ‘^port trunk permit vlan (.+)‘
‘‘‘
定义列表:分隔符行值
‘‘‘
splitstring_rowvalues = []
‘‘‘
循环文件内容,获得分隔符所在的行号
‘‘‘
for itemkey in xrange(0,len(filecontext)):
splitstring_rowvalue = re.findall(splitstring_keyword,filecontext[itemkey].strip())
if len(splitstring_rowvalue) != 0:
splitstring_rowvalues.append(itemkey)
‘‘‘
定义列表:接口名行值
‘‘‘
portname_rowvalues = []
‘‘‘
循环文件内容,获得接口名所在的行号
‘‘‘
for itemkey in xrange(0,len(filecontext)):
portname_value = re.findall(firms_interface,filecontext[itemkey].strip())
if len(portname_value) != 0:
portname_rowvalues.append(itemkey)
‘‘‘
定义列表:接口名所在行最近的分隔符的行值
例子:
interface Ten-GigabitEthernet2/8/0/30
port link-mode bridge
shutdown
# <-- 此分割符行号
‘‘‘
portnameEND_rowvalues = []
‘‘‘
列表开始位置值初始化为0
‘‘‘
startkey = 0
‘‘‘
循环提取接口名所在行最近的分隔符的行号
‘‘‘
for p_itemkey in xrange(0,len(portname_rowvalues)):
for s_itemkey in xrange(startkey,len(splitstring_rowvalues)):
if portname_rowvalues[p_itemkey] < splitstring_rowvalues[s_itemkey]:
portnameEND_rowvalues.append(splitstring_rowvalues[s_itemkey])
startkey = s_itemkey
break
‘‘‘
定义列表:接口信息
‘‘‘
portinfo = []
‘‘‘
循环提取接口名称,IP地址,接口描述,trunk或access相关的vlan信息 ----开始处
‘‘‘
for itemkey01,itemkey02 in zip(portname_rowvalues,portnameEND_rowvalues):
‘‘‘
提取接口名称
‘‘‘
port_name = re.findall(firms_interface,filecontext[itemkey01].strip())
port_name = port_name[0].replace(‘ ‘,‘-‘)
‘‘‘
提取接口描述
‘‘‘
for itemkey in xrange(itemkey01,itemkey02):
port_desc = re.findall(desc_keyword,filecontext[itemkey].strip())
if len(port_desc) != 0:
port_desc = port_desc[0].replace(‘ ‘,‘-‘)
break
else:
port_desc = ‘none‘
‘‘‘
提取接口IP
‘‘‘
for itemkey in xrange(itemkey01,itemkey02):
ip_list = []
ipvalue = re.findall(ip_keyword,filecontext[itemkey].strip())
if len(ipvalue) != 0:
‘‘‘
子网掩码位数转换
‘‘‘
ip_netmask = lambda netmask:sum(bin(int(i)).count(‘1‘) \
for i in netmask.split(‘.‘))
ip_list = ipvalue[0].replace(‘ ‘,‘/‘).split(‘/‘)
ip_list[1] = str(ip_netmask(ip_list[1]))
port_ip = ‘/‘.join(ip_list)
break
else:
port_ip = ‘none‘
‘‘‘
提取接口Trunk_VLAN
‘‘‘
trunk_vlan = []
for itemkey in xrange(itemkey01,itemkey02):
port_vlan = re.findall(trunk_vlan_keyword,filecontext[itemkey].strip())
if len(port_vlan) != 0:
port_vlan = port_vlan[0].replace(‘ ‘,‘-‘)
trunk_vlan.append(port_vlan)
‘‘‘
提取接口Access_VLAN
‘‘‘
for itemkey in xrange(itemkey01,itemkey02):
port_vlan = re.findall(access_vlan_keyword,filecontext[itemkey].strip())
if len(port_vlan) != 0:
port_vlan = ‘access‘ + ‘-‘ + port_vlan[0].replace(‘ ‘,‘-‘)
break
else:
port_vlan = ‘none‘
‘‘‘
将接口信息拼凑
‘‘‘
if len(trunk_vlan) != 0:
trunk_vlan = ‘trunk‘ + ‘-‘ + reduce(lambda x,y:x + ‘-‘ + y,trunk_vlan)
portinfo.append(port_name + ‘ ‘ + port_ip + ‘ ‘ + trunk_vlan + ‘ ‘+ port_desc)
else:
portinfo.append(port_name + ‘ ‘ + port_ip + ‘ ‘ + port_vlan + ‘ ‘ + port_desc)
‘‘‘
输出结果到csv文件
‘‘‘
headColContext = re.split(‘[/.]‘,file_path)[-2]
csv_file_path = ‘/root/log/Interfaceinfo.csv‘
with open(csv_file_path,‘ab‘) as cf:
cfwriter = csv.writer(cf,dialect=‘excel‘)
for p in portinfo:
cfwriter.writerow((headColContext + ‘ ‘ + p).split(‘ ‘))
if __name__ == ‘__main__‘:
Interfaceinfo(sys.argv[1],sys.argv[2])