python里的主函数 python中主函数( 二 )


sys.exit(2)# process options
for o, a in opts:if o in ("-h", "--help"):print __doc__
sys.exit(0)# process arguments
for arg in args:
process(arg) # process() is defined elsewhereif __name__ == "__main__":
main()1234567891011121314151617181920212223242526
Guido也承认之前自己写的main()函数也是类似的结构,但是这样写的灵活性还不够高,尤其是需要解析复杂的命令行选项时 。为此 , 他向大家提出了几点建议 。
添加可选的 argv 参数
首先,修改main()函数 , 使其接受一个可选参数 argv,支持在交互式shell中调用该函数:
def main(argv=None):
if argv is None:
argv = sys.argv# etc., replacing sys.argv with argv in the getopt() call.1234
这样做,我们就可以动态地提供 argv 的值,这比下面这样写更加的灵活:
def main(argv=sys.argv):
# etc.12
这是因为在调用函数时,sys.argv 的值可能会发生变化;可选参数的默认值都是在定义main()函数时,就已经计算好的 。
但是现在sys.exit()函数调用会产生问题:当main()函数调用sys.exit()时 , 交互式解释器就会推出!解决办法是让main()函数的返回值指示退出状态(exit status) 。因此 , 最后面的那行代码就变成了这样:
if __name__ == "__main__":
sys.exit(main())12
并且,main()函数中的sys.exit(n)调用全部变成return n 。
定义一个Usage()异常
另一个改进之处,就是定义一个Usage()异常,可以在main()函数最后的except子句捕捉该异常:
import sysimport getoptclass Usage(Exception):
def __init__(self, msg):
self.msg = msgdef main(argv=None):
if argv is None:
argv = sys.argvtry:try:
opts, args = getopt.getopt(argv[1:], "h", ["help"])except getopt.error, msg:raise Usage(msg)# more code, unchanged
except Usage, err:print sys.stderr, err.msgprint sys.stderr, "for help use --help"
return 2if __name__ == "__main__":
sys.exit(main())123456789101112131415161718192021222324
这样main()函数就只有一个退出点(exit)了,这比之前两个退出点的做法要好 。而且,参数解析重构起来也更容易:在辅助函数中引发Usage的问题不大,但是使用return 2却要求仔细处理返回值传递的问题 。
【python里的主函数 python中主函数】python里的主函数的介绍就聊到这里吧,感谢你花时间阅读本站内容,更多关于python中主函数、python里的主函数的信息别忘了在本站进行查找喔 。

推荐阅读