在 Python 中创建 .zip？

Question

我试图在我的脚本中创建一个函数，它将给定源目录 (src) 的内容压缩到一个 zip 文件 (dst)。例如zip('/path/to/dir', '/path/to/file.zip')，其中/path/to/dir 是一个目录，而/path/to/file.zip 尚不存在。我不想压缩目录本身，这在我的情况下完全不同。我想压缩目录中的文件（和子目录）。这就是我正在尝试的：

def zip(src, dst):
    zf = zipfile.ZipFile("%s.zip" % (dst), "w")
    for dirname, subdirs, files in os.walk(src):
        zf.write(dirname)
        for filename in files:
            zf.write(os.path.join(dirname, filename))
    zf.close()

这会创建一个本质上是/ 的zip。例如，如果我压缩了/path/to/dir，解压缩 zip 会创建一个目录，其中包含“path”，该目录中包含“to”，等等。

有没有人不会导致这个问题的功能？

我怎么强调都不过分，它需要压缩目录中的文件，而不是目录本身。

Answer 1

zipfile.write() 方法采用可选的arcname 参数 指定 zipfile 中的文件名。

您可以使用它在开头剥离src 的路径。我在这 使用os.path.abspath() 确保src 和 os.walk() 返回的文件名有一个共同的前缀。

#!/usr/bin/env python2.7

import os
import zipfile

def zip(src, dst):
    zf = zipfile.ZipFile("%s.zip" % (dst), "w", zipfile.ZIP_DEFLATED)
    abs_src = os.path.abspath(src)
    for dirname, subdirs, files in os.walk(src):
        for filename in files:
            absname = os.path.abspath(os.path.join(dirname, filename))
            arcname = absname[len(abs_src) + 1:]
            print 'zipping %s as %s' % (os.path.join(dirname, filename),
                                        arcname)
            zf.write(absname, arcname)
    zf.close()

zip("src", "dst")

使用这样的目录结构：

src
└── a
    ├── b
    │   └── bar
    └── foo

脚本打印：

zipping src/a/foo as a/foo
zipping src/a/b/bar as a/b/bar

生成的 zip 文件的内容是：

Archive:  dst.zip
  Length     Date   Time    Name
 --------    ----   ----    ----
        0  01-28-13 11:36   a/foo
        0  01-28-13 11:36   a/b/bar
 --------                   -------
        0                   2 files

Answer 2

据我所知，你很接近。您可以使用dirname 和basename 来确保获取正确的路径名：

>>> os.path.dirname("/path/to/dst")
'/path/to'
>>> os.path.basename("/path/to/dst")
'dst'

然后使用chdir，您可以确保您在父级中，因此路径是相对的。

def zip(src, dst):
    parent = os.path.dirname(dst)
    folder = os.path.basename(dst)

    os.chdir(parent):
    for dirname, subdirs, filenames in os.walk(folder):
        ...

这会创建：

dst/a.txt
dst/b
dst/b/c.txt
...etc...

如果不想包含名称“dst”，您可以先输入os.chdir(dst)，然后输入os.walk('.')。

希望对您有所帮助。

Answer 3

使用arcname parameter 控制压缩文件中的名称/路径。

例如，对于仅包含文件，不包含目录的 zip 文件：

zf.write(os.path.join(dirname, filename), arcname=filename)

或者在 zip 文件中创建一个新目录：

zf.write(os.path.join(dirname, filename), arcname=os.path.join("my_zip_dir", filename))