python 练习 : 帮我把硬盘里的图片都找出来

huangfen2002

要点:使用系统模块,遍历磁盘,遍历文件,找到符合条件的文件记录下来到一个Log文件中(XML格式).
python 代码
 

• import win32api  
  
• import os  
  
•   
  
• #constants and configs  
  
• IGNORE_PATH = [  
  
•     "C:\\WINDOWS",  
  
•     "C:\\Program Files",  
  
•     "C:\\Documents and Settings",  
  
•     "C:\\System Volume Information"  
  
•     ]  
  
•   
  
• #TODO use re replace  
  
• IMAGE_TYPE = [  
  
•     "jpg","gif","png","jpeg","bmp"  
  
•     ]  
  
• #find out the logic disks  
  
•   
  
• disks = win32api.GetLogicalDriveStrings().split("\x00")  
  
• for disk in disks:  
  
•     if disk in ['','A:\\']:  
  
•         disks.remove(disk)  
  
•          
  
• #TODO store to xml file  
  
• def store(filePath,fileName):  
  
•     print 'about to store %s\\%s' % (filePath,fileName)  
  
•   
  
• def scan(path):  
  
•     print 'scanding path : ' + path  
  
•     if not path.endswith('\\'):  
  
•         path = path + '\\'  
  
•     for _file in os.listdir(path):  
  
•         if os.path.isdir(path + _file) and path + _file not in IGNORE_PATH:  
  
•             scan(path + _file)  
  
•         else:  
  
•             for image_type in IMAGE_TYPE:  
  
•                 if _file.endswith(".%s" % image_type):  
  
•                     store(path,_file)  
  
•                     break  
  
•      
  
• # now iterat the disks  
  
• for disk in disks:  
  
•     print 'now scan disk : %s ' % disk  
  
•     scan(disk)  
  

到此为止,Python把硬盘里的图片都找出来了,还差把图片信息存储到XML文件里面去.
TODOs:
一,使用正则表达式匹配后缀名
二,保存信息到XML