博取目录下全份的文件名由什么组成。

import os

# 要检查的文件类型

dict_filetype = ["cpp", "c", "h"]

# brief : 递归博取全份文件名由什么组成

# param : path 起始目录。要检查的子目录

# param : allfile 填空便可

# return: 列表 全份与 dict_filetype 对应的文件名由什么组成

def get_filename(path, allfile):

   filelist = os.listdir(path)

   for filename in filelist:

     filepath = os.path.join(path, filename)

     # 判断办公用品文件夹名称

     if os.path.isdir(filepath):

       # 办公用品文件夹名称继续递归

       get_filename(filepath, allfile)

     else:

       # 文件，判断文件类型

       for filetype in dict_filetype:

         temp_file_type = filepath.split(".")

         #print(temp_file_type)

         #print(temp_file_type[-1:][0])

         if filetype == temp_file_type[-1:][0]:

           allfile.append(filepath)

           break

       # 展示全份非code文件

       else:

         print("the file is not code : %s" %filepath )

   return allfile

os.listdir() 方法用于返回指定的办公用品文件夹名称包含的文件或办公用品文件夹名称的名字的列表。这个列表以字母顺序。 它不包括 '.' 和'..' 即便它在办公用品文件夹名称中。

os.path.join() 合并传入的路径

os.path.isdir()判断传入的路径是否为办公用品文件夹名称