谁说 Python 的 shutil 不支持 7z 解压缩我来教你扩展它的功能

2020-07-30 12:03 AirPython






Python 的内置模块




在 Python 的标准库中,有哪些你常用并且觉得犀利无比的模块?

不要说 time、datetime、os、sys,这些模块常用是常用,但是逼格不够高啊。

举个例子,如果你经常在 LeetCode 上刷题,你会发现有时 Java、C 需要几十行的算法题,如果 Python 使用了 collections、itertools,可能只需要 3、4 代码就完成了!





shutil 的便利




日常的编码中,常会涉及到对文件、目录等的操作场景,如果我们使用 os,可能需要对文件、文件夹,非空等进行逐个判断。

举个例子: 我们现在要删除一个目录,目录中包含有文件与文件夹,如果使用 os 模块,没有现成可以使用的函数,需要我们进行判断与分类执行。

import os
# path是文件的路径,如果这个路径是一个文件夹,
# 则会抛出OSError的错误,这时需用用rmdir()来删除
os.remove(path)
# path是文件夹路径,注意文件夹需空的才能被删除
os.rmdir(path)

多数初学者遇到删除文件夹,想到的操作就是,创建两个列表,然后用 os.walk 遍历目录,将文件与文件夹分别存入初始化的两个列表中,然后先统一删除文件,最后删除文件夹。

如果有上面这样操作的同学,请面壁三分钟。明显没有好好学习 os.walk 函数

os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]]) top -- 是你所要遍历的目录的地址, 返回的是一个三元组(root,dirs,files)。

  • root 所指的是当前正在遍历的这个文件夹的本身的地址
  • dirs 是一个 list ,内容是该文件夹中所有的目录的名字(不包括子目录)
  • files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)

topdown --可选,为 True,则优先遍历 top 目录,否则优先遍历 top 的子目录(默认为开启)。如果 topdown 参数为 True,walk 会遍历top文件夹,与top 文件夹中每一个子目录。

onerror -- 可选,需要一个 callable 对象,当 walk 需要异常时,会调用。

followlinks -- 可选,如果为 True,则会遍历目录下的快捷方式(linux 下是软连接 symbolic link )实际所指的目录(默认关闭),如果为 False,则优先遍历 top 的子目录。

只需要将 topdown 设置为 False,这样在遍历目录时,就会从根节点进行遍历,然后我们逐个删除就 ok 了,哪里需要那么麻烦!

代码如下:

import os

for root, dirs, files in os.walk('D:\\software_temp', topdown=False):
    for name in files:
        os.remove(os.path.join(root, name))
    for name in dirs:
        os.rmdir(os.path.join(root, name))

说这么多,无外乎为了引出最简便的方式 : shutil 模块

如果换做 shutil 模块登场,那么执行删除目录的操作,只需要 0.1 秒的时间

import shutil
shutil.rmtree('D:\\software_temp')

就这样,完事儿了...




文件解压缩




日常工作中,我们经常会使用 Python 进行文件的解压缩处理。

Python 自带的解压缩模块有 zipfile, gzip, tarfile,如果我们需要解压 rar 文件则需要单独下载 rarfile 模块,针对每一种压缩文件,我们都需要针对文件类型进行对应模块的使用

是不是很繁琐?如果我们使用 shutil 呢?让我们先来看看 shutil 支持的解压类型:

import pprint
import shutil
pprint.pprint(shutil.get_unpack_formats())

output:
[('bztar', ['.tar.bz2''.tbz2'], "bzip2'ed tar-file"),
 ('gztar', ['.tar.gz''.tgz'], "gzip'ed tar-file"),
 ('tar', ['.tar'], 'uncompressed tar file'),
 ('xztar', ['.tar.xz''.txz'], "xz'ed tar-file"),
 ('zip', ['.zip'], 'ZIP file')]

shutil 已经包含了我们上面提到的所有文件。





.7z 文件是什么鬼?




众所周知,zip 的压缩率相比 rar 是比较低的,但是商业软件下载中,你很少会见到 .rar 的文件,why?因为专利啊...

RAR是一种专利文件格式,用于数据压缩与归档打包,开发者为尤金·罗谢尔(俄语:Евгений Лазаревич Рошал,拉丁转写:Yevgeny Lazarevich Roshal),RAR的全名是“Roshal ARchive”,即“罗谢尔的归档”之意。首个公开版本RAR 1.3发布于1993年。

所以,有很多产品在软件发布时,开始使用一种压缩率更高的 .7z 文件,这又是为什么?

7-zip 官方主页:

https://sparanoid.com/lab/7z/

在其中有一个许可协议是这样写的

许可协议:

7-Zip 是一款 开源 软件。大多数源代码都基于 GNU LGPL 许可协议下发布。AES 代码基于 BSD 许可下发布。unRAR 代码基于两种许可:GNU LGPL 和 unRAR 限制许可。更多下许可信息请查看:7-Zip 许可。您可以在任何一台计算机上使用 7-Zip ,包括用在商业用途的计算机,不对 7-Zip 进行捐赠或支付并不影响您的使用。





shutil 扩展 7z




说了这么多 7z 文件的好处,可我们看到shutil并不能解压该类型的文件啊。

我们能否让 shutil 支持 .7z 文件,达到无脑解压缩呢?

此时,你需要 py7zr 模块,养成好习惯,遇到模块先找 GitHub

https://github.com/miurahr/py7zr

1. 模块下载

pip install py7zr

2. 基本使用

当我们安装好 py7zr 后,它可以在 cmd 下直接运行该命令

List archive contents
$ py7zr l test.7z
Extract archive
$ py7zr x test.7z
Extract archive with password
$ py7zr x -P test.7z
  password?: ****
Create and compress to archive
$ py7zr c target.7z test_dir
Create multi-volume archive
$ py7zr c -v 500k target.7z test_dir
Test archive
$ py7zr t test.7z
Show information
$ py7zr i
Show version
$ py7zr --version

单独使用模块

import py7zr

archive = py7zr.SevenZipFile('sample.7z', mode='r')
archive.extractall(path="/tmp")
archive.close()

with py7zr.SevenZipFile('target.7z''w'as z:
    z.writeall('./base_dir')
3. shutil 集成

之所以推荐 py7zr 给大家,不仅因为他的简单好用,更是由于他可以轻松集成于 shutil,来看看它的使用方式吧:

from py7zr import pack_7zarchvie, unpack_7zarchive
import shutil

# register file format at first.
shutil.register_archive_format('7zip',
                               pack_7zarchive,
                               description='7zip archive')

shutil.register_unpack_format('7zip',
                              ['.7z'],
                              unpack_7zarchive,
                              description='7zip archive')

# extraction
shutil.unpack_archive('test.7z''/tmp')

# compression
shutil.make_archive('target''7zip''src')

pprint.pprint(shutil.get_unpack_formats())

# output:
[('7zip', ['.7z'], '7zip archive'),
 ('bztar', ['.tar.bz2''.tbz2'], "bzip2'ed tar-file"),
 ('gztar', ['.tar.gz''.tgz'], "gzip'ed tar-file"),
 ('tar', ['.tar'], 'uncompressed tar file'),
 ('xztar', ['.tar.xz''.txz'], "xz'ed tar-file"),
 ('zip', ['.zip'], 'ZIP file')]

通过注册我们看到,shutil 已经支持 7z 文件的解压了,就是如此简单

本文章转载自公众号:AirPython

首页 - Python 相关的更多文章: