CDays–3 通过函数进行功能化 Python基础教程图片来自 《可爱的Python》 P16 如果我们希望Python 的程序像CMD中的命令一样,可以添加运行参数,那么我们需要将程序做成函数, 就像C语言的函数。 这种问题《可爱的Python》是不会告诉我们的,让我们去参考书上翻一下。《Python 学习手册》 首先明白一个问题,函数是什么。 可以这么理解,在程序中,函数是一个通用的功能块,在程序的任何一个位置调用一个函数,他的功能是一样的。我们可以把一段总在程序中重复的代码定义成函数。 然后在之后的程序中调用。这样就可以省去了很多麻烦。 那么如果你学过任何一门语言,那么函数是很容易理解的。 那么根据参考书,我们看一下Python的函数是如何定义的。 def 语句 def语句将创建一个函数名,并且定义函数使用的变量。 函数的主体往往都含有return语句。 def <name> (arg1, arg2 ,..... argN ): ... return < value > 有一点值得说明,python 的def语句并不像C语言定义函数一样,def 可以在程序的任何一个位置,甚至在if 中,根据条件不同,配置不同的子函数。 我们给出一个def的使用例子。 def test(x ,y): return x*y print test(5,66) 在这里个例子中,我们定义函数test 的功能就是将两个变量相乘。 让我们尝试在命令行中调用它。 我想大家已经可以理解函数是怎么被调用的了。
那么现在我们看下我们的需求。
我们需要指定扫描目录 以及保存的文件名,让我们改一下昨天的代码。 import os export = "" for root, dirs, files in os.walk('/media/cdrom0'): export+="\n %s;%s;%s" % (root,dirs,files) open('mycd2.cdc', 'w').write(export) 下面我们加入def import os def cdWalker (cdrom , result_file ): export = "" for root, dirs, files in os.walk(cdrom): export+="\n %s;%s;%s" % (root,dirs,files) open(result_file, 'w').write(export) 我们运行一下,然后在命令行中调用。 我们得到了两个文件。 如此看来,我们定义的函数是没错的。 那又怎么从命令行获取输入的参数呢。 根据书上的指引,我们使用 sys.argv 让我们查一下sys.argv是干嘛的。 sys.argv 是用来获取命令行的参数的。 sys.argv 的用法 我们从《Python简明教程》上取得一个例子看一下他是怎么用的。 import sys def readfile(filename): ''' Print a file to the standard output. ''' f = file(filename) while True: line = f.readline() if len(line) == 0: break print line f.close() #Script starts from here if len(sys.argv) < 2: print ' NO action specified.' sys.exit() if sys.argv[1].startswith('--'): option = sys.argv[1][2:] if option == 'version': print ' version 1.2 ' elif option == 'help': print '''This program prints files to the standard output. Any number of files can be specified. Options include: --version : Prints the version number --help : Display this help''' else: print 'Unknow option.' else: for filename in sys.argv[1:]: readfile(filename) 这个程序要从cmd里面执行,让我们试一下。 功能好像挺简单的,那么我们现在分析一下这个程序。 首先,我们看到了def, 它定义了一个函数,这个函数从名称上看来是读取文件的,参数是文件名。 下面出现了两个IF , 第一个if是判断sys.argv的长度,那么我们有了一个新问题。sys.argv得到的是什么,让我们再上一个程序上面补写一句话 print sys.argv 我们看到了 sys.argv 的输出内容是一个list , 如果没有 参数,应该只有一项。 如果有参数,长度肯定超过1 . 第一个if 就是解决是否提供了参数的问题,如果没有参数,程序则退出,使用函数 sys.exit( ) 第二个if 是用来判断参数是什么。可以看到这里使用了偏移来索引字符串。 到这里,我们应该就很清楚了 sys.argv 的实质是什么了。
现在我们用 sys.argv 来修改最初的程序。
import os,sys CDROM = 'D:\\CDROM' def cdWalker (cdrom , result_file ): export = "" for root, dirs, files in os.walk(cdrom): export+="\n %s;%s;%s" % (root,dirs,files) open(result_file, 'w').write(export) if '-e'==sys.argv[1] : cdWalker (CDROM,sys.argv[2]) print "CDROM detals is saved in %s" % sys.argv[2] else: print 'ERROR INPUT !' 保存它,现在我们在CMD中运行试试看。 一个可以与用户互动的小程序暂时完成了。 大年初一啊。>< 绿色通道 |
|
来自: 昵称QAb6ICvc > 《python》