Python标准库之zipfile和tarfile模块的使用

Python标准库之zipfile和tarfile模块的使用攻略

简介

压缩文件在网页和文件传输中是很常见的，很多时候我们需要用到Python对压缩文件的操作。Python标准库中提供了两个模块zipfile和tarfile，可以用来读取、写入、解压缩、压缩zip、tar和其他常见的压缩格式文件。

本篇攻略将介绍zipfile和tarfile模块的使用方法，包括它们的常用的方法和实际应用的示例。

zipfile模块

zipfile模块提供了一种在Python中管理zip文件(存档，或存档文件)的方式。它使用标准的zip文件格式。使用zipfile模块可以：

• 向zip文件添加文件或文件夹
• 从zip文件中提取文件或文件夹
• 列出zip文件中的文件
• 读取zip文件中的文件的内容
• 在zip文件中删除文件或文件夹

ZipFile打开和关闭

打开压缩文件并返回ZipFile对象，可以使用以下语法：

import zipfile
with zipfile.ZipFile("example.zip", "r") as zip_file:
    # operations to do with zip file

zipfile.ZipFile 接收两个参数：

• zipfile 必须指定待打开的压缩文件名或文件对象。
• mode 是可选的字符串参数。它指定zip在打开文件时的模式。默认模式是"r"，表示打开zip文件，以便读取其内容。

使用with语句来确保文件在使用完成后被关闭。

ZipFile增加和提取文件

要向zip文件中添加文件或文件夹，请使用 write() 方法。此方法接收文件名或目录名，注意：目录名是将添加目录下的所有文件。例如：

with zipfile.ZipFile("example.zip", "w") as zip_file:
    zip_file.write("file1.txt")
    zip_file.write("directory1")

在上面的代码示例中，使用了"example.zip"打开zip文件，然后使用 write() 方法，添加了一个 file1.txt 文件和一个名为 directory1 的文件夹。

要从zip文件中提取文件或文件夹，请使用 extract() 方法。此方法接受文件名或目录名作为参数。例如：

with zipfile.ZipFile("example.zip", "r") as zip_file:
    zip_file.extract("file1.txt")
    zip_file.extract("directory1")

在上面的代码示例中，使用"example.zip"打开zip文件，然后使用 extract() 方法，从zip文件中提取出了一个 file1.txt 文件和一个名为 directory1 的文件夹。

ZipFile列表

要列出zip文件中的所有文件，请使用 namelist() 方法。例如：

with zipfile.ZipFile("example.zip", "r") as zip_file:
    files_list = zip_file.namelist()
    print(files_list)

在上面的代码示例中，使用"example.zip"打开zip文件，使用 namelist() 方法获取压缩文件中的所有文件，然后打印出来。

ZipFile读取文件内容

要在zip文件中读取文件的内容，请使用 open() 方法。此方法接收文件名和模式作为参数。这里的模式与Python内置的"open()"方法中的模式相同。例如：

with zipfile.ZipFile("example.zip", "r") as zip_file:
    with zip_file.open("file1.txt", "r") as file:
        print(file.read().decode("utf-8"))

在上面的代码示例中，使用 "example.zip" 打开zip文件，使用 open() 方法读取文件1.txt中的内容，然后打印出来。

ZipFile删除文件

要从zip文件中删除文件，请使用 remove() 方法。此方法接收文件名作为参数。例如：

with zipfile.ZipFile("example.zip", "w") as zip_file:
    zip_file.write("file1.txt")
    zip_file.write("file2.txt")

with zipfile.ZipFile("example.zip", "a") as zip_file:
    zip_file.remove("file2.txt")

在上面的代码示例中，首先使用 "example.zip" 打开zip文件，使用 write() 方法添加了两个文件。然后，使用 remove() 方法，从zip文件中删除了一个文件。最后，使用 append 打开zip文件，并且删除的文件不再在zip文件中。

tarfile模块

tarfile模块提供了一种在Python中管理 tar存档文件的方式。它使用标准的tar文件格式。使用tarfile模块可以：

• 创建tar文件
• 向tar文件添加文件或文件夹
• 从tar文件中提取文件或文件夹
• 列出tar文件中的文件
• 读取tar文件中的文件的内容
• 在tar文件中删除文件或文件夹

tarfile打开和关闭

打开一个 tar 文件并返回一个 TarFile 对象的方法与 zipfile.ZipFile 相似，只是操作的文件类型不同。可以使用以下语法：

import tarfile
with tarfile.open("example.tar", "r") as tar_file:
    # operations to do with tar file

tarfile.open 接收两个参数：

• tarfile 必须指定待打开的tar文件名或文件对象。
• mode 是可选的字符串参数。它指定tar在打开文件时的模式。默认模式是 "r" ，它表示打开 tar 文件，以便读取其内容。

使用with语句来确保文件在使用完成后被关闭。

tarfile增加和提取文件

要将文件或文件夹添加到 tar 文件中，请使用 add() 方法：

with tarfile.open("example.tar", "w") as tar_file:
    tar_file.add("file1.txt")
    tar_file.add("directory1")

在上面的代码示例中，使用 "example.tar" 打开 tar 文件，然后使用 add() 方法，添加了一个 file1.txt 文件和一个名为 directory1 的文件夹。

要从 tar 文件中提取文件或文件夹，请使用 extract() 方法。

with tarfile.open("example.tar", "r") as tar_file:
    tar_file.extract("file1.txt")
    tar_file.extract("directory1")

在上面的代码示例中，使用 "example.tar" 打开 tar 文件，然后使用 extract() 方法，从 tar 文件中提取出了一个 file1.txt 文件和一个名为 directory1 的文件夹。

tarfile列表

要列出tar文件中的所有文件，请使用 getnames() 方法。例如：

with tarfile.open("example.tar", "r") as tar_file:
    files_list = tar_file.getnames()
    print(files_list)

在上面的代码示例中，使用 "example.tar" 打开tar文件，使用 getnames() 方法获取tar文件中的所有文件，然后打印出来。

tarfile读取文件内容

要在tar文件中读取文件的内容，请使用 extractfile() 方法。此方法接收文件名作为参数。例如：

with tarfile.open("example.tar", "r") as tar_file:
    with tar_file.extractfile("file1.txt") as file:
        print(file.read().decode("utf-8"))

在上面的代码示例中，使用 "example.tar" 打开tar文件，使用 extractfile() 方法读取文件1.txt中的内容，然后打印出来。

tarfile删除文件

要从tar文件中删除文件，请使用 remove() 方法。此方法接收文件名作为参数。例如：

with tarfile.open("example.tar", "w") as tar_file:
    tar_file.add("file1.txt")
    tar_file.add("file2.txt")

with tarfile.open("example.tar", "a") as tar_file:
    tar_file.remove("file2.txt")

在上面的代码示例中，首先使用 "example.tar" 打开tar文件，使用 add() 方法添加了两个文件。然后，使用 remove() 方法，从tar文件中删除了一个文件。最后，使用 append 打开tar文件，并且删除的文件不再在tar文件中。

示例

示例一：将文件夹压缩成zip文件

下面是一个示例，将当前工作目录下的文件夹 test 压缩成 test.zip 文件:

import zipfile  
import os  

def zip_dir(dirname, zipfilename):  
    filelist = []  

    if os.path.isfile(dirname):  
        filelist.append(dirname)  
    else :  
        for root, dirs, files in os.walk(dirname):  
            for name in files:  
                filelist.append(os.path.join(root, name))  

    zf = zipfile.ZipFile(zipfilename, "w", zipfile.zlib.DEFLATED)  
    for tar in filelist:  
        arcname = tar[len(dirname):]  
        print arcname  
        zf.write(tar, arcname)  
    zf.close()  

zip_dir("test","test.zip") 

示例二：将文件夹压缩成tar文件

下面是一个示例，将当前工作目录下的文件夹 test 压缩成 test.tar文件:

import tarfile
import os

def make_tarfile(output_filename, source_dir):
    with tarfile.open(output_filename, "w:gz") as tar:
        tar.add(source_dir, arcname=os.path.basename(source_dir))

make_tarfile("test.tar", "test")

小结

以上介绍了zipfile和tarfile模块的使用方法，这两个模块都是Python标准库的一部分，所以不需要额外安装。在实际应用中，可以按需使用这些模块，处理压缩文件。