Python 解压缩Zip和Rar文件到指定目录

Posted 2020-08-16

#__author__ = ‘Joker‘
# -*- coding:utf-8 -*-
import urllib
import os
import os.path
import zipfile
from zipfile import *
import sys
reload(sys)
sys.setdefaultencoding(‘gbk‘)

rootdir = "F:/50_GIS/1000_Tools"    # 指明被遍历的文件夹
zipdir = "F:/000_Terrain/zipdir"    # 存储解压缩后的文件夹

#Zip文件处理类
class ZFile(object):
    def __init__(self, filename, mode=‘r‘, basedir=‘‘):
        self.filename = filename
        self.mode = mode
        if self.mode in (‘w‘, ‘a‘):
            self.zfile = zipfile.ZipFile(filename, self.mode, compression=zipfile.ZIP_DEFLATED)
        else:
            self.zfile = zipfile.ZipFile(filename, self.mode)
        self.basedir = basedir
        if not self.basedir:
            self.basedir = os.path.dirname(filename)

    def addfile(self, path, arcname=None):
        path = path.replace(‘//‘, ‘/‘)
        if not arcname:
            if path.startswith(self.basedir):
                arcname = path[len(self.basedir):]
            else:
                arcname = ‘‘
        self.zfile.write(path, arcname)

    def addfiles(self, paths):
        for path in paths:
            if isinstance(path, tuple):
                self.addfile(*path)
            else:
                self.addfile(path)

    def close(self):
        self.zfile.close()

    def extract_to(self, path):
        for p in self.zfile.namelist():
            self.extract(p, path)

    def extract(self, filename, path):
        if not filename.endswith(‘/‘):
            f = os.path.join(path, filename)
            dir = os.path.dirname(f)
            if not os.path.exists(dir):
                os.makedirs(dir)
            file(f, ‘wb‘).write(self.zfile.read(filename))

#创建Zip文件
def createZip(zfile, files):
    z = ZFile(zfile, ‘w‘)
    z.addfiles(files)
    z.close()

#解压缩Zip到指定文件夹
def extractZip(zfile, path):
    z = ZFile(zfile)
    z.extract_to(path)
    z.close()

#解压缩rar到指定文件夹
def extractRar(zfile, path):
    rar_command1 = "WinRAR.exe x -ibck %s %s" % (zfile, path)
    rar_command2 = r‘"C:\\WinRAR.exe" x -ibck %s %s‘ % (zfile, path)
    if os.system(rar_command1) == 0:
        print "Path OK."
    else:

        if os.system(rar_command2) != 0:
            print "Error."
        else:
            print "Exe OK"

#获得文件名和后缀
def GetFileNameAndExt(filename):
    (filepath,tempfilename) = os.path.split(filename);
    (shotname,extension) = os.path.splitext(tempfilename);
    return shotname,extension

#定义文件处理数量-全局变量
fileCount = 0

#递归获得rar文件集合
def getFiles(filepath):
#遍历filepath下所有文件，包括子目录
  files = os.listdir(filepath)
  for fi in files:
    fi_d = os.path.join(filepath,fi)
    if os.path.isdir(fi_d):
        getFiles(fi_d)
    else:
        global fileCount
        global zipdir
        fileCount = fileCount + 1
        # print fileCount
        fileName = os.path.join(filepath,fi_d)
        filenamenoext = GetFileNameAndExt(fileName)[0]
        fileext = GetFileNameAndExt(fileName)[1]
        # 如果要保存到同一个文件夹，将文件名设为空
        filenamenoext = ""
        zipdirdest = zipdir + "/" + filenamenoext + "/"
        if fileext in [‘.zip‘,‘.rar‘]:
            if not os.path.isdir(zipdirdest):
                os.mkdir(zipdirdest)
        if fileext == ".zip" :#
            print str(fileCount) + " -- " + fileName
           # unzip(fileName,zipdirdest)
            extractZip(fileName,zipdirdest)
        elif fileext == ".rar":
            print str(fileCount) + " -- " + fileName
            extractRar(fileName, zipdirdest)

#递归遍历“rootdir”目录下的指定后缀的文件列表
getFiles(rootdir)

文件处理参考：http://www.cnblogs.com/txw1958/archive/2012/03/08/2385540.html