Python 提取 zip 文件时出错

Question

我只知道如何为 GIS 目的编写 python。使用 arcpy 和地理处理工具的代码还有更多内容......这只是我坚持的开始部分，它是为了尝试准备好数据，这样我就可以将压缩文件夹中的 shapefile 用于我的其余部分脚本

我正在尝试提示用户输入要搜索的目录。为了使用这个脚本，它将搜索一个压缩的 zip 文件，然后将所有文件解压缩到同一个目录。

import zipfile, os


# ask what directory to search in 
mypath = input("Enter .zip folder path: ")
extension = ".zip"

os.chdir(mypath) # change directory from working dir to dir with files

for item in os.listdir(mypath):
    if item.endswith(extension):
        filename = os.path.abspath(item)
        zip_ref = zipfile.ZipFile(filename)
        zip_ref.extractall(mypath)
        zip_ref.close()

尝试了你们的建议，但仍然存在以下问题：

import zipfile, os

mypath = input("Enter folder: ")


if os.path.isdir(mypath):
    for dirpath, dirname, filenames in os.listdir(mypath):
        for file in filenames:
            if file.endswith(".zip"):
                print(os.path.abspath(file))
                with zipfile.ZipFile(os.path.abspath(file)) as z:
                    z.extractall(mypath)

else:
    print("Directory does not exist.")

Answer 1

我不确定arcpy 的使用。不过……

要遍历目录中的条目，请使用os.listdir：

for entry_name in os.listdir(directory_path): 
    # ...

在循环中，entry_name 将是directory_path 目录中项目的名称。

检查是否以“.zip”结尾时，请记住比较区分大小写。在使用str.endswith 时，您可以使用str.lower 有效地忽略大小写：

if entry_name.lower().endswith('.zip'):
        # ...

要获取条目的完整路径（在本例中为您的 .zip），请使用 os.path.join：

entry_path = os.path.join(directory_path, entry_name)

将此完整路径传递给zipfile.ZipFile。

Answer 2

如果位置无效，您的第一个 if 块将否定 else 块。我会完全删除“else”运算符。如果你保留它，if-check 会有效地杀死程序。 "if folderExist" 足以替代 else。

import arcpy, zipfile, os


# ask what directory to search in 
folder = input("Where is the directory? ")
# set workspace as variable so can change location
arcpy.env.workspace = folder

# check if invalid entry - if bad, ask to input different location
if len(folder) == 0:
    print("Invalid location.")
    new_folder = input("Try another directory?")
    new_folder = folder
    # does the above replace old location and re set as directory location?



# check to see if folder exists
folderExist = arcpy.Exists(folder)
if folderExist:
    # loop through files in directory
        for item in folder:
        # check for .zip extension
            if item.endswith(".zip"):
                 file_name = os.path.abspath(item) # get full path of files
                 print(file_name)
                 zip_ref = zipfile.ZipFile(file_name) # create zipfile object
                 zip_ref.extractall(folder) # extract all to directory
                 zip_ref.close() # close file

如果您可以不使用原始代码，这可能会更简洁：

import zipfile, os
from tkinter import filedialog as fd

# ask what directory to search in 
folder = fd.askdirectory(title="Where is the directory?")

# loop through files in directory
for item in os.listdir(folder):
     # check for .zip extension
     if zipfile.is_zipfile(item):
         file_name = os.path.abspath(item) # get full path of files
         # could string combine to ensure path
         # file_name = folder + "/" + item
         print(file_name)

         zip_ref = zipfile.ZipFile(file_name) # create zipfile object
         zip_ref.extractall(folder) # extract all to directory
         zip_ref.close() # close file

Answer 3

[更新]

我可以用下面的代码解决这个问题

import os
import zipfile

mypath = raw_input('Enter Folder: ')

if os.path.isdir(mypath):
    for file in os.listdir(mypath):
        if file.endswith('.zip'):
            with zipfile.ZipFile(os.path.join(mypath, file)) as z:
                z.extractall(mypath)
else:
    print('Directory does not exist')