19

コマンドラインからパラメーター、つまりユーザー名、パスワード、データベース名を入力する必要があります。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

フラグを使用してコマンド ラインから引数を取得するにはどうすればよいですか?

4

3 に答える 3

73

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
        ))
于 2012-07-22T23:18:41.040 に答える
11

まず、「argparse」コードの Mark に感謝します。フラグで指定されたコマンドラインから引数を取得する方法を見つけました。変更された「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

私の懸念の 1 つは、引数からコードへの変数の使用方法でした。そのためには、2 つのことを行う必要があります

  1. add_argument 行 ex dest ="username" に「dest」を追加します。これは、コマンド ラインから収集された値が変数「username」に割り当てられることを意味します。
  2. 「dest」から変数を割り当てたコマンドラインを使用するときはいつでも、変数の前に「args」を追加するだけです。 ex con=sql.connect(host=args.hostname, user= args.username, passwd= args.password)
于 2012-07-23T18:52:40.427 に答える