合理主義的グルメブログ

学生起業家の日常をツラツラと書いています。主に食事情報です。

【Python】argparseで,選択肢を与える方法

python使っている人なら,コマンドライン引数の取得に,argparseを使うと思います.

僕も使っているので,その際に,
このオプションは,一定の選択肢の中から選んでほしいときに使えるオプションがあるので紹介します.

choices

--choicesオプションにより,引数には許される値を指定することができます.

import argparse

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--mode', help='set mode, [real] or [sim]', type=str, choices=['real', 'sim'])
args = parser.parse_args()

このようにすると,--modeオプションは,realsimをしていなければいけなくなります.