python实现批量获取指定文件夹下的所有文件的厂商信息

本文实例讲述了python实现批量获取指定文件夹下的所有文件的厂商信息的方法。分享给大家供大家参考。具体如下：

功能代码如下：

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程序设计有所帮助。

赞 (0)

相关推荐

  • Python判断文件和文件夹是否存在的方法

    一.python判断文件和文件夹是否存在.创建文件夹 复制代码 代码如下: >>> import os >>> os.path.exists('d:/assist') True >>> os.path.exists('d:/assist/getTeacherList.py') True >>> os.path.isfile('d:/assist') False >>> os.path.isfile('d:/assis

  • Python实现递归遍历文件夹并删除文件

    思路: 遍历文件夹下面的文件夹 如果文件夹名称等于".svn",则修改文件夹的属性(因为".svn"的文件都是只读的,你不能直接删除) 删除此文件夹 如果文件夹名称不等于".svn",则递归上面的方法 Python的实现 代码 import os import shutil import os.path import stat rootdir="F:\\work\\Test" for parent,dirnames,filen

  • Python简单计算文件夹大小的方法

    本文实例讲述了Python简单计算文件夹大小的方法.分享给大家供大家参考.具体如下: import os, re """ 查看文件夹下的所有文件及文件夹 join为拼接函数 """ def Look_File(path): for root , dirs, files in os.walk(path, True): print root #主目录 for item in files: #主目录下的文件夹 print os.path.join(ro

  • Python发送以整个文件夹的内容为附件的邮件的教程

    由于我经常需要备份文件夹下的内容到邮件里面,每个打开邮件,上传文件,发送,太过麻烦,其实每次发送的文件都是放在固定 置的,只是邮件标题不同而已,于是用 python 为自己写了个发送文件到邮箱的小工具,在任意目录下执行该脚本,并指定邮件标 ,就将指定文件夹下的文件发送到邮箱中备份起来 . #!/usr/bin/env python # coding: utf-8 from smtplib import SMTP, quotedata, CRLF, SMTPDataError from email

  • Python列出一个文件夹及其子目录的所有文件

    python简介 Python是一种解释型.面向对象.动态数据类型的高级程序设计语言. Python由Guido van Rossum于1989年底发明,第一个公开发行版发行于1991年. 像Perl语言一样, Python 源代码同样遵循 GPL(GNU General Public License)协议. >>> import os >>> for i in os.walk("."): ... print i[0],"\n##"

  • python实现的文件夹清理程序分享

    使用: 复制代码 代码如下: foldercleanup.py -d 10 -k c:\test\keepfile.txt c:\test 表示对c:\test目录只保留最近10天的子文件夹和keepfile.txt中指定的子文件夹. 代码: 复制代码 代码如下: import os import os.path import datetime   def getOption():   from optparse import OptionParser     des   = "clean up

  • python创建临时文件夹的方法

    本文实例讲述了python创建临时文件夹的方法.分享给大家供大家参考.具体实现方法如下: import tempfile, os tempfd, tempname = tempfile.mkstemp('.suffix') os.write(tempfd, "aString") # or, if you want a file-object: os.fdopen(tempfd, 'w+') os.close(tempfd) os.unlink(tempname) 希望本文所述对大家的P

  • python实现批量获取指定文件夹下的所有文件的厂商信息

    本文实例讲述了python实现批量获取指定文件夹下的所有文件的厂商信息的方法.分享给大家供大家参考.具体如下: 功能代码如下: 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_Pa

  • Python实现合并同一个文件夹下所有PDF文件的方法示例

    Python实现合并同一个文件夹下所有PDF文件的方法示例

    本文实例讲述了Python实现合并同一个文件夹下所有PDF文件的方法.分享给大家供大家参考,具体如下: 一.需求说明 下载了网易云课堂的吴恩达免费的深度学习的pdf文档,但是每一节是一个pdf,我把这些PDF文档放在一个文件夹下,希望合并成一个PDF文件.于是写了一个python程序,很好的解决了这个问题. 二.数据形式 三.合并效果 四.python代码实现 # -*- coding:utf-8*- import sys reload(sys) sys.setdefaultencoding('

  • Python合并同一个文件夹下所有PDF文件的方法

    Python合并同一个文件夹下所有PDF文件的方法

    一.需求说明 下载了网易云课堂的吴恩达免费的深度学习的pdf文档,但是每一节是一个pdf,我把这些PDF文档放在一个文件夹下,希望合并成一个PDF文件.于是写了一个python程序,很好的解决了这个问题. 二.数据形式 三.合并效果 四.python代码实现 # -*- coding:utf-8*- import sys reload(sys) sys.setdefaultencoding('utf-8') import os import os.path from pyPdf import P

  • Python读取文件夹下的所有文件实例代码

    Python读取文件夹下的所有文件实例代码

    Python读取文件夹下的所有文件 os.listdir(path)是得到在path路径下所以文件的名称列表. open(path)是打开某个文件. iter是python的迭代器. 所以读取某文件夹下的所有文件如下: import os path = "D:/Python34/news" #文件夹目录 files= os.listdir(path) #得到文件夹下的所有文件名称 s = [] for file in files: #遍历文件夹 if not os.path.isdir

  • Python实现合并同一个文件夹下所有txt文件的方法示例

    Python实现合并同一个文件夹下所有txt文件的方法示例

    本文实例讲述了Python实现合并同一个文件夹下所有txt文件的方法.分享给大家供大家参考,具体如下: 一.需求分析 合并一个文件夹下所有txt文件 二.合并效果 三.python实现代码 # -*- coding:utf-8*- import sys reload(sys) sys.setdefaultencoding('utf-8') import os import os.path import time time1=time.time() ########################

  • 使用脚本实现自动清除指定文件夹下丢失链接文件的符号链接

    使用脚本实现自动清除指定文件夹下丢失链接文件的符号链接 脚本可清除,指定文件夹下,对视链接文件的符号链接. 在使用Linux时,常常会为自己许多文件或者程序建立符号链接,这样就不用每次都到对应的文件夹下去寻找对应的文件而只需要对需要经常访问的文件建立符号链接就可以了,这样就能把你常需要访问的文件放到桌面上,或者指定到另外一个文件夹中. 这样访问时,方便了很多,但是用完之后往往留下许多的符号链接,这些链接需要用户自己手动确认是否可以删除,为linux的使用带来很多的不方便,使用该脚本就能解放你的双

  • python引入其他文件夹下的py文件具体方法

    python引入其他文件夹下的py文件具体方法

    红色方框要引入箭头里面的 import sys sys.path.append('../../config/') from database import * print(MYSQL_CONFIG) 内容扩展: Python的import包含文件功能就跟PHP的include类似,但更确切的说应该更像是PHP中的require,因为Python里的import只要目标不存在就报错程序无法往下执行.要包含目录里的文件,PHP中只需要给对路径就OK.Python中则不同 a.py 要 import

  • JAVA实现遍历文件夹下的所有文件(递归调用和非递归调用)

    JAVA 遍历文件夹下的所有文件(递归调用和非递归调用) 1.不使用递归的方法调用. public void traverseFolder1(String path) { int fileNum = 0, folderNum = 0; File file = new File(path); if (file.exists()) { LinkedList<File> list = new LinkedList<File>(); File[] files = file.listFile

  • php 读取文件夹下所有图片、文件的实例

    如下所示: $hostdir=dirname(__FILE__).'/data/upload/admin/20170517/'; //要读取的文件夹 $url = '/data/upload/admin/20170517/'; //图片所存在的目录 $filesnames = scandir($hostdir); //得到所有的文件 // print_r($filesnames);exit; //获取也就是扫描文件夹内的文件及文件夹名存入数组 $filesnames $www = 'http:/

  • 如何配置vue.config.js 处理static文件夹下的静态文件

    最近开发过程中遇到个小需求,需要根据需求动态配置一些静态数据,但我不想直接把这个静态文件引入进来,build时就会把数据打包到js文件中,造成js文件很大:呃呃....(ps:就想给自己找点别扭) 回头看一下项目,果断的在根目录下创建一个static/test.json文件,在目标页面下fetch获取这个json文件里的数据: fetch("../static/test.json") .then(res => res.json()) .then(res => { //获取到

随机推荐