42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import os
|
||
from fnmatch import fnmatchcase
|
||
|
||
if __name__ == "__main__":
|
||
base_path = "D:\Code\Learn\practice"
|
||
|
||
# 想要建立一个处理管道，需要将几个生成器函数堆叠，比如，我现在要输出所有practice文件夹下的注释行内容
|
||
# 1.首先生成文件夹路径
|
||
def gen_dir_path(fpath):
|
||
for path, dir, files in os.walk(fpath):
|
||
for file in files:
|
||
if (not fnmatchcase(path, "*.[git][idea]*")) and (not fnmatchcase(file, "*.md")):
|
||
yield os.path.join(path, file)
|
||
|
||
# 2.打开文件
|
||
def gen_file(paths):
|
||
for path in paths:
|
||
f = open(path, 'rt', encoding='utf-8')
|
||
yield f
|
||
f.close()
|
||
|
||
# 3.读取文件内容
|
||
def gen_file_txt(opened_files):
|
||
for file in opened_files:
|
||
yield from file
|
||
print("done")
|
||
|
||
# 4.匹配符合条件的文件行
|
||
def read_file_lines(file_lines):
|
||
for line in file_lines:
|
||
if "#" in line:
|
||
print(line)
|
||
|
||
# 将这些函数堆在一起就组成了一条管道，每次生成一个对象进行流水线处理，
|
||
# 这样避免了for循环的嵌套
|
||
paths = gen_dir_path(base_path)
|
||
files = gen_file(paths)
|
||
lines = gen_file_txt(files)
|
||
read_file_lines(lines)
|
||
|
||
|