Python Zip: 简单高效的文件压缩与解压方案

Python 笔记

文件压缩与解压是我们在日常开发和生活中常常会遇到的问题，比如，我们需要将文件打包传递，或者需要将大容量文件压缩以便于传输，这些都是常见的需求。而Python语言自带的zipfile模块，提供了一种简单高效的文件压缩与解压方式，本文将对其进行详细介绍。

一、zipfile模块基本用法

zipfile模块是Python自带的一个压缩文件处理模块，我们可以通过它来创建、读取和解压zip格式的压缩文件。下面是一些基本用法：

import zipfile

# 创建一个压缩文件
with zipfile.ZipFile('test.zip', 'w') as myzip:
    myzip.write('file1.txt')
    myzip.write('file2.txt')

# 读取一个压缩文件
with zipfile.ZipFile('test.zip', 'r') as myzip:
    myzip.printdir()
    myzip.extractall()

上面的代码展示了zipfile模块的两个基本操作，分别是创建一个压缩文件和读取一个压缩文件。我们可以通过with语句来打开指定的文件并进行压缩/解压/读取等操作。其中，ZipFile类提供了多个方法，例如：write()用于将一个文件写入压缩包，extractall()用于将压缩文件全部解压，printdir()用于打印压缩文件的目录结构。

二、压缩文件夹

除了压缩单个文件，zipfile还支持压缩整个文件夹，这对于需要压缩多个相关文件的场景非常有用。例如，我们需要将某个项目的所有源代码打包成一个zip文件，就可以使用下面的代码：

import zipfile
import os

# 压缩整个文件夹
def zipdir(dirname, zipfilename):
    with zipfile.ZipFile(zipfilename, 'w') as myzip:
        for root, dirs, files in os.walk(dirname):
            for file in files:
                myzip.write(os.path.join(root, file))

zipdir('myproject', 'myproject.zip')

上面的代码定义了一个zipdir方法，通过os.walk函数遍历整个文件夹，并将所有文件写入压缩包中。这样，就能够将整个文件夹压缩成一个zip文件。

三、解压某个文件

有时候我们不需要将整个压缩文件解压，而只需要解压其中一个文件。这时，我们可以使用ZipFile类提供的extract()方法：

import zipfile

# 解压某个文件
with zipfile.ZipFile('test.zip', 'r') as myzip:
    myzip.extract('file1.txt')

上面的代码解压了test.zip中的file1.txt文件，解压后的文件会放在当前目录下。

四、压缩文件时设置密码

对于一些需要保密的文件，我们可能需要加上密码才能够解压。zipfile提供了这个功能，我们可以使用ZipFile类的setpassword()方法来设置密码：

import zipfile

# 设置密码
password = 'mypassword'
with zipfile.ZipFile('test.zip', 'w') as myzip:
    myzip.setpassword(password)
    myzip.write('file1.txt')

# 解压需要密码的压缩文件
with zipfile.ZipFile('test.zip', 'r') as myzip:
    myzip.extractall(pwd=password.encode())

上面的代码分别演示了如何设置密码和如何解压需要密码的压缩文件。需要注意的是，密码必须是字符串类型，并且需要进行编码。

五、压缩文件时设置压缩方式

除了默认的zip压缩方式，zipfile还支持其他压缩方式，例如gzip压缩方式、bz2压缩方式等。我们可以使用ZipFile类的write()方法中的参数来指定压缩方式：

import zipfile

# 使用gz压缩方式
with zipfile.ZipFile('test.gz', 'w') as myzip:
    myzip.write('file1.txt', compress_type=zipfile.ZIP_DEFLATED)

上面的代码演示了如何使用gzip压缩方式，需要注意的是，在写入文件的时候要将压缩方式指定为zipfile.ZIP_DEFLATED。

六、小结

本文介绍了Python自带的zipfile模块，展示了它的基本用法和一些高级特性。通过本文的学习，我们可以轻松地进行文件压缩和解压，并且对压缩方式、密码等进行定制化设置。zipfile是Python中非常实用的一个模块，在实际工作和生活中经常会用到，希望本文能够帮助大家更好地掌握这个模块。