python 函数 argparse 使用 - 针对命令行参数调用 [备忘]

python 具有 argarse 模块， 专门针对命令行下的参数传递

参见执行例子:

[root@gd02-zabbix-db-research python_api]# ./app.py
usage: ./app.py --help

[root@gd02-zabbix-db-research python_api]# ./app.py --help
usage: app.py [-h] [-t] [-u] [-s SCHEDULER] [-p PERSISTENT] [-r REALSERVER]
              [-f FLOAPINGIP] [-g] [-m] [-S SRCPORT] [-R DESTPORT]

optional arguments:
  -h, --help            show this help message and exit
  -t, --tcp             tcp service
  -u, --udp             udp service
  -s SCHEDULER, --scheduler SCHEDULER
                        one of rr|wrr|lc|wlc|lblc|lblcr|dh|sh|sed|nq,the
                        default scheduler is wlc
  -p PERSISTENT, --persistent PERSISTENT
                        persistent service, default:1500
  -r REALSERVER, --realserver REALSERVER
                        server-address is host (and port) Example: -r '1.1.1.1
                        2.2.2.2 3.3.3.3'
  -f FLOAPINGIP, --floapingip FLOAPINGIP
                        vip address of lvs
  -g, --gatewaying      gatewaying (direct routing)
  -m, --masquerading    masquerading (NAT)
  -S SRCPORT, --srcport SRCPORT
                        listen on floapip port
  -R DESTPORT, --destport DESTPORT
                        listen on realserver port

#!/apps/svr/python/python
#coding:utf-8
# Terry zeng <terry.zeng@xxx.com>
#
#  2014 04 10
#
#  lvs 调用接口


import argparse

#### 参数调用说明
parser = argparse.ArgumentParser()
parser.add_argument('-t','--tcp', help='tcp service',action='store_true')
parser.add_argument('-u', '--udp', help='udp service', action='store_true')
parser.add_argument('-s', '--scheduler', help='one of rr|wrr|lc|wlc|lblc|lblcr|dh|sh|sed|nq,the default scheduler is wlc')
parser.add_argument('-p', '--persistent', help='persistent service, default:1500', type = int)
parser.add_argument('-r', '--realserver', help="server-address is host (and port) Example: -r '1.1.1.1 2.2.2.2 3.3.3.3'", type = str )
parser.add_argument('-f', '--floapingip', help='vip address of lvs' )
parser.add_argument('-g', '--gatewaying', help='gatewaying (direct routing)',action='store_true')
parser.add_argument('-m','--masquerading', help='masquerading (NAT)', action='store_true')
parser.add_argument('-S','--srcport', help='listen on floapip port', type = int )
parser.add_argument('-R','--destport', help='listen on realserver port', type = int )
args = parser.parse_args()

例子说明：

parser = argparse.ArgumentParser()   [打开解释器]
parser.add_argument('-t','--tcp', help='tcp service',action='store_true')

parser.add_argument 用于添加可用参数， 所有需要调用的参数都需要利用该函数进行添加

'-t', '--tcp' 参数长短语法
help = 'xxxxx' 属于该参数的语法说明 
action='store_true' 表明该参数不接受参数传递， 即， 只接受 -t 或 --tcp 下面是错误例子 -t tcp  或 --tcp=yes

假如需要接受参数， 如 -t tcp 或 --tcp=yes 那么不可以使用 action 进行定义

假如需定义参数传入格式， 可使用 type = int|str|complax 进行定义

参数处理技巧 1

参数穿入后如何调用？参见上文例子

args.srcport  
假如执行过程中， 调用参数 -S 80 或 --srcport=80  传递
即: args.srcport = 80

参数处理技巧 2

假如上述参数中， 可能只接受一种， 如 -t -u 只能够接受一个而且必须接受一个

if args.tcp == args.udp:
  print 'usage: ' + __file__ + ' --help'
  sys.exit(2)

上述判断两个参数要么都存在， 要么都不存在， 即可满足要求

参数处理技巧 3

假如需要传递多个参数值,  多个值 切记使用 " " 定义

执行命令时， 效果如下

./app.py --realserver="1.1.1.1         2.2.2.2        3.3.3.3       4.4.4.4"

参数处理技巧 4

参数传递时， 可能会出现多个空格， 可用下面方法进行避免

执行时可能存在多个空格
./app.py --realserver="1.1.1.1         2.2.2.2        3.3.3.3       4.4.4.4"


参见语法
realip =  re.sub(r'\s+', ' ',args.realserver)
realip 格式变成 1.1.1.1 2.2.2.2 3.3.3.3 4.4.4.4 只存在一个空格分隔符

参数处理技巧 5

参数类型转换, 如上文中需要把值1.1.1.1    2.2.2.2   3.3.3.3   4.4.4.4 转换成为数组进行处理

realserver = args.realserver.split(' ')

转换已经成功
print len(realserver)  参见数组个数