python-对指定文件夹进行压缩

目的

首先，我试验了一下【1】的效果：

import os

from zipfile import ZipFile

def backupZip(folder): #这个函数只做文件夹打包的动作，不判断压缩包是否存在

zipfile_name = os.path.basename(folder) + '.zip' #压缩包和文件夹同名

with ZipFile(zipfile_name, 'w') as zfile: #以写入模式创建压缩包

for foldername, subfolders, files in os.walk(folder): #遍历文件夹

print('Adding files in ' + foldername +'...')

zfile.write(foldername)

for i in files:

zfile.write(os.path.join(foldername,i))

print('Adding ' + i)

print('Done.')

def folder2zip(folder): #文件夹打包为zip的函数

zipfile_name = os.path.basename(folder) + '.zip'

if not os.path.exists(zipfile_name): #检查压缩包是否存在，如果已存在则询问是否继续

backupZip(folder)

else:

response = input("Zipfile exists. Coutinue?('q' for quit): ")

if response != 'q':

backupZip(folder)

但是当我使用绝对路径 folder2zip("C:/Users/10073\Desktop/knn") 测试效果时，其效果却是这样的：

直接双击打开 knn.zip 结果却是这样的，而我想要的是打开zip文件出现的直接是该文件的内容。那么就需要改写代码了。

首先我们得了解ZipFile.write函数的用法。

ZipFile.write()的简单介绍

ZipFile.write(filename[, arcname[, compress_type]])

其中：filename为文件路径，arcname为添加到zip文档之后保存的名称,参数compress_type表示压缩方法，它的值可以是zipfile.ZIP_STORED或zipfile.ZIP_DEFLATED

示例代码：

import zipfile,os

fileName = r"E:\\WORK\\js\\2.html"

zipName = r"E:\\WORK\\js\\2.zip"

zf = zipfile.ZipFile(zipName, 'w')

zf.write(fileName,'3.html')#fileName-文件名，3.html-压缩后的文件名称

zf.close()

要想达到我的目的，需要一些递归操作。

思路与代码编写

首先，由于要进行递归写入，所以我把生成zipfile实例这一步放到了解压函数外面，并把它作为了解压函数的参数。(注意我的第5行和第10行)

def folder2zip(folder): #文件夹打包为zip的函数

zipfile_name = os.path.basename(folder) + '.zip'

if not os.path.exists(zipfile_name): #检查压缩包是否存在，如果已存在则询问是否继续

zfile=ZipFile(zipfile_name,'w')

myOwnZip(zfile,folder)

else:

response = input("Zipfile exists. Coutinue?('q' for quit): ")

if response != 'q':

zfile = ZipFile(zipfile_name, 'w')

myOwnZip(zfile,folder)

一开始我的 myOwnZip 函数是这样实现的：

def myOwnZip(zfile,folder):

print("folder:",folder)

fileList=os.listdir(folder)

for file in fileList:

if os.path.isfile(os.path.join(folder,file)): # 如果是文件

zfile.write(os.path.join(folder,file),file)

else:

zfile.write(os.path.join(folder,file),baseDir+"/"+file) # 创建文件夹

print("将文件夹",file,"添加到压缩包中")

myOwnZip(zfile,os.path.join(folder,file),file)

print('Done.')

但是这样会出现一个问题，就是不管是在第几层的文件都写在了同一级下(没能递归)。

也就是说如果一个文件存放在knn/test/mm.py，那么在zfile.write中的file也应该是knn/test/mm.py。但是在我们上述函数中，file输出为mm.py，所以zfile.write就直接把它放在第一层里了。

于是我又加了一个参数，basedir。

def myOwnZip(zfile,folder,baseDir=""):

print("folder:",folder)

print("baseDir:",baseDir)

fileList=os.listdir(folder)

for file in fileList:

if os.path.isfile(os.path.join(folder,file)): # 如果是文件

zfile.write(os.path.join(folder,file),os.path.join(baseDir,file))

else:

zfile.write(os.path.join(folder,file),baseDir+"/"+file) # 创建文件夹

print(baseDir+"/"+file)

print("将文件夹",file,"添加到压缩包中")

# baseFolderName=os.path.basename(folder)

myOwnZip(zfile,os.path.join(folder,file),baseDir=os.path.join(baseDir,file))

print('Done.')

这样就实现了递归压缩。

再接下来是，就是要实现把文件压缩到指定文件夹。这个思路就很简单了，我们把它压缩到当前目录后，再shutil.move到指定文件夹就行了。专业代写，扫码了解

参考资料