PythonのタプルをPostgreSQLデータベースに挿入する方法

PostgreSQLはデフォルトでポート番号5432を使用して動作します。PythonからPostgreSQLを操作するには、psycopg2モジュールをインストールします。この記事では、「test」データベースと、fname（名）、sname（姓）、age（年齢）、gender（性別）、salary（給与）というフィールドを持つ「employee」テーブルが既に用意されているものとして解説を進めます。

1. データベースへの接続とカーソルオブジェクトの取得

まず、Pythonスクリプト内で以下のコードを実行し、データベースへの接続を確立してカーソルオブジェクトを取得します。

import psycopg2
conn = psycopg2.connect(database="test", user="postgres", password="pass123", host="localhost", port="5432")
cur = conn.cursor()

2. 挿入するデータをタプルとして準備する

employeeテーブルに挿入したいデータは、タプルオブジェクトとして保持しておきます。

t1 = ('Mac', 'Mohan', 20, 'M', 2000)

3. INSERTクエリを作成する

次に、このタプルを使ってINSERT文となるSQLクエリを組み立てます。

sql = "insert into employee values(%s,%s,%d,%s,%d)" % t1

4. クエリを実行する

組み立てたクエリを、カーソルオブジェクトを使って実行します。

cur.execute(sql)
conn.commit()

このコードを実行すると、employeeテーブルに新しいレコードが追加されます。conn.commit()を呼び出して変更を確定することを忘れないようにしましょう。コミットしない場合、変更はデータベースに反映されません。

より安全な方法：パラメータ化クエリの活用

上記のような文字列フォーマットによるSQL構築は、SQLインジェクションのリスクがあるため、実際の開発では推奨されません。代わりに、プレースホルダを使ったパラメータ化クエリを利用するのが安全です。

sql = "insert into employee values(%s,%s,%s,%s,%s)"
cur.execute(sql, t1)
conn.commit()

この方法では、psycopg2が自動的に値を適切にエスケープしてくれるため、セキュリティ面でも安心です。