python学习1-列出指定目录下的指定类型文件

列出指定目录下的指定类型文件

import sys
import os
import getopt

 

def DisplayFileAndPath(directory, pfix, pfixset=False):
    try:
        dirlist = os.listdir(os.path.abspath(directory))
    except:
        print 'Access denied or No directory: %s' %directory
        sys.exit(2)
    else:
        for item in dirlist:
#            print os.path.isdir(item)
            new = directory+os.sep+item   #very important: isdir must use the absolute path...
            if os.path.isdir(new):               
                DisplayFileAndPath(new, pfix, pfixset)
            else:
                #postfix              
                if pfixset == True:
                    if os.path.splitext(item)[1][1:] == pfix:
                        print "%-30s\t\t%s" %(item, os.path.abspath(directory))
                else:
                    print "%-30s\t\t%s" %(item, os.path.abspath(directory))
 
def usage():
    print 'This program is to list all the files of same type in specified directory.\
           If no arguments, it will list all the files in current directory.'
   
    print '    -c    list all the files with specified postfix'
    print '    -d    list all the files in specified folder'

def main():
    path = os.path.dirname(sys.argv[0])
    pfixset = False
    postfix = []
 
    try:
        opts, args = getopt.getopt(sys.argv[1:], "c:d:")
    except:
        usage()
        sys.exit(2)

    for opt, arg in opts:
        if opt == "-c":
            postfix = arg
            pfixset = True
        elif opt == "-d":
            path = os.path.abspath(arg)
        else:
            assert False
   
    DisplayFileAndPath(path, postfix, pfixset)

if __name__=='__main__':
    '''
   
    -c    list all the files with specified postfix
    -d    list all the files in specified folder
    '''
    main()

 

刚看python，学习笔记，错误之处，希望不吝赐教。