#coding:utf-8
#壓縮解壓文件模塊
#支持中文路徑，支持多級目錄
#支持跨平臺，在linux和window下都可直接使用
#python 2.7.2
#author:xieShuxu
#QQ:258356793
#Email:sondx@qq.com
import zipfile,os
class ZipException(Exception):
    pass
def unZipFile(zipPath,unZipPath=''):
    '''解壓文件
    zipPath 要解壓的文件路徑
    unZipPath 解壓目標路徑 默認解壓到zipPath所在目錄
    '''
    try:
        filePath=filePath.decode('utf-8');
        zipFilePath=zipFilePath.decode('utf-8');
    except:
        print '================'
    if not os.path.exists(zipPath):
        raise ZipException,'function unZipFile:not exists file or dir(%s)' %zipPath;
    if unZipPath=='':
        unZipPath=os.path.splitext(zipPath)[0];
    if not unZipPath.endswith(os.sep):
        unZipPath+=os.sep;
    z = zipfile.ZipFile(zipPath, 'r')
    #zipInfolist=z.namelist();
    for k in z.infolist():
        savePath=unZipPath+k.filename;
        saveDir=os.path.dirname(savePath);
        if not os.path.exists(saveDir):
            os.makedirs(saveDir);
        f=open(savePath,'wb');
        f.write(z.read(k));
        f.close();
    z.close();
    #print unZipPath
global _iterateExeZipFile;
def exeZipFile(filePath,zipFilePath=''):
    '''壓縮文件
    filePath 要解壓的文件路徑 可以是文件或者目錄
             os.sep結尾表示壓縮該目錄下的子文件和文件夾 不包含該文件夾,否則包含該文件夾壓縮
    ZipFilePath 壓縮包文件路徑
                也可只傳文件名
                默認壓縮到filePath的父級目錄下
    '''
    filePath=filePath.decode('utf-8');
    zipFilePath=zipFilePath.decode('utf-8');
    #壓縮文件不存在直接返回
    if not os.path.exists(filePath):
        raise ZipException,'function exeZipFile:not exists file or dir(%s)' %filePath;
    # 是否包含父級目錄壓縮
    hasPDir=not filePath.endswith(os.sep);
    if not hasPDir:
        filePath=os.path.dirname(filePath);
        print filePath
    #校驗備份文件路徑
    if zipFilePath=='':
        zipFilePath=os.path.splitext(filePath)[0]+'.zip';
    elif zipFilePath.find(os.sep)==-1:#只傳文件名的處理
        zipFilePath=os.path.dirname(filePath)+os.sep+zipFilePath;
    #校驗創(chuàng)建備份路徑目錄
    if not os.path.exists(os.path.dirname(zipFilePath)):
        os.makedirs(os.path.dirname(zipFilePath));
    #初始化壓縮包中的根目錄
    zipRoot='';
    if hasPDir:
        zipRoot=os.path.split(filePath)[1];
    #開始壓縮
    z = zipfile.ZipFile(zipFilePath, 'w')
    if os.path.isfile(filePath):
        z.write(filePath,os.path.split(filePath)[1]);
    else:
        _iterateExeZipFile(filePath,zipRoot,z);
    z.close();
def _iterateExeZipFile(dirPath,zipRoot,z):

if __name__=='__main__':
    #壓縮包不帶級父文件夾目錄
    testdir='D:\\codeSource\\linuxAgent\\'
    zipFilePath='D:\\codeSource\\壓縮包不帶父級目錄.zip'
    exeZipFile(testdir,zipFilePath);
    #壓縮包帶父級文件夾目錄
    testdir='D:\\codeSource\\linuxAgent'#不帶后綴斜線
    zipFilePath='D:\\codeSource\\壓縮包帶父級目錄.zip'
    exeZipFile(testdir,zipFilePath);
    #不指定目標文件與路徑壓縮
    testdir='D:\\codeSource\\linuxAgent'
    exeZipFile(testdir);
    #指定壓縮包名稱不指定路徑壓縮
    testdir='D:\\codeSource\\linuxAgent\\'
    exeZipFile(testdir,'僅指定名稱壓縮包.zip');

    #指定解壓目錄解壓文件
    testdir=u'D:\\codeSource\\僅指定名稱壓縮包\\'
    zipFilePath=u'D:\\codeSource\\僅指定名稱壓縮包.zip'
    unZipFile(zipFilePath,testdir);
    #不指定目錄解壓
    zipFilePath=u'D:\\codeSource\\僅指定名稱壓縮包.zip'
    unZipFile(zipFilePath);