python实现批量获取指定文件夹下的所有文件的厂商信息
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了python实现批量获取指定文件夹下的所有文件的厂商信息,小编现在分享给大家,供广大互联网技能从业者学习和参考。文章包含2347字,纯文字阅读大概需要4分钟。
内容图文
![python实现批量获取指定文件夹下的所有文件的厂商信息](/upload/InfoBanner/zyjiaocheng/434/7f01887b81b04d05b0ed3738b932d8a7.jpg)
功能代码如下:
import os, string, shutil,re import pefile import codecs, sys import wx import struct #输出中打印Unicode字符 #sys.stdout = codecs.lookup('utf-8')[-1](sys.stdout) def addToDict(theDict,PEfile_Path,strCompanyName): theDict.setdefault(PEfile_Path, [ ]).append(strCompanyName) #存在就在基础上加入列表,不存在就新建个字典key def IsPeFile(inputFileName): '''''判断一个文件是否为PE文件''' file = open(inputFileName, 'r') dosSign = hex(struct.unpack("h",file.read(2))[0]) if (dosSign == "0x5a4d"): file.seek(0x3c) date_fNew = struct.unpack("l",file.read(4))[0] file.seek(date_fNew) peSign = hex(struct.unpack("h",file.read(2))[0]) if (peSign == "0x4550"): return 1 else: return 0 else: return 0 #得到一个文件的厂商信息 #输入:文件路径 #输出:字典 def getCompanyName(PEfile_Path): if not IsPeFile(PEfile_Path): return {} else: dictCompany = {} pe = pefile.PE(PEfile_Path) p = re.compile('''''CompanyName:(.+)''') for name in p.findall(pe.__str__()): uniCompanyName = name.replace('\\x', '\\u').strip() #strTemp = uniCompanyName.decode('unicode_escape') addToDict(dictCompany, PEfile_Path, uniCompanyName) writeDicToFile(dictCompany) #写入文件 return dictCompany #得到文件夹中所有文件的厂商信息 #输入:文件夹路径 #输出:字典 def getCompanyNameFromDir(dir, dir_callback=None, file_callback=None): dictAll = {} for root, dirs, files in os.walk(dir): for f in files: file_path = os.path.join(root, f) if file_callback: file_callback(file_path) dictAll.update(getCompanyName(file_path)) return dictAll def writeDicToFile(dicName, outputFileName="company.txt"): """将字典写入文件中""" fileOutput = open(outputFileName, "a+") for key, value in dicName.items(): strTemp2 = '' + value[0] strChina2 = strTemp2.decode('unicode_escape') try: fileOutput.write("%-*s" % (110, key)) fileOutput.write(strChina2.encode('gb2312')) except UnicodeEncodeError, e: pass fileOutput.write("\n") fileOutput.close() #主函数 if __name__ == "__main__": getCompanyNameFromDir(u"D:\\everydaySample\\1221\\10white") print "ok finish"
这里不解释,代码很简单.
出现的问题如下:
1. 写入中文.str.encode('gb2212')解决
2. 出现UnicodeEncodeError 的错误,用了try给忽略了
希望本文所述对大家的Python程序设计有所帮助。
内容总结
以上是互联网集市为您收集整理的python实现批量获取指定文件夹下的所有文件的厂商信息全部内容,希望文章能够帮你解决python实现批量获取指定文件夹下的所有文件的厂商信息所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错,欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 gblab@vip.qq.com 举报,一经查实,本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。