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 DATABASE USERNAME PASWORD

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

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

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

Answer 1

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

您应该使用的是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
        ))

Answer 2

首先感谢Mark为'argparse'代码。我已经找到了从flags指定的命令行获取参数的方法。我粘贴下面的代码，该代码使用修改后的“Mark”代码和一个小模块，该模块显示如何使用通过命令行收集的参数：

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

我关心的一个问题是如何使用参数中的变量来代码。为此，需要做两件事

将'dest'添加到add_argument行e.x. dest =“username”表示从comamnd行收集的值将被分配给变量'username'
每当你使用这样的命令行从'dest'分配变量时，只需在变量ex con = sql.connect（host = args.hostname，user = args.username，passwd = args.password）之前添加'args'

完成

由于

AK

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

2 个答案: