import os
import sys

__all__ = ['DirPath']

'''
    在指定路径下递归查找包含指定字符串的文件
    可以指定查找的文件类型category-默认为'.py'
    可以指定查找的字符串str-默认为'python'
'''


class DirPath(object):
    # 初始化参数查找路径-path
    def __init__(self, path):
        self.show = self.show()
        self.path = path

    # 开启func协程的装饰器
    def on(func):
        def wrapper(*args):
            res = func(*args)
            next(res)
            return res

        return wrapper

    @on
    # 搜索path路径下的python文件
    def search(self, target, category):
        while True:
            path = yield
            for entry in os.scandir(path):
                if entry.is_file():
                    if entry.name.endswith(category):
                        target.send(entry.path)
                if entry.is_dir():
                    self.search(target, category).send(entry.path)

    @on
    # 找到f文件中包含str的行信息并发送给target
    def find_str(self, target, str):
        while True:
            path = yield
            with open(path, "r", encoding='utf-8') as f:
                for (name, value) in enumerate(f):
                    if str in value:
                        target.send(f"[{path}][{name+1}]:{value}")

    @on
    # 展示查询结果
    def show(self):
        while True:
            res = yield
            print(res)

    # 默认在'.py'类型文件中查找字符串-可以指定文件类型category
    # 默认查找字符串'python'-可以指定查找字符串str
    def code_search(self, category=".py", str="python"):
        self.search(self.find_str(self.show, str), category).send(self.path)


if __name__ == '__main__':
    path = sys.argv[1]
    Dir = DirPath(path)
    Dir.code_search(str=sys.argv[2], category=sys.argv[3])