Python: SQLite3のSELECT文で値を動的に指定する

SQLite3を使用してPythonでデータを検索する方法。

テーブル名はusers、カラムは(id, name, age)。 指定した「id」のデータを検索します。

import sqlite3

def search_user_by_id(user_id):
	# データベースファイル名（実際のファイル名に置き換えてください）
	db_file = 'your_database.db'

	try:
		# with文を使用してデータベースに接続
		with sqlite3.connect(db_file) as conn:
			# カーソルを取得
			cursor = conn.cursor()

			# SQLクエリを準備（パラメータ化クエリを使用）
			query = "SELECT * FROM users WHERE id = ?"
			
			# クエリを実行
			cursor.execute(query, (user_id,))
			
			# 結果を取得
			result = cursor.fetchone()
			
			if result:
				# 結果を表示
				print(f"ID: {result}")
				print(f"Name: {result}")
				print(f"Age: {result}")
			else:
				print(f"User with ID {user_id} not found.")

	except sqlite3.Error as e:
		print(f"An error occurred: {e}")

# 使用例
user_id = input("検索するユーザーのIDを入力してください: ")
search_user_by_id(user_id)

このコードの説明：

  1. sqlite3モジュールをインポートします。
  2. search_user_by_id関数を定義します。この関数は指定されたIDに基づいてユーザーを検索します。
  3. with sqlite3.connect(db_file) as conn:を使用してデータベースに接続します。with文を使用することで、ブロックの終了時に自動的に接続がクローズされます。
  4. カーソルを取得し、SQLクエリを準備します。パラメータ化クエリを使用してSQLインジェクション攻撃を防ぎます。
  5. cursor.execute()を使用してクエリを実行し、指定されたIDのユーザーを検索します。
  6. fetchone()メソッドを使用して結果を取得します。
  7. 結果が存在する場合、ユーザーの情報を表示します。存在しない場合は、その旨を表示します。
  8. エラーハンドリングを行い、例外が発生した場合にエラーメッセージを表示します。
  9. 最後に、ユーザーに検索したいIDの入力を求め、その入力値を使用してsearch_user_by_id関数を呼び出します。

この方法では、with文を使用することで、データベース接続の管理が簡潔になり、接続のクローズを忘れるリスクが軽減されます。また、コードの可読性も向上します。

Python