Python - 搜索文件和文件ZIP，跨多个目录

Question

这是我第一次将零碎的代码组合在一起形成我需要的实用程序（我是一名设计师），尽管我觉得我已经很接近了，但我在让以下内容正常工作时遇到了困难。

我通常需要压缩我创建的目录结构中带有 .COD 扩展名的文件。作为示例，结构可能如下所示：（

单个根文件夹）-> （多个文件夹）-> （两个文件夹）-> （一个文件夹）-> COD 文件

我需要将所有 COD 文件压缩到 COD.zip 中，并将该 zip 文件放置在当前文件所在目录的上一级目录中。完成后，文件夹结构将如下所示：

EXPORT 文件夹 -> 9800文件夹-> 6文件夹-> OTA 文件夹（+ 新 COD.zip）-> COD 文件

我的问题 -

首先，它创建的 COD.zip 似乎适合其中的 COD 文件，但当我解压它时，里面只有 1 个 .cod，但该 ZIP 的文件大小是所有压缩包的大小COD 压缩在一起。

其次，我需要将 COD 文件压缩，不带任何文件夹结构 - 直接在 COD.zip 中。目前，我的脚本创建了整个目录结构（以“users/mysuername/etc 等”开头）。

任何帮助将不胜感激 - 并且解释会更好，因为我正在努力学习:)

谢谢。

import os, glob, fnmatch, zipfile


def scandirs(path):
for currentFile in glob.glob( os.path.join(path, '*') ):
    if os.path.isdir(currentFile):
        scandirs(currentFile)
    if fnmatch.fnmatch(currentFile, '*.cod'):
            cod = zipfile.ZipFile("COD.zip","a")
            cod.write(currentFile)


scandirs(os.getcwd())

Answer 1

对于问题#1，我认为您的问题可能是这一部分：

cod = zipfile.ZipFile("COD.zip","a")
cod.write(currentFile)

每次写入新文件时，您都会创建一个新的 zip（并可能覆盖现有的）。相反，您希望为每个目录创建一次 zip，然后重复附加到该目录（请参见下面的示例）。

对于问题#2，您的问题是，当您将文件名写入存档时，您可能需要将文件名扁平化。一种方法是使用 os.chdir 将 CD 插入到 scandirs 中的每个目录中。一种更简单的方法是使用 os.path 模块分割文件路径并获取基本名称（不带路径的文件名），然后您可以使用 cod.write 的第二个参数 更改放入实际 zi​​p 中的文件名（请参见下面的示例）。

import os, os.path, glob, fnmatch, zipfile

def scandirs(path):

   #zip file goes at current path, then up one dir, then COD.zip
   zip_file_path = os.path.join(path,os.path.pardir,"COD.zip")
   cod = zipfile.ZipFile(zip_file_path,"a") #NOTE: will result in some empty zips at the moment for dirs that contain no .cod files

   for currentFile in glob.glob( os.path.join(path, '*') ):
      if os.path.isdir(currentFile):
         scandirs(currentFile)
      if fnmatch.fnmatch(currentFile, '*.cod'):
         cod.write(currentFile,os.path.basename(currentFile))

   cod.close()
   if not cod.namelist(): #zip is empty
      os.remove(zip_file_path)

scandirs(os.getcwd())

因此，创建一次 zip 文件，在扁平化文件名的同时重复附加到该文件，然后将其关闭。您还需要确保调用 close，否则可能无法写入所有文件。

目前我没有一个好的方法在本地测试这个，所以请随意尝试并报告回来。我确信我可能弄坏了东西。 ;-)

Answer 2

以下代码具有相同的效果，但可重用性更高，并且不会创建多个 zip 文件。

import os,glob,zipfile

def scandirs(path, pattern):
    result = []
    for file in glob.glob( os.path.join( path, pattern)):
        if os.path.isdir(file):
            result.extend(scandirs(file, pattern))
        else:
             result.append(file)
     return result


zfile = zipfile.ZipFile('yourfile.zip','w')
for file in scandirs(yourbasepath,'*.COD'):
    print 'Processing file: ' + file
    zfile.write(file)                   # folder structure
    zfile.write(file, os.path.split(file)[1])   # no folder structure

zfile.close()