在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 
     ))