Python使用zipfile模块压缩和解压缩文件及目录
压缩多个文件
import zipfile
import os
def zip_files(files, output):
"""压缩多个文件"""
zip = zipfile.ZipFile(output, 'w', zipfile.ZIP_DEFLATED )
for file in files:
zip.write( file )
zip.close()
压缩指定目录
def zip_dir(path, output=None):
"""压缩指定目录"""
output = output or os.path.basename(path) + '.zip' # 压缩文件的名字
zip = zipfile.ZipFile(output, 'w', zipfile.ZIP_DEFLATED)
for root, dirs, files in os.walk(path):
relative_root = '' if root == path else root.replace(path, '') + os.sep # 计算文件相对路径
for filename in files:
zip.write(os.path.join(root, filename), relative_root + filename) # 文件路径 压缩文件路径(相对路径)
zip.close()
解压目录
def unzip(zip_file, output=None):
"""解压zip文件"""
zip = zipfile.ZipFile(zip_file)
output = output or os.path.dirname(zip_file) # 默认解压到当前目录同名文件夹中
zip.extractall(output) # 会被解压到输入的路径中
zip.close()
处理MacOS系统压缩目录生成的__MACOSX目录
def unzip_file(file_path):
file_dir = os.path.dirname(file_path)
new_zip_file = os.path.join(file_dir, 'new_archive.zip')
with ZipFile(file_path) as original_zip:
with ZipFile(new_zip_file, 'w') as new_zip:
for item in original_zip.infolist():
buffer = original_zip.read(item.filename)
if not str(item.filename).startswith('__MACOSX/'):
new_zip.writestr(item, buffer)
new_zip.extractall(os.path.dirname(file_path)) #解压到当前工作目录
os.remove(new_zip_file)
os.remove(file_path)
return os.path.join(file_dir, os.listdir(file_dir)[0])