生成器版本的文件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()