python\列表和生成器表达式

import os
 
def search(dir_name, partten):
    g = os.walk(dir_name)
    res = {"",}
    for i in g:
        for j in i[-1]:
            file_path = i[0] + "\\" +j
            with open(file_path) as f:
                for line in f:
                    if partten in line:
                        res.add(file_path)
    for line in res:
        if line:
            print(line)
 
search("c:\\test", "python")

利用生成器来处理
import os
 
def firstNext(func):
    def wrapper(*args, **kwargs):
        g = func(*args, **kwargs)
        next(g)
        return g
    return wrapper
 
@firstNext
def search(target):
    while True:
        dir_name = yield
        g = os.walk(dir_name)
        for i in g:
            for j in i[-1]:
                file_path = i[0] + "\\" +j
                target.send(file_path)
 
@firstNext
def opener(target):
    while True:
        file_path = yield
        with open(file_path) as f:
            target.send((file_path, f))
 
@firstNext
def cat(target):
    while True:
        file_path, f = yield
        for line in f:
            target.send((file_path, line))
 
@firstNext
def grep(partten, target):
    while True:
        file_path, line = yield
        if partten in line:
            target.send(file_path)
 
@firstNext
def printer():
    while True:
        file_path = yield
        print(file_path)
 
g = search(opener(cat(grep("python",printer()))))
g.send("c:\\test")