首先感谢Mark提供的"argparse"代码.我已经找到了从由标志指定的命令行获取参数的方法.我正在粘贴下面的代码,其中使用了一个修改过的"标记"代码和一个小模块,该模块显示了如何使用通过命令行收集的参数:
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_参数行e.x.dest="username",这意味着从命令行收集的值将被分配给变量'username'
- 无论何时,只要在变量e.x con=sql之前添加"args",就可以使用命令行从"dest"分配的变量.连接(主机=args.hostname,用户=args.username,密码=args.password)