きっと続かんブログ

勉強したことや人に言いたいことを書く。

【python】オプションでパラメータをコマンドラインから受け取る【argparse】

目的

プログラム中のパラメータを、外部ファイル指定があればその内容を見て更新する。無ければデフォルト値を用いる。

import argparse

GLOBAL_PARAM = 1
GLOBAL_PARAM2 = 2

def update_params(param_path):
    # ファイルからパラメータを読み込み更新

if __name__ == "__main__":
    parser = argparse.ArgumentParser()

    parser.add_argument('--param', dest='param_path', help='option : assign parameter file', type=str)
    args = parser.parse_args()

    if args.param_path!=None:
        update_params(args.param_path)

オプション引数は無ければNoneを返すので、それによって分岐する。