python3 -- subprocess，excel读写库实践整理

实践项目：统计VS编译环境下程序的全局变量大小

1. 项目需求

2. 需求分解

3. 各需求点涉及到的python知识

3.1 寻找特定尾缀的文件

>>>import os
>>>os.path.splitext("filename.obj")
('filename', '.obj')

if len(sys.argv) != 2:
print("Param number is invalid! Please input 1 param(dir to search)")
sys.exit()

if not os.path.exists(sys.argv[1]):
print("path(%s) is not exist!" % sys.argv[1])
sys.exit()

root, dirs, files = genRDF.send(None)
for file in files:
if os.path.splitext(file)[1] == ".obj":
#以下为对检索出的以.obj为尾缀的文件file做操作

>>>import os
>>> os.path.split("d:/software/filename.txt")
('d:/software', 'filename.txt')

3.2 利用VS工具dumpbin.exe对文件做解析

D:\vsPrj\dump\dump\dump\Debug>dumpbin /summary dump.obj
Microsoft (R) COFF/PE Dumper Version 14.00.23026.0
Copyright (C) Microsoft Corporation.  All rights reserved.

Dump of file dump.obj

File Type: COFF OBJECT

Summary

8 .data
BA8 .debug$S
44 .debug$T
74 .drectve
4 .rtc$IMZ
4 .rtc$TMZ
27 .text$mn

subprocess

import subprocess
proc = subprocess.Popen(args = ['dumpbin', '/summary', file], stdout=subprocess.PIPE, shell=True, encoding='gbk')
filelines = proc.stdout.readlines()
#filelines 是以行为元素的列表
#encoding 需要以gbk编码解码才可以正确显示

3.3 获取BSS段及DATA段大小

BSS_SEGMENT = '.bss'
DATA_SEGMENT = '.data'

for line in filelines:
if line.find(DATA_SEGMENT) != -1:                     #'.data'关键字在该行
sizeDATA = int(line[0:line.find(DATA_SEGMENT)])   #获取DATA段大小
if line.find(BSS_SEGMENT) != -1:                      #'.bss'关键字在该行
sizeBSS = int(line[0:line.find(BSS_SEGMENT)])     #获取BSS段大小

3.4 将段大小数据写入excel中

Copy an :class:

into an :class:

#读文件函数定义
def open_excl_read(file = 'file.xls'):
try:
data = xlrd.open_workbook(file)
return data
except Exception as e:
print(str(e))
return None

#excel读取
excel2read = open_excl_read("./x1.xls")
if excel2read == None:
print("Excel file not exist")
sys.exit()

#excel转换
excel2write = copy(excel2read)
wtTable = excel2write.get_sheet(0)            #获取Excel的第一个Sheet(编号为0)
wtTable.write(Row1, Column1, siz
4000
eBSS)         #将sizeBSS写入对应行列格中(行列从0开始算起)
wtTable.write(Row2, Column2, sizeDATA)        #将sizeDATA写入对应行列格中(行列从0开始算起)
excel2write.save('./x1.xls')                  #写入后，进行存储

4. 完整代码

#-*- coding: utf-8 -*-
import sys
import os
import xlrd
import xlwt
from xlutils.copy import copy
import subprocess

FILE_NAME_COL = 2
BSS_COL = 3
DATA_COL = 4
START_ROW = 4
BSS_SEGMENT = '.bss'
DATA_SEGMENT = '.data'

def open_excl_read(file = 'file.xls'):
try:
data = xlrd.open_workbook(file)
return data
except Exception as e:
print(str(e))
return None
def open_excl_write(file = 'file.xls'):
try:
data = xlwt.Workbook(encoding='utf-8', style_compression=0)
return data
except Exception as e:
print(str(e))
return None

if __name__ == '__main__':
if len(sys.argv) != 2:
print("Param number is invalid! Please input 1 param(dir to search)")
sys.exit()

if not os.path.exists(sys.argv[1]):
print("path(%s) is not exist!" % sys.argv[1])
sys.exit()

#root,dirs,files = os.walk(sys.argv[1])
curRow = START_ROW

excel2read = open_excl_read("./x1.xls")
if excel2read == None:
print("Excel file not exist")
sys.exit()

excel2write = copy(excel2read)
wtTable = excel2write.get_sheet(0)

genRDF = os.walk(sys.argv[1])
root, dirs, files = genRDF.send(None)
for file in files:
if os.path.splitext(file)[1] == ".obj":
sizeBSS = 0
sizeDATA = 0
#print(file)
#open excel

#get the size of data and bss field
filename = os.path.splitext(file)[0] + '.c'

proc = subprocess.Popen(args = ['dumpbin', '/summary', file], stdout=subprocess.PIPE, shell=True, encoding='gbk')

filelines = proc.stdout.readlines()
for line in filelines:
if line.find(DATA_SEGMENT) != -1:
sizeDATA = int(line[0:line.find(DATA_SEGMENT)])
if line.find(BSS_SEGMENT) != -1:
sizeBSS = int(line[0:line.find(BSS_SEGMENT)])

wtTable.write(curRow, FILE_NAME_COL, filename)
wtTable.write(curRow, BSS_COL, sizeBSS)
wtTable.write(curRow, DATA_COL, sizeDATA)

curRow += 1

excel2write.save('./x1.xls')