思路:先把第一级目录中的文件进行遍历,如果是文件,则把它连同当前路径一起加入result(list),如果是子目录,则在整个目录上继续深度优先遍历,直到所有的文件都被加入。主要使用python3中zipfile模块:
class zipfile.
ZipFile
(file [,mode [,compression [,allowZip64 ] ] ] )
参数说明:
1.第一个参数file可以是文件(字符串)的路径,也可以是类文件对象
2.第二个参数mode模式参数可以是'r'
读取现有的文件,'w'
以截断并写入一个新的文件,或'a'
将追加到现有文件。如果mode是'a'
和file引用现有的ZIP文件,则会向其中添加其他文件,如果文件未引用ZIP文件,则会将新的ZIP存档附加到该文件,这用于将ZIP存档添加到另一个文件
3.第三个参数compression是编写存档时使用的ZIP压缩方法,可以是ZIP_STORED
或ZIP_DEFLATED
Demo:
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import zipfile
def get_zip_file(input_path, result):
"""
对目录进行深度优先遍历
:param input_path:
:param result:
:return:
"""
files = os.listdir(input_path)
for file in files:
if os.path.isdir(input_path + '/' + file):
get_zip_file(input_path + '/' + file, result)
else:
result.append(input_path + '/' + file)
def zip_file_path(input_path, output_path, output_name):
"""
压缩文件
:param input_path: 压缩的文件夹路径
:param output_path: 解压(输出)的路径
:param output_name: 压缩包名称
:return:
"""
f = zipfile.ZipFile(output_path + '/' + output_name, 'w', zipfile.ZIP_DEFLATED)
filelists = []
get_zip_file(input_path, filelists)
for file in filelists:
f.write(file)
# 调用了close方法才会保证完成压缩
f.close()
return output_path + r"/" + output_name
if __name__ == '__main__':
zip_file_path(r"./testing", 'F:', '123.zip')