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

Question

我必须从命令行输入参数，例如用户名，密码和数据库名称。我知道如何做到这一点，而不使用标志，通过使用'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数据库用户名PASWORD

但问题是，我必须使用 '标志'。所以，脚本可以像这样运行：

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

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

Answer 1

第一。我已经想出了从标志指定的命令行参数的方法。我粘贴低于使用修改“标记”的代码和一个小模块，展示了如何使用通过命令行收集到的参数代码：

import argparse 

parser = argparse.ArgumentParser() 

#-db DATABASE -u USERNAME -p PASSWORD -size 20000 
parser.add_argument("-host", "--hostname", dest = "hostname", default = "xyz.edu", help="Server name") 
parser.add_argument("-db", "--database", dest = "db", default = "ding_dong", help="Database name") 
parser.add_argument("-u", "--username",dest ="username", help="User name") 
parser.add_argument("-p", "--password",dest = "password", help="Password") 
parser.add_argument("-size", "--binsize",dest = "binsize", help="Size", type=int) 

args = parser.parse_args() 

print("Hostname {} db {} User {} Password {} size {} ".format(
     args.hostname, 
     args.db, 
     args.username, 
     args.password, 
     args.binsize 
     )) 
def ConnectToDB(): 
    print ('Trying to connect to mySQL server') 
    # Try to connect to the database 
    try: 
     con=sql.connect(host=args.hostname, user= args.username, passwd= args.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 

我的一个关注的是如何使用的参数变量的代码。对于这两件事情需要完成

1. 将'dest'添加到add_argument行e.x. dest =“username”，这意味着从comamnd行收集的值将被分配给变量'username'
2. 每当您使用命令行分配的变量来自'dest'时，只需在变量前添加'args'ex con = sql.connect （主机= args.hostname，用户= args.username，passwd文件= args.password）

完成

感谢

AK

Answer 2

python 3库包含3个模块用于解析命令行，因此无需额外添加到您的设置中。

，你应该使用一个一切都归功于标记为“argparse”代码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 
     ))