在Python3中添加带有标志的命令行参数

我必须从命令行输入参数,即用户名,密码和数据库名称.我知道如何在不使用标志的情况下使用’sys.argv'(如下所示).

##Test.py
hostname = str(sys.argv[1])
username = str(sys.argv[2])
password = str(sys.argv[3])

def ConnecttoDB():
    try:
        con=sql.connect(host=hostname, user= username, passwd= password)
        print ('\nConnected to Database\n')

# If we cannot connect to the database, send an error to the user and exit the program.
    except sql.Error:
        print ("Error %d: %s" % (sql.Error.args[0],sql.Error.args[1]))
        sys.exit(1)

    return con

因此,它可以运行为：

$test.py DATABASE USERNAME PASWORD

但问题是我必须使用’标志’.所以,脚本可以像这样运行：

$test.py -db DATABSE -u USERNAME -p PASSWORD -size 20

如何使用标志从命令行获取参数？我很感激你的帮助.

你应该使用的是 argparse

import argparse
parser = argparse.ArgumentParser()

#-db DATABSE -u USERNAME -p PASSWORD -size 20
parser.add_argument("-db", "--hostname", help="Database name")
parser.add_argument("-u", "--username", help="User name")
parser.add_argument("-p", "--password", help="Password")
parser.add_argument("-size", "--size", help="Size", type=int)

args = parser.parse_args()

print( "Hostname {} User {} Password {} size {} ".format(
        args.hostname,
        args.username,
        args.password,
        args.size
        ))