【每日一坑 4】 查找文件

Shen_SF

1. import os
   
2. from os.path import join
   
3. 
   
4. the_directory = raw_input('Please enter the directory\n')
   
5. base_directory = os.path.basename(the_directory)
   
6. os.chdir(os.path.dirname(the_directory))
   
7. all_files = []
   
8. for root, dirs, files in os.walk(base_directory):
   
9.     for name in files:
   
10.         all_files.append(join(root, name))
    
11. for i in all_files:
    
12.     if '.txt' in i:
    
13.         print i
    

复制代码