[已解决]Python练习题，请问如何深入子文件夹？找到特定格式的文件？

mchl.gao

各位好。

我初学Python，在做一个Python练习题,要求首先定位桌面文件夹位置，然后再把桌面文件夹（包括子文件夹在内）的，符合所有指定格式的文件显示出来，指定的文件格式是TXT、DOC、PDF文件。

我写了一段，代码如下，现在问题是，
1，现在能显示出桌面TXT格式文件，但是无法深入桌面的子文件夹。
2，只能找到txt文件，glob多条件选择一直不能成功。无法找到DOC和PDF格式的文件。

import winreg
def get_desktop():
    key = winreg.OpenKey(winreg.HKEY_CURRENT_USER,r'Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders')
    return winreg.QueryValueEx(key, "Desktop")[0]

if __name__=='__main__':
    Desktop_path=str(get_desktop())
   
import glob
import os
os.chdir(Desktop_path)
for file in glob.glob('*.txt'):
    print (file)

请各位多多赐教，谢谢

最佳答案

月排行榜 / 总排行榜

jackz007

2019-4-20 17:21:47

1. #!/bin/python
   
2. #coding:gbk
   
3. 
   
4. import os , os . path
   
5. import winreg
   
6. 
   
7. def get_desktop():
   
8.     key = winreg . OpenKey(winreg . HKEY_CURRENT_USER , r'Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders')
   
9.     return winreg . QueryValueEx(key , "Desktop")[0]
   
10. 
    
11. def search(pwd , ext):
    
12.     try:
    
13.         for e in os . listdir(pwd):
    
14.             x = os . path . join(pwd , e)
    
15.             if os . path . isfile(x):
    
16.                 if os . path . splitext(e)[1] . lower() in ext:
    
17.                     print x
    
18.             elif os . path . isdir(x):
    
19.                 search(x , ext)
    
20.     except:
    
21.         pass
    
22. 
    
23. if __name__=='__main__':
    
24.     search(get_desktop() , (".txt" , ".pdf" , ".doc"))
    

复制代码

跳转到最佳答案楼层

大裤衩子

层层递进，判断文件后缀名啊

jackz007

1. #!/bin/python
   
2. #coding:gbk
   
3. 
   
4. import os , os . path
   
5. import winreg
   
6. 
   
7. def get_desktop():
   
8.     key = winreg . OpenKey(winreg . HKEY_CURRENT_USER , r'Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders')
   
9.     return winreg . QueryValueEx(key , "Desktop")[0]
   
10. 
    
11. def search(pwd , ext):
    
12.     try:
    
13.         for e in os . listdir(pwd):
    
14.             x = os . path . join(pwd , e)
    
15.             if os . path . isfile(x):
    
16.                 if os . path . splitext(e)[1] . lower() in ext:
    
17.                     print x
    
18.             elif os . path . isdir(x):
    
19.                 search(x , ext)
    
20.     except:
    
21.         pass
    
22. 
    
23. if __name__=='__main__':
    
24.     search(get_desktop() , (".txt" , ".pdf" , ".doc"))
    

复制代码

mchl.gao

jackz007 发表于 2019-4-20 17:21

非常感谢