CDays–3 通过函数进行功能化 Python基础教程

friendlessstar

　　图片来自 《可爱的Python》 P16
　　如果我们希望Python 的程序像CMD中的命令一样，可以添加运行参数，那么我们需要将程序做成函数， 就像C语言的函数。
　　这种问题《可爱的Python》是不会告诉我们的，让我们去参考书上翻一下。《Python 学习手册》
　　首先明白一个问题，函数是什么。
　　可以这么理解，在程序中，函数是一个通用的功能块，在程序的任何一个位置调用一个函数，他的功能是一样的。我们可以把一段总在程序中重复的代码定义成函数。
　　然后在之后的程序中调用。这样就可以省去了很多麻烦。
　　那么如果你学过任何一门语言，那么函数是很容易理解的。
　　那么根据参考书，我们看一下Python的函数是如何定义的。
　　def 语句
　　def语句将创建一个函数名，并且定义函数使用的变量。
　　函数的主体往往都含有return语句。
  
def  (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 = &quot;&quot;
for root, dirs, files in os.walk('/media/cdrom0'):
export+=&quot;\n %s;%s;%s&quot; % (root,dirs,files)
open('mycd2.cdc', 'w').write(export)
　　下面我们加入def



import os
def cdWalker (cdrom , result_file ):
export = &quot;&quot;
for root, dirs, files in os.walk(cdrom):
export+=&quot;\n %s;%s;%s&quot; % (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 = &quot;&quot;
for root, dirs, files in os.walk(cdrom):
export+=&quot;\n %s;%s;%s&quot; % (root,dirs,files)
open(result_file, 'w').write(export)
if '-e'==sys.argv[1] :
cdWalker (CDROM,sys.argv[2])
print &quot;CDROM detals is saved in  %s&quot; % sys.argv[2]
else:
print 'ERROR INPUT !'
　　保存它，现在我们在CMD中运行试试看。

　　一个可以与用户互动的小程序暂时完成了。
　　大年初一啊。><