最近在做安全测试时,需要对项目工程类的文件进行检查,所以根据项情况 ,整理代码搜索关键词,并对代码进行搜索,检查是否符合安全要求,并将查询出的结果写入到excel表格中,以方便查看结果。
对excle操作之前选用xlwt库进行处理,但有以下xlwt缺点:
版本
只能处理Excel97-2003或Excel 97之前版本的xls格式
存储数据过大
存储数据过大时,会报错Exception: String longer than 32767 characters
因查询数据较大,因此使用的是XlsxWrite库进行( https://github.com/jmcnamara/XlsxWriter ),以此来弥补以上缺点:
代码如下:
# -*- coding: utf-8 -*-
import os
def findAllFiles(path):
find_files = []
for root, dirs, files in os.walk(path): #
for file in files:
pathWithFile = os.path.join(root,file)
if os.path.splitext(pathWithFile)[1] in [".java",".xml",".iml",".bat"]:
find_files.append(pathWithFile)
else: