命令行参数

命令行参数

1.sys

​ sys模块主要用于获取和Python解释器相关的信息，可在Python交互式解释器中先导入sys模块 import sys.

通过内置函数dir()查询到模块内定义的所有名称，并以一个字符串列表形式返回

sys.argv               # 以列表的形式获取运行Python程序的命令行参数

import sys
def start(argv):
	print('程序名',argv[0])
	print('第一个参数',argv[1])
	print('第二个参数',argv[2])
	print('第三个参数',argv[3])

if __name__ == '__main__':
	try:
		start(sys.argv[0:])
	except KeyboardInterrupt:
		print("interrupted by user,killing all threads...")

2.getopt

​ getopt是对sys.argv获取的命令行参数进行二次处理。在程序运行时，可能需要根据不同的条件，输入不同的命令行选项来实现不同的功能。调用getopt.getopt()会返回两个列表，第一个列表赋值给opts,第二个列表赋值给args

# opts:  是(选项，选项值)的元组为一个元素的列表。如果没有选项值为空字符串
# args:  存放没用选项单参数的列表。不属于格式信息的剩余的命令行参数。

import sys,getopt

opts,args = getopt.getopt(sys.argv[1:],"u:p:",["file:"])
print("opts输出结果为",opts)
print("args输出结果为",args)

3.optparse

​ optparse模块主要用来为脚本传递命令参数，采用预先定义号的选项来解析命令，与getopt函数不同的是optparse可以自动生成帮助信息。

# optparse.OptionParse(usage).add_option()参数

import optparse

def banner():
    print('\033[1;34m##########################################################\033[0m\n'
          '\033[1;34m####################\033[1;32mlalalalala\033[1;34m###################\033[0m\n'
          '\033[1;34m#############################################################\033[0m\n')

def start():
	banner()
	usage = "python %prog -u/ --user <target user> -p/--password <targert password>"
	parser = optparse.OptionParser(usage)
	parser.add_option('-u','--user',dest='User',type='string',help='target user',default='root')
	parser.add_option('-p','--password',dest='Pwd',type='string',help='target password')
	option, args = parser.parse_args()
	
	print("用户名为", options.User)
	print("密码为",options.Pwd)

if __name__ == '__main__':
	try:
		start()
	except KeyboardInterrupt:
		print("interrupted by user, killing all threads...")

4.argparse

​ argparse是optparse模块的升级版，与optparse相比，argparse模块使用更加简单便捷

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-u', '--user',dest='User',type=str,default='root',help='target User')
parser.add_argument('-s', '--sex', dest='Sex', type=str, choices=['男','女'], default='男',help='target Sex' )
parser.add_argument('-n', '--number', dest='Num', nargs=2, required=True, type=int, help='target Two Numbers')
print(parser.parse_args())