##貌似是判断文件那有问题?可是不知道怎么写啊。。。有没有大神帮忙看看
class loadFiles(object):
def __init__(self,par_path): # par_path 文件夹的路径
self.par_path = par_path
def __iter__(self):
for file in os.listdir(self.par_path):
print('1. ',file)
file_abspath = os.path.join(self.par_path,file)
print('2. ',file_abspath)
if os.path.isdir(file_abspath): #如果还是一个文件夹
loadFiles(file_abspath)
elif os.path.isfile(file_abspath): # 如果还是一个文件夹
for f in os.listdir(file_abspath): # secondary directory
catg = file_abspath.split(os.sep)[-2]
print(catg, f)
yield catg, f