首页 > 代码库 > 逐步实现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命令
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。