python学习重温(1)---自动case生成工具
因为最近工作中，需要设计根据wirshark截获的报文自动产生测试case的小工具，我重温了python。这个笔记，就是根据这个工具的实现，来记录python语法的学习过程。
这个工具的功能如下图所示，其中wirshark cap可以通过wirshark自身的功能导出为pmsl格式，也就是一种XML，同时公司使用的case script也是一种XML，只是添加了一些循环，判断等功能。
所以这个工具涉及文件操作，XML解析和一些python的基本语法功能
一、PYTHON 基本语法概念
1.什么是main函数
if __name__ == "__main__":
上面这个就是python的main函数的声明
2. 变量申明
python对变量申明不像那么严格，但也不像shell那么随便。一般变量通过首次赋值产生，当超出作用范围时自动消亡。这几句话的意思是，变量如果要使用就一定要先给他赋值。比如
print str1
一定会报错
Traceback (most recent call last):
File "<pyshell#0>", line 1, in <module>
print str1
NameError: name 'str1' is not defined
所以一定要先赋值
str1 = 'hello' print str1
python一样有全局变量和局部变量的概念。
3. python 常用的变量类型
数值型
value = 12
string
str1 = 'hello'
str1 = "hello"
这里有个双引号还是单引号的问题，我的理解是除非设计转义字符，两者都是一样的。
a = 'let\'s us' a = "let's us"
list
也就是C语言中的数组
a = ['spam', 'eggs', 100, 1234]
支持下标访问
a[0] a[1]
Dictionary
大家称呼这个味字典型，其实可以类比为C语言中的结构体。因为python里面是没有struct这类东西的，当然python可以用class的方法替代struct。但Dictionary这个类型也可以在某些时候当做struct这个概念来用
d = {"server":"mpilgrim", "database":"master"}
这个类型是可以作为不同类型数据的container的，但要注意一个问题dictionary在插入的时候是乱序的，也就是说当我做
d["newdata"]='fda'
这个操作后，这个新加入的项会随机加在d中，不一定在最后。
4.Python的层次结构
python依靠代码缩进和：来确定代码的前后层次关系
如下面这个函数，缩进起到了C语言中{}的作用。
def GetEnumName(attribename,valuestr):
"""
@author: limim
@param attribename:attribname
@type attribename:str
@param valuestr:mmt log value
@type valuestr:str
@return:the enum name
@rtype:str
"""
tree = ReadFromXml('PegasusEnumMapping.xml')
root = tree.getroot()
value = str(int(valuestr,16))
errorret = ''
for listnode in root.iter('attrib'):
strlist = listnode.attrib['name'].split('.')
attriblist = attribename.split('.')
substrlist = strlist[(len(strlist)-4):]
subattriblist = attriblist[(len(attriblist)-4):]
if substrlist == subattriblist:
for enumlistnode in listnode:
if enumlistnode[0].text == value:
return enumlistnode[1].text
else:
errorret = enumlistnode[1].text
print "error"
print attribename
print 'mmt log value' + ' ' + valuestr
print 'set default value' + ' ' + errorret
print '##############'
return errorret
二、elementTree
这个工具处理的主要文件格式是XML，就用elementTree这个库来处理，下面是要处理文件格式的一部分。
<?xml version="1.0"?>
<pdml version="0" creator="wireshark/1.20.0.1">
<packet>
<proto name="geninfo" pos="0" showname="General information" size="98">
<field name="num" pos="0" show="1246" showname="Number" value="4de" size="98"/>
<field name="len" pos="0" show="98" showname="Frame Length" value="62" size="98"/>
<field name="caplen" pos="0" show="98" showname="Captured Length" value="62" size="98"/>
<field name="timestamp" pos="0" show="Mar 6, 2013 18:28:28.729395000 China Standard Time" showname="Captured Time" value="1362565708.729395000" size="98"/>
</proto>
<proto name="frame" showname="Frame 1246: 98 bytes on wire (784 bits), 98 bytes captured (784 bits)" size="98" pos="0">
<field name="frame.time" showname="Arrival Time: Mar 6, 2013 18:28:28.729395000 China Standard Time" size="0" pos="0" show="Mar 6, 2013 18:28:28.729395000"/>
<field name="frame.time_epoch" showname="Epoch Time: 1362565708.729395000 seconds" size="0" pos="0" show="1362565708.729395000"/>
<field name="frame.time_delta" showname="Time delta from previous captured frame: 0.000475000 seconds" size="0" pos="0" show="0.000475000"/>
<field name="frame.time_delta_displayed" showname="Time delta from previous displayed frame: 0.000000000 seconds" size="0" pos="0" show="0.000000000"/>
<field name="frame.time_relative" showname="Time since reference or first frame: 93.072253000 seconds" size="0" pos="0" show="93.072253000"/>
<field name="frame.number" showname="Frame Number: 1246" size="0" pos="0" show="1246"/>
<field name="frame.len" showname="Frame Length: 98 bytes (784 bits)" size="0" pos="0" show="98"/>
<field name="frame.cap_len" showname="Capture Length: 98 bytes (784 bits)" size="0" pos="0" show="98"/>
<field name="frame.marked" showname="Frame is marked: False" size="0" pos="0" show="0"/>
<field name="frame.ignored" showname="Frame is ignored: False" size="0" pos="0" show="0"/>
<field name="frame.protocols" showname="Protocols in frame: eth:ip:udp:mmtss:sicap" size="0" pos="0" show="eth:ip:udp:mmtss:sicap"/>
<field name="frame.coloring_rule.name" showname="Coloring Rule Name: SICAP" size="0" pos="0" show="SICAP"/>
<field name="frame.coloring_rule.string" showname="Coloring Rule String: sicap" size="0" pos="0" show="sicap"/>
</proto>
<proto name="eth" showname="Ethernet II, Src: 192.168.254.1 (00:0f:bb:69:93:ee), Dst: DCT-INF (00:10:18:cb:b5:fd)" size="14" pos="0">
<field name="eth.dst" showname="Destination: DCT-INF (00:10:18:cb:b5:fd)" size="6" pos="0" show="00:10:18:cb:b5:fd" value="001018cbb5fd">
<field name="eth.addr" showname="Address: DCT-INF (00:10:18:cb:b5:fd)" size="6" pos="0" show="00:10:18:cb:b5:fd" value="001018cbb5fd"/>
<field name="eth.ig" showname=".... ...0 .... .... .... .... = IG bit: Individual address (unicast)" size="3" pos="0" show="0" value="0" unmaskedvalue="001018"/>
<field name="eth.lg" showname=".... ..0. .... .... .... .... = LG bit: Globally unique address (factory default)" size="3" pos="0" show="0" value="0" unmaskedvalue="001018"/>
</field>
<field name="eth.src" showname="Source: 192.168.254.1 (00:0f:bb:69:93:ee)" size="6" pos="6" show="00:0f:bb:69:93:ee" value="000fbb6993ee">
<field name="eth.addr" showname="Address: 192.168.254.1 (00:0f:bb:69:93:ee)" size="6" pos="6" show="00:0f:bb:69:93:ee" value="000fbb6993ee"/>
<field name="eth.ig" showname=".... ...0 .... .... .... .... = IG bit: Individual address (unicast)" size="3" pos="6" show="0" value="0" unmaskedvalue="000fbb"/>
<field name="eth.lg" showname=".... ..0. .... .... .... .... = LG bit: Globally unique address (factory default)" size="3" pos="6" show="0" value="0" unmaskedvalue="000fbb"/>
</field>
<field name="eth.type" showname="Type: IP (0x0800)" size="2" pos="12" show="0x0800" value="0800"/>
</proto>
1. 文件读取
from xml.etree.ElementTree import ElementTree, Element
import sys
def ReadFromXml(path):
''''read from xml and prase it
author:limin
path: the file path
return ElementTree'''
tree = ElementTree()
tree.parse(path);
return tree
通过
tree = ElementTree() tree.parse(path);
就获得了XML文件的指针
在调试模式下，可以看到tree就是根据读入的XML文件用不同层级的NODE构成的结构体。这个时候操作就很方便了，我们可以先拿到root，在遍历整个tree就可以方便的读取XML中任意一个节点的attrib，text，tag这些内容。
def ReadPacketContent(tree,framenum):
'''
read the packet content
node:the xml node
num: the index of the node
'''
SicCapMessage = []
root = tree.getroot() #check the boundary
num = 0
flag = 0
for node in root: # proto level node
if flag == 1:
break
num = num + 1
#print protonode.tag,protonode.attrib
for protonode in node:
if protonode.attrib['name'] == 'frame':
frameList = parseFrameHead(protonode)
if frameList[1] != str(framenum):
break
else:
SicCapMessage.insert(0,frameList)
print "find the frame, number is" + frameList[1]
flag = 1
print num
print len(root)
if num == 0 or num > len(root) - 1:
print "can not find the frame num"
return SicCapMessage
for protonode in root[num - 1]:
if protonode.attrib['name'] == 'sicap':
#dictPrint(parseSiCapMessage(protonode))
SicCapMessage.extend(parseSiCapMessage(protonode))
return SicCapMessage
可以看到
protonode.attrib
就是一个dictionary类型的变量
2.文件写入
读取的问题解决了，剩下的就是写入了。只用调用
tree.write('test.xml','UTF-8')
就可以将内存中的tree写入文件
