提取並重命名zip文件文件夾

Question

我想從一個zip文件中提取python中的特定文件夾，然後在原始文件名後重命名它。

比如我有一個名爲包含幾個文件夾和子文件夾test.zip：

xl/media/image1.png 
xl/drawings/stuff.png 
stuff/otherstuff.png 

我想提取到一個文件夾名爲test的媒體文件夾的內容： test/image1.png

Answer 1

使用

• zipfile模塊，特別是ZipFile.extractall()
• os.path.splitext() 從字符串test1.zip
• tmpfile.mkdtemp()得到test1創建一個臨時目錄
• shutil.move()移動整個目錄樹。

例如：

#!/usr/bin/env python 
"""Usage: 
./extract.py test.zip 
""" 

from zipfile import ZipFile 
import os 
import sys 
import tempfile 
import shutil 


ROOT_PATH = 'xl/media/' 

zip_name = sys.argv[1] 
zip_path = os.path.abspath(zip_name) 
extraction_dir = os.path.join(os.getcwd(), os.path.splitext(zip_name)[0]) 
temp_dir = tempfile.mkdtemp() 


with ZipFile(zip_path, 'r') as zip_file: 
    # Build a list of only the members below ROOT_PATH 
    members = zip_file.namelist() 
    members_to_extract = [m for m in members if m.startswith(ROOT_PATH)] 
    # Extract only those members to the temp directory 
    zip_file.extractall(temp_dir, members_to_extract) 
    # Move the extracted ROOT_PATH directory to its final location 
    shutil.move(os.path.join(temp_dir, ROOT_PATH), extraction_dir) 

# Uncomment if you want to delete the original zip file 
# os.remove(zip_path) 

print "Sucessfully extracted '%s' to '%s'" % (zip_path, extraction_dir) 

使用try..except塊來處理創建目錄時，刪除文件和提取的zip可能發生的各種異常。