Python3标准库：shutil高层文件操作(3)

当前位置:

首页 > Python基础教程 >

Python3标准库：shutil高层文件操作(3)

 pprint
import shutil

print('BEFORE:')

pprint.pprint(glob.glob('/tmp/example/*'))

shutil.rmtree('/tmp/example')

print('\nAFTER:')

pprint.pprint(glob.glob('/tmp/example/*'))

默认的，如果出现错误，则会作为异常抛出，不过如果第二个参数为true，则可以忽略这些错误。可以在第三参数中提供一个特殊的错误处理函数。

要把一个文件或目录从一个位置移动到另一个位置，可以使用move()。

import glob
import shutil
with open('example.txt', 'wt') as f:
f.write('contents')
print('BEFORE: ', glob.glob('example*'))
shutil.move('example.txt', 'example.out')
print('AFTER : ', glob.glob('example*'))

其语义与UNIX命令mv类似。如果源和目标都在同一个文件系统中，则会重命名源文件。否则，源文件会被复制到目标文件，然后被删除。

1.4 查找文件

which()函数会扫描一个搜索路径以查找一个命名文件。典型的用法是在环境变量PATH定义的shell搜索路径中查找一个可执行程序。

import shutil
print(shutil.which('python'))
print(shutil.which('time'))
print(shutil.which('java'))

如果无法找到与搜索参数匹配的文件，则which()返回None。

1.5 归档

Python的标准库包含很多模块来管理归档文件，如tarfile和zipfile。另外，shutil中还提供了很多更高层函数来创建和解压归档文件。get_archive_formats()返回当前系统上支持的所有格式的名字和描述。

import shutil
for format, description in shutil.get_archive_formats():
print('{:<5}: {}'.format(format, description))

支持的格式取决于有哪些模块和底层库。因此，根据这个例子在哪里运行，它的输出可能会有变化。

shutil维护了一个格式注册表，可以在当前系统上解包；可以通过get_unpack_formats()访问这个注册表。

import shutil
for format, exts, description in shutil.get_unpack_formats():
print('{:<5}: {}, names ending in {}'.format(
format, description, exts))

shutil管理的注册表不同于创建归档所用的注册表，因为它还包括对应各种格式的常见文件扩展名。解压缩归档文件的函数使用这个注册表来根据文件扩展名猜测应当使用哪种格式。

1.6 文件系统空间

完成一个长时间运行的可能耗尽可用的空间的操作之前，最好先检查本地文件系统，来看看有多少可用的空间，这会很有用。disk_usage()会返回一个元组，包括总空间、当前正在使用的空间以及未用的空间(自由空间)。

import shutil
total_b, used_b, free_b = shutil.disk_usage('.')
gib = 2 ** 30 # GiB == gibibyte
gb = 10 ** 9 # GB == gigabyte
print('Total: {:6.2f} GB {:6.2f} GiB'.format(
total_b / gb, total_b / gib))
print('Used : {:6.2f} GB {:6.2f} GiB'.format(
used_b / gb, used_b / gib))
print('Free : {:6.2f} GB {:6.2f} GiB'.format(
free_b / gb, free_b / gib))

disk_usage()返回的值以字节为单位，所以这个示例程序在打印之前先把它转换为更可读的一些单位。

栏目列表