python 命令行参数学习（二）

照着例子看看打打，码了就会。写了个命令行参数调用进行运算的脚本。

参考文章链接：http://www.jianshu.com/p/a50aead61319

#-*-coding:utf-8-*-
__author__ = 'Deen'

'''
import argparse
parser = argparse.ArgumentParser()
# 这个action的作用为，如果使用verbosity，则赋值为true，否则为false
parser.add_argument("--verbosity", help="increase output verbosity", action = "store_true")
args = parser.parse_args()
# 对这个程序法而言，只有两个参数有用，true，和false
if args.verbosity:
    print "verbosity turned on"
'''

'''
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-v', '--verbose', help='increase output verbosity', 
                    action='store_true')
args = parser.parse_args()
if args.verbose:
    print "Verbosity turned on"
'''

# 差不多了，自己写一个加减乘除的命令行参数脚本吧
import argparse


def plus(x, y):
    print("Plus is %d" % ((x + y)))


def minus(x, y):
    print("Minus is %s" % (x - y))


def mul(x, y):
    print("Mul is %s" % (x * y))


def divi(x, y):
    print("Divi is %s" % float(x / y))

if __name__ == '__main__':
    # 添加python字典映射，调用函数
    choices = {'plus': plus, 'minus': minus, 'mul': mul, 'divi': divi}

    # description对脚本进行说明
    parser = argparse.ArgumentParser(description='this a test')

    # 开始为脚本添加参数
    # 添加函数调用参数
    parser.add_argument('function', choices=choices,
                        help='To count x,y,input fucntion in [plus][minus][mul][divi]', default='plus')

    # 添加命令行参数位置
    parser.add_argument('x', help='number x', type=int)
    parser.add_argument('y', help='nubmer y', type=int)

    args = parser.parse_args()
    func = choices[args.function]

    # 调用参数
    func(args.x, args.y)

感觉差不多了，自己会用了