reader.py 文件源码

def get_paths(path: str, ext: str) -> Iterator[str]:
        """
        ????????? ???? ?????? ????????? ???? ?? ????????? ????.

        :param path: ???? ? ?????/?????.
        :param ext: ????????? ??????????.
        """
        if os.path.isfile(path):
            if ext == os.path.splitext(path)[1]:
                yield path
        else:
            for root, folders, files in os.walk(path):
                for file in files:
                    if ext == os.path.splitext(file)[1]:
                        yield os.path.join(root, file)
                for folder in folders:
                    return Reader.get_paths(folder, ext)