【Python】コマンドライン引数を取得する方法【→プログラム初心者向け】

プログラム女子

使う人が直接入力するような情報をプラグラムに入力するにはどうしたらいいのかな...

もし、入力するその時まで分からない情報や、ユーザーしか知らない情報をPythonプログラムに渡すには、「コマンドライン引数」というのが使えるよ。今回はこのコマンドライン引数の使い方を説明するね。

Tommy

Pythonでコマンドライン引数を取得する概要

この記事では、Pythonプログラムをコマンドラインから実行した場合に、コマンドライン引数の値をPythonファイル内部のプログラムに反映する方法を説明します。

Pythonファイルを実行する時に、同時に引数（パラメータ）を渡し、内部の処理で、渡されたパラメータを使うといった処理ができるようになります。これによって、Pythonで予めプログラムできないものを処理することができます。

例えば、下記のように、Pythonファイル「test.py」を実行する時、第１引数に「Tommy」という文字列を、第２引数に「hoge」という文字列を入力することができます。

$ python test.py Tommy hoge

或いは、Pythonファイルを実行するユーザーが複数人数居る場合に、ユーザーごとのユーザー名やパスワードは、ユーザーによって変わります。Pythonファイル実行時にどのユーザーがPythonファイルを実行しているかが分かるので、Pythonプログラム処理内部でユーザー名やパスワードを使った処理ができるようになります。

Pythonでコマンドライン引数を取得する前提環境

Pythonで引数を使った実行方法の前提の環境は、下の記事で説明している私の環境と同じ環境で確認しています。

レンタルサーバなどのLinuxサーバを使っている場合は、下の記事でプログラミングする環境の作り方を説明しています。

もし、まだプログラムする環境ができていない方は、先にこちらの記事を読んで、プログラミングできる環境をつくることをおすすめします。

Pythonでコマンドライン引数を取得するサンプルプログラム

サンプルプログラムの作成

ここでは、引数を取得するサンプルプログラムを作成します。手順としては、Pythonのプラグラムファイルを実行する時に引数を入力するように設計を前提として、その引数をプログラムのロジックに取り込めるような関数を定義します。そしてその関数の呼び出すようなMainの処理を実装します。詳細には次のように順番でサンプルプログラムを作成し、説明します。

1. ライブラリインポート
2. 引数入力を取得する関数の定義
3. 定義した関数の実行

1.ライブラリインポート

ここでは、Pythonファイル実行時に指定された引数の値を取得する為に、「sys」ライブラリをインポートしています。

import sys

2.引数入力を取得する関数の定義

今回のサンプルプログラムは、Pythonファイルの実行時にユーザーのユーザー名を第一引数に、パスワードを第二引数に指定されることを想定したつくりにします。

「sys」ライブラリの「argv」と呼ばれる関数で引数全体（ここでは変数「args」と定義している）を取得し、その引数全体はリストで渡される為、変数「args」の２番目の要素をユーザー名として取得、３番目の要素をパスワードとして取得します。また、変数「args」の１番目にはPythonのファイル名が入っています。

def get_input_param():

	user = ''
	passwd = ''
	result = False

	# 引数の取得
	args = sys.argv

	# 引数の個数が足りない場合
	if 2 >= len(args):
		result = False

	else:
		result = True
		user = args[1]
		passwd = args[2]

	return user,passwd,result

3.定義した関数の実行

ここでは、先程定義した関数の実行処理を説明します。まず、引数で指定された値をを取得する関数「get_input_param」を呼び出します。関数「get_input_param」の結果が値「False」だった場合は、入力が異常である旨のメッセージを表示し、入力が正常であれば、ユーザー名とパスワードを表示するというプログラムです。

if __name__ == '__main__':

	username, password, result = get_input_param()

	if False == result:
		print("Incorrect input.")

	else:
		print("User Name: " + username)
		print("User Pswd: " + password)

サンプルプログラムの実行

ここでは、上で作成したサンプルプログラム「test.py」を実行して、その結果を確認します。引数に設計した通りに入力した時の結果と、引数になにも指定せずにエラー処理結果を確かめる結果の２つの結果を示します。

1. サンプルプログラム実行の結果【期待通りの処理】
2. サンプルプログラム実行の結果【エラー処理】

1.サンプルプログラム実行の結果【期待通りの処理】

下の結果は、第一引数に文字列「Tommy」、第二引数に文字列「hoge」を指定してPythonファイル「test.py」を実行した際の結果を表しています。User Nameの項目に「Tommy」が、UserPswdの項目に「hoge」が表示されていて、２つの引数が意図通りに表示されていることが分かります。

$ python test.py Tommy hoge
User Name: Tommy
User Pswd: hoge

2.サンプルプログラム実行の結果【エラー処理】

下の結果は、引数なしでPythonファイル「test.py」を実行した際の結果を表しています。引数が２つ未満だとエラー処理（入力が間違っているメッセージを表示）にしているので、エラー処理されていることが分かります。

$ python test.py
Incorrect input.