在特定文件夹中拥有所需的所有文件后,我希望我的 python 脚本压缩文件夹内容。
这可能吗?
我该怎么做呢?
在 python 2.7 上,您可以使用:shutil.make_archive(base_name, format[, root_dir[, base_dir[, verbose[, dry_run[, owner[, group[, logger]]]]]]])。
base_name档案名称减去扩展名
归档的格式格式
root_dir目录进行压缩。
例如
shutil.make_archive(target_file, format="bztar", root_dir=compress_me)
脚本的改编版本是:
#!/usr/bin/env python
from __future__ import with_statement
from contextlib import closing
from zipfile import ZipFile, ZIP_DEFLATED
import os
def zipdir(basedir, archivename):
assert os.path.isdir(basedir)
with closing(ZipFile(archivename, "w", ZIP_DEFLATED)) as z:
for root, dirs, files in os.walk(basedir):
#NOTE: ignore empty directories
for fn in files:
absfn = os.path.join(root, fn)
zfn = absfn[len(basedir)+len(os.sep):] #XXX: relative path
z.write(absfn, zfn)
if __name__ == '__main__':
import sys
basedir = sys.argv[1]
archivename = sys.argv[2]
zipdir(basedir, archivename)
例子:
C:\zipdir> python -mzipdir c:\tmp\test test.zip
它使用目录的内容创建'C:\zipdir\test.zip'
存档'c:\tmp\test'
。
这是一个递归版本
def zipfolder(path, relname, archive):
paths = os.listdir(path)
for p in paths:
p1 = os.path.join(path, p)
p2 = os.path.join(relname, p)
if os.path.isdir(p1):
zipfolder(p1, p2, archive)
else:
archive.write(p1, p2)
def create_zip(path, relname, archname):
archive = zipfile.ZipFile(archname, "w", zipfile.ZIP_DEFLATED)
if os.path.isdir(path):
zipfolder(path, relname, archive)
else:
archive.write(path, relname)
archive.close()
jfs 的解决方案和 Kozyarchuk 的解决方案都适用于 OP 的用例,但是:
因此,这是一个解决方案,只需将源文件夹(以及任何深度的任何子文件夹)添加到 zip 存档。这是因为您不能将文件夹名称传递给内置方法ZipFile.write()
- 下面的函数add_folder_to_zip()
, 提供了一种简单的方法来将文件夹及其所有内容添加到 zip 存档中。下面的代码适用于 Python2 和 Python3。
import zipfile
import os
def add_folder_to_zip(src_folder_name, dst_zip_archive):
""" Adds a folder and its contents to a zip archive
Args:
src_folder_name (str): Source folder name to add to the archive
dst_zip_archive (ZipFile): Destination zip archive
Returns:
None
"""
for walk_item in os.walk(src_folder_name):
for file_item in walk_item[2]:
# walk_item[2] is a list of files in the folder entry
# walk_item[0] is the folder entry full path
fn_to_add = os.path.join(walk_item[0], file_item)
dst_zip_archive.write(fn_to_add)
if __name__ == '__main__':
zf = zipfile.ZipFile('myzip.zip', mode='w')
add_folder_to_zip('zip_this_folder', zf)
zf.close()