python从目录树打印.c文件的路径名

时间:2016-07-21 11:09:18

标签: python recursion tree directory

我是Python的初学者。 我正在尝试使用包含可变深度的测试用例(。文件)的目录树,例如,test(top目录)有2个子test1和test 2. test1有更多分支t1和t2,其中包含t1.c和t2 .c分别。同样,test2有2个孩子test21和test22。测试21具有t3和t4,其中包含t3.c和t4.c。 test22的t5和t6有t5.c和t6.c. 我正在尝试搜索.c文件,然后使用递归打印其路径。

我得到的错误是代码仅通过一个分支进行跟踪,并且不会返回到上一级别以通过其他分支。请帮帮我。

以下是我的代码:

# assume, at no level folders and .c files are present at the same time

import os import fnmatch


# move to test directory 
os.chdir('/home/gyadav/test') 


# function to create a list of sub directories 
def create_subdir_list():
     subdirs = os.listdir('.')
     len_subdirs = len(subdirs) - 1
     # while i != length
     for i in range(0, len_subdirs):
         # move to next folder
         os.chdir(subdirs[i])
         print os.getcwd()
         subdirs1 = (os.listdir('.'))
         print subdirs1
         # call function - open thedirectory and check for .c file
         open_dir('.')


 # definition of check for . c file 
 def check_for_c():
     cfile = fnmatch.filter(os.listdir('.'), '*.c')
     # if file found
     if len(cfile) != 0:
         # save the path
         path = os.path.dirname(os.path.abspath(cfile[0]))
         print path
         os.chdir('..')
         print os.getcwd()

     else:
         create_subdir_list()


 def open_dir(name):
     check_for_c() 
     open_dir('.')

1 个答案:

答案 0 :(得分:1)

最好用自己的方式练习python。但有效的方法是:

>>> PATH = '/home/sijan/Desktop/'
>>> import os
>>> from glob import glob
>>> result = [y for x in os.walk(PATH) for y in glob(os.path.join(x[0], '*.py'))]

它将列出给定路径中文件夹中的所有.py文件。