python版WC

#!/usr/bin/env python
#coding:utf8
from optparse import OptionParser
import sys,os
def opt():
    parser = OptionParser()
    parser.add_option("-c","--char",
                      dest="chars",
                      action="store_true",
                      default=False,
                      help="only count chars")
    
    parser.add_option("-w","--word",
                      dest="words",
                      action="store_true",
                      default=False,
                      help="only count words")
    
    parser.add_option("-l","--line",
                      dest="lines",
                      action="store_true",
                      default=False,
                      help="only count lines") 
    parser.add_option("-n","--no-total",
                      dest="nototal",
                      action="store_true",
                      default=False,
                      help="no total")
    options,args = parser.parse_args() 
    return options,args
‘‘‘参数模块‘‘‘
def get_count(data):
    chars = len(data)
    words = len(data.split())
    lines = data.count(‘\n‘)
    return lines,words,chars
‘‘‘统计模块‘‘‘    
def print_wc(options,lines,words,chars,fn):
    if options.lines:
        print lines,
    if options.words:
        print words,
    if options.chars:
        print chars,
    print fn
‘‘‘判断参数‘‘‘
def main():
    options,args = opt()
    if not (options.lines or options.words or options.chars):
        options.lines , options.words , options.chars = True,True,True
    ‘‘‘判断有没有参数，若没加参数，则输出所有参数‘‘‘
    if args:
        total_lines,total_words,total_chars = 0,0,0
        for fn in args:
            if os.path.isfile(fn):
                with open(fn) as fd:
                    data = fd.read()
                lines,words,chars = get_count(data)
                print_wc(options,lines,words,chars,fn)
                total_lines += lines
                total_words += words
                total_chars += chars
            elif os.path.isdir(fn):
                print >> sys.stderr,"%s: is a directory" %fn
            else:
                sys.stderr.write("%s: No such file or direcotry\n" %fn)
        if len(args) > 1 and not options.nototal:
            print_wc(options,total_lines,total_words,total_chars,‘total‘)
    else:
        data = sys.stdin.read()
        fn = " "
        lines,words,chars = get_count(data)
        print_wc(options,lines,words,chars,fn)
if __name__ == ‘__main__‘:
    main()