生成器版本的文件MD5校验

生成器是一个可迭代的对象，可以对可迭代的对象进行便利，比如字符串、列表等，都是可迭代对象

 

　def f(n):

　　for i in range(n):

　　　　yield i

 

特点：

　　1、当调用这个函数的时候，函数内部的代码并不立马执行，这个函数只是返回一个生成器对象

　　2、当使用for进行迭代的时候，函数内的代码才会被执行

 

yield与return的区别：

　　1、return的时候这个函数的局部变量就都销毁了

　　2、所有return是得到所有结果之后的返回

　　3、yield是产生一个可以恢复的函数（生成器），恢复了局部变量

　　4、生成器只有在调用next()时才运行函数生成一个结果

　　5、return 不能出现在生成器中

 

yield版本的MD5：

 #!/usr/bin/env python

 import os
 import sys
 import hashlib

 def md5sum(f):
     m = hashlib.md5()
     with open(f) as fd:
         while True:
             data = fd.read(4096)
             if data:
                 m.update(data)
             else:
                 break
 return m.hexdigest()

 def file_md5(dirpath):
     pathobj = os.walk(dirpath)
     for roots,dirs,files in pathobj:
         for file in files:
             filename = os.path.join(roots,file)
             md5 = md5sum(filename)
             yield "%s %s" % ( md5,filename)

 def main():
     try:
         dirpath = sys.argv[1]
     except IndexError as e:
         sys.exit('need one dirctory')
     if os.path.isfile(dirpath):
         sys.exit('Please input a directory')
     for info in file_md5(dirpath):
         print info

 if __name__ == '__main__':
     main()