Python Zipfile模塊：如何壓縮和解壓文件

在日常的軟件開發過程中，文件的壓縮和解壓是我們經常遇到的任務。在Python中，使用Zipfile模塊來完成文件的壓縮和解壓非常方便。在本文中，我們將從以下幾個方面詳細介紹Zipfile模塊的使用方法：

一、壓縮單個文件

我們首先看一下如何使用Zipfile模塊來壓縮單個文件。Zipfile模塊提供了一個ZipFile類，我們可以通過實例化一個ZipFile類來進行文件的壓縮。下面是一個示例代碼：

import zipfile
 
def zip_file(file_path):
    zf = zipfile.ZipFile(file_path + '.zip', mode='w', allowZip64=True)
    try:
        zf.write(file_path, compress_type=zipfile.ZIP_DEFLATED)
    finally:
        zf.close()

在上面的代碼中，我們使用Zipfile模塊的ZipFile類創建了一個zip文件對象，並指定了文件的壓縮模式（mode）。使用write()方法來將文件添加到壓縮文件中，並指定壓縮的方式（compress_type）。

二、壓縮多個文件

除了壓縮單個文件之外，Zipfile模塊還提供了壓縮多個文件的方法。我們可以使用ZipFile類的write()方法，每次傳入一個文件路徑即可實現多個文件的壓縮。下面是一個示例代碼：

import os
import zipfile
  
def zip_dir(dirname, zipfilename):
    filelists = []
  
    if os.path.isfile(dirname):
        filelists.append(dirname)
    else:
        for root, dirs, files in os.walk(dirname):
            for name in files:
                filelists.append(os.path.join(root, name))
  
    zf = zipfile.ZipFile(zipfilename, "w", zipfile.ZIP_DEFLATED)
    for tar in filelists:
        arcname = tar[len(dirname):]
        zf.write(tar, arcname)
    zf.close()

在上面的代碼中，我們使用os.walk()方法遍歷目錄下所有的文件，然後使用ZipFile類的write()方法將文件添加到壓縮文件中。由於ZipFile類支持壓縮目錄，我們在壓縮時需要指定文件的根目錄。

三、解壓縮文件

如果我們需要對已經壓縮的文件進行解壓，同樣可以使用Zipfile模塊。Zipfile模塊提供了ZipFile類的extract()和extractall()方法，可以用來分別解壓縮單個文件和壓縮目錄下的所有文件。下面是一個示例代碼：

import zipfile
 
def unzip_file(zip_file_path, unzip_path):
    zf = zipfile.ZipFile(zip_file_path)
    try:
        zf.extractall(path=unzip_path)
    except Exception as e:
        print(e)
    finally:
        zf.close()

在上面的代碼中，我們使用ZipFile類的extractall()方法解壓縮文件到指定的路徑。在解壓縮時，我們需要注意文件的編碼以及路徑的長度等問題，以避免解壓縮失敗。

四、壓縮文件時保留文件權限

在壓縮文件時，有時我們需要保留原有的文件權限。Zipfile模塊提供了ZipFile類的writepy()方法，可以在壓縮文件時保留文件權限。下面是一個示例代碼：

import os
import zipfile
  
def zip_dir_with_permission(dirname, zipfilename):
    filelists = []
  
    if os.path.isfile(dirname):
        filelists.append(dirname)
    else:
        for root, dirs, files in os.walk(dirname):
            for name in files:
                filelists.append(os.path.join(root, name))
  
    zf = zipfile.ZipFile(zipfilename, "w", zipfile.ZIP_DEFLATED, True)
    for tar in filelists:
        arcname = tar[len(dirname):]
        zf.write(tar, arcname)
    zf.close()

在上面的代碼中，我們在創建ZipFile對象時，將保留權限參數(True)傳給ZipFile類的構造函數。在每次向壓縮文件中添加文件時，ZipFile類都會保留原有的文件權限。

五、結語

通過本文的介紹，相信大家已經掌握了使用Python ZipFile模塊進行文件壓縮和解壓縮的基本技巧。如果大家還有什麼問題或者疑惑，歡迎在評論區留言，我們一起學習探討。