web-dev-qa-db-ja.com

入力引数argparse pythonの形式を指定します

コマンドライン入力を必要とするpythonスクリプトがあり、それらの解析にargparseを使用しています。ドキュメントが少し混乱していることがわかり、入力パラメーターの形式を確認する方法が見つかりませんでした。フォーマットをチェックすることの意味は、このサンプルスクリプトで説明されています。

parser.add_argument('-s', "--startdate", help="The Start Date - format YYYY-MM-DD ", required=True)
parser.add_argument('-e', "--enddate", help="The End Date format YYYY-MM-DD (Inclusive)", required=True)
parser.add_argument('-a', "--accountid", type=int, help='Account ID for the account for which data is required (Default: 570)')
parser.add_argument('-o', "--outputpath", help='Directory where output needs to be stored (Default: ' + os.path.dirname(os.path.abspath(__file__)))

オプション-sおよび-eをチェックする必要があります。ユーザーによる入力はYYYY-MM-DDという形式です。 argparseには、これを達成する方法がわからないオプションがありますか?.

90
Sohaib

ドキュメント ごと:

add_argument()typeキーワード引数により、必要な型チェックと型変換を実行できます... type=は、単一の文字列引数を取り、変換された値を返す呼び出し可能オブジェクトを受け取ることができます

次のようなことができます:

def valid_date(s):
    try:
        return datetime.strptime(s, "%Y-%m-%d")
    except ValueError:
        msg = "Not a valid date: '{0}'.".format(s)
        raise argparse.ArgumentTypeError(msg)

次に、それをtypeとして使用します。

parser.add_argument("-s", 
                    "--startdate", 
                    help="The Start Date - format YYYY-MM-DD", 
                    required=True, 
                    type=valid_date)
207
jonrsharpe

上記の答えに追加するだけで、ラムダ関数を1行に保持する場合はラムダ関数を使用できます。例えば:

parser.add_argument('--date', type=lambda d: datetime.strptime(d, '%Y%m%d'))

古いスレッドですが、質問は少なくとも私にとっては関連性がありました!

64
Evan V

検索エンジンを介してこれをヒットした他の人:Python 3.7では、ISO-8601準拠の日付のホイールを再発明する代わりに、標準の.fromisoformatクラスメソッドを使用できます。

parser.add_argument('-s', "--startdate",
    help="The Start Date - format YYYY-MM-DD",
    required=True,
    type=datetime.date.fromisoformat)
parser.add_argument('-e', "--enddate",
    help="The End Date format YYYY-MM-DD (Inclusive)",
    required=True,
    type=datetime.date.fromisoformat)
21
Michał Górny