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)  #存在就在基礎(chǔ)上加入列表，不存在就新建個(gè)字典key  def IsPeFile(inputFileName):   '''''判斷一個(gè)文件是否為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     #得到一個(gè)文件的廠商信息 #輸入：文件路徑 #輸出：字典 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()    #主函數(shù) if __name__ == "__main__":   getCompanyNameFromDir(u"D://everydaySample//1221//10white")   print "ok finish"