python rpa_（RPA学习）Python 之 Pathlib 模块

原标题：(RPA学习)Python 之 Pathlib 模块

艺赛旗 RPA9.0全新首发免费下载点击下载

http://www.i-search.com.cn/index.html?from=line9

读文件和写文件

在我们使用 open 来操作文件读写操作的时候, 不仅可以使用字符串格式的路径，对于 pathlib 生成的路径完全可以直接使用:

path = pathlib.Path.cwd() / 'test.md'

with open(path, mode='r') as fid:

headers = [line.strip() for line in fid if line.startswith('#')]

print('\n'.join(headers))

或者在 pathlib 的基础使用 open, 我推荐使用下面的方式:

import pathlib

DIR_PATH = pathlib.Path("/Users/chennan/CDM") / "2000" / "hehe.txt"

with DIR_PATH.open("r") as fs:

data = fs.read()

print(data)

这样写的好处就是 open 里面我们不需要再去传入路径了，直接指定文件读写模式即可。实际上这里的 open 方法，底层也是调用了 os.open 的方法。使用哪种方式看个人的喜好。

pathlib 还提供几种文件的读写方式:

可以不用再使用 with open 的形式即可以进行读写。

.read_text(): 找到对应的路径然后打开文件，读成str格式。等同open操作文件的"r"格式。

.read_bytes(): 读取字节流的方式。等同open操作文件的"rb"格式。

.write_text(): 文件的写的操作，等同open操作文件的"w"格式。

.write_bytes(): 文件的写的操作，等同open操作文件的"wb"格式。

使用 resolve 可以通过传入文件名，来返回文件的完整路径，使用方式如下

import pathlib

py_path =pathlib.Path("superdemo.py")

print(py_path.resolve())

输出

/Users/chennan/pythonproject/demo/superdemo.py

需要注意的是 "superdemo.py" 文件要和我当前的程序文件在同一级目录。

选择路径的不同组成部分

pathlib 还提供了很多路径操作的属性，这些属性可以选择路径的不用部位，如

.name: 可以获取文件的名字，包含拓展名。

.parent: 返回上级文件夹的名字

.stem: 获取文件名不包含拓展名

.suffix: 获取文件的拓展名

.anchor: 类似盘符的一个东西,

import pathlib

now_path = pathlib.Path.cwd() / "demo.txt"

print("name",now_path.name)

print("stem",now_path.stem)

print("suffix",now_path.suffix)

print("parent",now_path.parent)

print("anchor",now_path.anchor)

输出：

name demo.txt

stem demo

suffix .txt

parent /Users/chennan/pythonproject/demo

anchor /

移动和删除文件

当然 pathlib 还可以支持文件其他操作，像移动，更新，甚至删除文件，但是使用这些方法的时候要小心因为，使用过程不用有任何的错误提示即使文件不存在也不会出现等待的情况。

使用 replace 方法可以移动文件，如果文件存在则会覆盖。为避免文件可能被覆盖，最简单的方法是在替换之前测试目标是否存在。

import pathlib

destination = pathlib.Path.cwd() / "target"

source = pathlib.Path.cwd() / "demo.txt"

if not destination.exists():

source.replace(destination)

但是上面的方法存在问题就是，在多个进程多 destination 进行的操作的时候就会现问题，可以使用下面的方法避免这个问题。也就是说上面的方法适合单个文件的操作。

import pathlib

destination = pathlib.Path.cwd() / "target"

source = pathlib.Path.cwd() / "demo.txt"

with destination.open(mode='xb') as fid:

#xb表示文件不存在才操作

fid.write(source.read_bytes())

当 destination 文件存在的时候上面的代码就会出现 FileExistsError 异常。

从技术上讲，这会复制一个文件。要执行移动，只需在复制完成后删除源即可。

使用 with_name 和 with.shuffix 可以修改文件名字或者后缀。

import pathlib

source = pathlib.Path.cwd() / "demo.py"

source.replace(source.with_suffix(".txt")) #修改后缀并移动文件，即重命名

可以使用.rmdir()和.unlink() 来删除文件。

几个 pathlib 的使用例子

统计文件个数

我们可以使用.iterdir 方法获取当前文件下的所以文件.

import pathlib

from collections import Counter

now_path = pathlib.Path.cwd()

gen = (i.suffix for i in now_path.iterdir())

print(Counter(gen))

输出：

Counter({'.py': 16, '': 11, '.txt': 1, '.png': 1, '.csv': 1})

通过配合使用 collections 模块的 Counter 方法，我们获取了当文件夹下文件类型情况。

前面我们说过 glob 模块点这里了解【glob 模块的使用】，同样的 pathlib 也有 glob 方法和 rglob 方法，不同的是 glob 模块里的 glob 方法结果是列表形式的，iglob 是生成器类型，在这里 pathlib 的 glob 模块返回的是生成器类型，然后 pathlib 还有一个支持递归操作的 rglob 方法。

下面的这个操作我通过使用 glob 方法，设定规则进行文件的匹配。

import pathlib

from collections import Counter

gen =(p.suffix for p in pathlib.Path.cwd().glob('*.py'))

print(Counter(gen))

展示目录树

下一个示例定义了一个函数 tree()，该函数的作用是打印一个表示文件层次结构的可视树，该树以一个给定目录为根。因为想列出其子目录，所以我们要使用.rglob() 方法：

import pathlib

from collections import Counter

def tree(directory):

print(f'+ {directory}')

for path in sorted(directory.rglob('*')):