python迭代列出某文件夹下所有文件

# -*- coding: UTF-8 -*- 
# 输出一个文件夹下所有文件
import os
import re
def filterFileName(filename):
if re.search('.doc\w{0,1}',filename,re.I)!=None:
return filename

def getalldocfilename(path):
    docfilenames = []
    for dirpath, dirnames, filenames in os.walk(path):
        filenames = filter(filterFileName, filenames)
        filenames = map(lambda filename: os.path.join(dirpath, filename), filenames)
        docfilenames.extend(filenames)
        return docfilenames
def main():
print 'begin'
path="C:/Users/sun/Desktop"
files =getalldocfilename(path)
print files

if __name__=='__main__':
main()