首页 > 代码库 > 逐步实现python版wc命令

逐步实现python版wc命令

  
Python 如何处理管道输入输出
  sys.stdin 等于打开了一个文件对象,所有输入的文件都会写入到标准输入文件中(键盘)
  sys.stdout 等于打来了一个文件对象,使用.write()把信息写入到标准输出文件中(屏幕)
 
  判断行数:
 1 #!/usr/bin/env python
 2 #_*_ coding:UTF-8 _*_
 3  
 4 import sys
 5  
 6 #遍历文件对象,并统计行数
 7 def lineCount(f):
 8     n = 0
 9     for i in f:
10       n += 1
11     return n
12  
13 input = sys.stdin
14 print(lineCount(input))
15  

 

    文件对象的方法:
  fd.read() 一次读取所有行,返回字符串
  fd.readline() 一次读取一行,返回一个字符串
  fd.readlines() 一次读取所有行,返回一个列表,每一行为一个元素
  f.write() 写入文件
  f.close() 关闭文件(每次打开文件,最好都要手动关闭文件)
   
  利用while循环遍历文件
1 while True:
2 ....: data =http://www.mamicode.com/ fd.readline()
3 ....: if not data:
4 ....: break
5 ....: print(data)
 
 
文件输出:
    sys.stdout.write() 文件写入到标准输出文件中去
    print和stdout的区别:
  1、print通常是调用一个stdout对象的write方法
  2、print会先进行格式转换
  3、print会在最后加上换行符
   stdout的buffer
   通过stdout输出的信息一般会先放在stdout的buffer中,然后等待输出完毕后,一次性输出
     这里可以通过两种方式禁止写入buffer
    1、sys.stdout.flush(),每次写入buffer,然后强制刷新到文件中
    2、python -u scripts.py 执行python解释器的时候,指明不使用buffer,python 2.x 适用
例子:
1 import sys
2 import time
3  
4 for i in range(10):
5     sys.stdout.write(>) #当然这里加上\n就会一个一个输出,因为sys.stdout是正行正行输出(加\n,就类似于print了)
6     sys.stdout.flush() #强制i刷新到stdout中去
7     time.sleep(1)

 

 
计算字符:
  获取字符数 len(data)
  获取单词数 len(data.split()) #以空格分隔,计算有几段
  获取行数 data.count(‘\n‘) #统计\n出现的次数即可
 
1 #!/usr/bin/env python
2 import sys
3 data =http://www.mamicode.com/ sys.stdin.read()
4 chars = len(data)
5 words = len(data.split())
6 lines = data.count(\n)
7 print(%s %s %s  % (lines,words,chars)) #传统的字符串替换
8 print(%(lines)s %(words)s %(chars)s % locals()) #高级用法,%(key)s,表示格式化关键字替换,后面就需要以字典的方式传入对应的key值,而locals(),表示当前环境下所有的变量和值的字典,所以这里可以进行替换
9 print(%(lines)s %(words)s %(chars)s % {lines:lines,words:words,chars:chars}) 这种方法和上面利用locals的方式是一样的,只不过locals的变量更多而已
    没有命令和参数版本的wc:
 1 #!/usr/bin/env python
 2 import sys,os
 3  
 4 if len(sys.argv) == 1:
 5     data =http://www.mamicode.com/ sys.stdin.read()
 6 else:
 7     try:
 8         filename = sys.argv[1]
 9     except IndexError as e:
10         sys.exit(The %s need one parameter % __file__)
11  
12 if os.path.exists(filename):
13     try:
14         fd = open(filename)
15         data =http://www.mamicode.com/ fd.read()
16         fd.close()
17     except IOError as e:
18         sys.exit(The Parameter is a file,not a Directory)
19 else:
20     sys.exit(The %s is not exist % filename)
21  
22 chars = len(data)
23 words = len(data.split())
24 lines = data.count(\n)
25  
26 print(%(lines)s %(words)s %(chars)s % locals())

 

Python的命令行参数
  利用optparse模块,来添加参数和选项
 
 1 #!/usr/bin/env python
 2 # Author:Lee Sir
 3  
 4 import sys
 5 from optparse import OptionParser #导入optparser模块中的OptionParser类
 6  
 7 parser = OptionParser() #实例化一个OptionParser类的对象parser,这里括号里可以添加一些提示信息,用户在执行help时输出(%prog表示脚本名称。例子:%prog [ -c| -l | -d] [file1])
 8 parser.add_option(-c,--char,dest=chars,action=store_true,default=False,help=only user to count chars)
 9  
10 #add_option 表示添加一个选项,-c为选项名称,--char为对应的长选项(可选),dest 表示在程序内引用该变量时的名称,action表示参数后面是否有值(有的话store,没有的话store_true/store_false),default表示该参数默认是添加还是不添加,help(执行-help会显示的内容)
11 parser.add_option(-w,--word,dest=words,action=store_true,default=False,help=only user to count words)
12 parser.add_option(-l,--line,dest=lines,action=store_true,default=False,help=only user to count lines)
13  
14 #parse_args() 会返回一个元组,第一个元素为对象,存储着参数的使用情况,第二个为列表,存储着参数对应的值。(注意,第一个元素为对象,呈现形式很像字典,但不能用字典的方式读取,只能使用option.dest来读写)
15 options,args = parser.parse_args()
16  
17 #默认参数,当同时没有-c,-l,-w时,设置这三个参数都是True
18 if not (options.chars or options.words or options.lines):
19 options.chars,options.words,options.lines = True,True,True
20  
21 data =http://www.mamicode.com/ sys.stdin.read()
22 chars = len(data)
23 words = len(data.split())
24 lines = data.count(\n)
25  
26 if options.chars: #脚本后添加了-c,则option.chars = True
27     print(chars,end=\t)
28 if options.words:
29     print(words,end=\t)
30 if options.lines:
31     print(lines)
32  

 

添加判断完善脚本:

  os.Path对文件路径的处理
  os.path.isdir 判断是否是目录
  os.path.isfile 判断是否是文件

 1 #!/usr/bin/env python
 2  
 3 import os,sys
 4 from optparse import OptionParser
 5 def opt():
 6     Get Command line parser
 7     parser = OptionParser()
 8     parser.add_option(-c,--char,dest=chars,action=store_true,default=False,help=used to count chars)
 9     parser.add_option(-w,--word,dest=words,action=store_true,default=False,help=used to count words)
10     parser.add_option(-l,--line,dest=lines,action=store_true,default=False,help=used to count lines)
11     option,args = parser.parse_args()
12     return option,args
13  
14 def get_count(data):
15     count for lines ,words or chars
16     chars = len(data)
17     words = len(data.split())
18     lines = data.count(\n)
19     return lines,words,chars
20  
21 def print_wc(option,lines,words,chars,filename):
22     print lines,words or chars
23     if option.lines:
24         print lines,
25     if option.words:
26         print words,
27     if option.chars:
28         print chars,
29     print filename
30  
31 def main():
32     main functions
33     option,args = opt()
34     if not (option.chars or option.words or option.lines):
35         option.chars , option.words, option.lines = True,True,True
36     if args:
37         total_lines,total_words,total_chars = 0, 0, 0
38         for filename in args:
39         if os.path.isfile(filename):
40             with open(filename) as fd:
41                 data =http://www.mamicode.com/ fd.read()
42                 lines,words,chars = get_count(data)
43                 print_wc(option,lines,words,chars,filename)
44                 total_lines += lines
45                 total_words += words
46                 total_chars += chars
47         elif os.path.isdir(filename):
48             print >> sys.stderr,%s is a directory % filename #利用print写入到文件中去,注意这里仅仅适用于Python 2.x,python3是不支持的(可以用print(i,file=sys.stdout) 或者sys.stdout.write())
49         else:
50             sys.exit(%s : No such file or Directory % filename)
51         if len(args) > 1:
52             print_wc(option,total_lines,total_words,total_chars,total)
53     else:
54         data =http://www.mamicode.com/ sys.stdin.read()
55         filename = ‘‘
56         lines,words,chars = get_count(data)
57         print_wc(option,lines,words,chars,filename)
58  
59  
60 if __name__ == __main__:
61     main()

 

逐步实现Python版本的wc命令
  添加-n参数,来禁止显示total
 1 #!/usr/bin/env python
 2 
 3 import os,sys
 4 from optparse import OptionParser
 5 def opt():
 6     Get Command line parser
 7     parser = OptionParser()
 8     parser.add_option(-c,--char,dest=chars,action=store_true,default=False,help=used to count chars)
 9     parser.add_option(-w,--word,dest=words,action=store_true,default=False,help=used to count words)
10     parser.add_option(-l,--line,dest=lines,action=store_true,default=False,help=used to count lines)
11     parser.add_option(-n,"--no-total",dest="nototal",action=store_true,default=False,help=not print total)
12     option,args = parser.parse_args()
13     return option,args
14 
15 def get_count(data):
16     count for lines ,words or chars
17     chars = len(data)
18     words = len(data.split())
19     lines = data.count(\n)
20     return lines,words,chars
21 
22 def print_wc(option,lines,words,chars,filename):
23     print lines,words or chars
24     if option.lines:
25         print lines,
26     if option.words:
27         print words,
28     if option.chars:
29         print chars,
30     print filename
31 
32 def main():
33     main functions
34     option,args = opt()
35     if not (option.chars or option.words or option.lines):
36         option.chars , option.words, option.lines = True,True,True
37     if args:
38         total_lines,total_words,total_chars = 0, 0, 0
39         for filename in args:
40             if os.path.isfile(filename):
41                 with open(filename) as fd:
42                     data =http://www.mamicode.com/ fd.read()
43                     lines,words,chars = get_count(data)
44                     print_wc(option,lines,words,chars,filename)
45                     total_lines += lines
46                     total_words += words
47                     total_chars += chars
48             elif os.path.isdir(filename):
49                 print >> sys.stderr,%s is a directory % filename
50             else:
51                 sys.exit(%s : No such file or Directory % filename)
52         if len(args) > 1:
53             if not option.nototal:
56 print_wc(option,total_lines,total_words,total_chars,total) 57 else: 58 data =http://www.mamicode.com/ sys.stdin.read() 59 filename = ‘‘ 60 lines,words,chars = get_count(data) 61 print_wc(option,lines,words,chars,filename) 62 63 64 if __name__ == __main__: 65 main()

 

逐步实现python版wc命令