Question

下面的python代码会获取一个文件列表并将它们拉上来。我需要的唯一文件地理数据库（基于文件的数据库）称为“数据”，那么如何修改循环以仅包含名为Data的基于文件的数据库？更具体地说，文件地理数据库存储为包含存储和管理空间数据的二进制文件的系统文件夹。所以我需要名为Data.gdb的整个系统文件夹。

非常感谢

#**********************************************************************
# Description:
#    Zips the contents of a folder, file geodatabase or ArcInfo workspace
#    containing coverages into a zip file.
# Parameters:
#   0 - Input workspace
#   1 - Output zip file. It is assumed that the caller (such as the
#       script tool) added the .zip extension.
#
#**********************************************************************

# Import modules and create the geoprocessor
import sys, zipfile, arcgisscripting, os, traceback
gp = arcgisscripting.create()

# Function for zipping files 
def zipws(path, zip):
    isdir = os.path.isdir

    # Check the contents of the workspace, if it the current
    # item is a directory, gets its contents and write them to
    # the zip file, otherwise write the current file item to the
    # zip file
    #
    for each in os.listdir(path):
        fullname = path + "/" + each
        if not isdir(fullname):
            # If the workspace is a file geodatabase, avoid writing out lock
            # files as they are unnecessary
            #
            if not each.endswith('.lock'):
                # gp.AddMessage("Adding " + each + " ...")
                # Write out the file and give it a relative archive path
                #
                try: zip.write(fullname, each)
                except IOError: None # Ignore any errors in writing file
        else:
            # Branch for sub-directories
            #
            for eachfile in os.listdir(fullname):
                if not isdir(eachfile):
                    if not each.endswith('.lock'):
                        # gp.AddMessage("Adding " + eachfile + " ...")
                        # Write out the file and give it a relative archive path
                        #
                        try: zip.write(fullname + "/" + eachfile, \
                                       os.path.basename(fullname) + "/" + eachfile)
                        except IOError: None # Ignore any errors in writing file


if __name__ == '__main__':
    try:
        # Get the tool parameter values
        #
        inworkspace = sys.argv[1]
        outfile = sys.argv[2]     

        # Create the zip file for writing compressed data
        #
        zip = zipfile.ZipFile(outfile, 'w', zipfile.ZIP_DEFLATED)
        zipws(inworkspace, zip)
        zip.close()

        # Set the output derived parameter value for models
        #
        gp.setparameterastext(1, outfile)
        gp.AddMessage("Zip file created successfully")

    except:
        # Return any python specific errors as well as any errors from the geoprocessor
        #
        tb = sys.exc_info()[2]
        tbinfo = traceback.format_tb(tb)[0]
        pymsg = "PYTHON ERRORS:\nTraceback Info:\n" + tbinfo + "\nError Info:\n    " + \
                str(sys.exc_type)+ ": " + str(sys.exc_value) + "\n"
        gp.AddError(pymsg)

        msgs = "GP ERRORS:\n" + gp.GetMessages(2) + "\n"
        gp.AddError(msgs)

Answer 1

遍历目录树的最佳方法是os.walk - 为您执行文件/目录分离，并为您执行递归到子目录。

所以：

def zipws(path, zip, filename='Data.gdb'):
  for root, dirs, files in os.walk(path):
    if filename in files:
      zip.write(os.path.join(root, filename),
                os.path.join(os.path.basename(root), filename))
      return

我不确定我是否已经捕获了您想要确定zip.write的两个参数的整个逻辑（这对我来说对您来说并不明显），但是，如果没有，那应该很容易调整。

此外，我不确定你最后是否想要return：效果只是压缩一个文件，而不是压缩所有< / strong>以树的形式命名的文件（在各自的子目录中）。如果你知道只有一个这样的文件，也可以留下return（它会加快一点速度）。如果您想要所有此类文件，请删除return。

编辑：事实证明OP想要的“一件事”是目录，而不是文件。在这种情况下，我建议，作为最简单的解决方案：

def zipws(path, zip, dirname='Data.gdb'): for root, dirs, files in os.walk(path): if os.path.basename(root) != dirname: continue for filename in files: zip.write(os.path.join(root, filename), os.path.join(dirname, filename)) return

再次用类似的猜测来确定你想要用于存档名称的的完全谜团。

Answer 2

从这一行开始：

    zipws(inworkspace, zip)

您不希望使用此函数从多个文件构建zip文件。看起来你想要只用一个成员构建一个zip文件。

将其替换为此。

     try: 
         zip.write(os.path.join('.', 'Data.gdb'))
     except IOError: 
         pass # Ignore any errors in writing file

丢掉您显然不想使用的zipws功能。

阅读本文，它可能会有所帮助：http://docs.python.org/library/zipfile.html

在Python中更新For Each循环

2 个答案: