【Python】sqlite3モジュールでSQLiteデータベースにデータを保存・取得する方法

Pythonでは、標準ライブラリのsqlite3モジュールを使用することで、SQLiteデータベースを手軽に操作できます。以下のコードでは、sqlite3モジュールをインポートしてデータベース接続を確立し、テーブルを作成した後、データの挿入と取得を行い、最後に接続を閉じるまでの一連の流れを紹介します。

サンプルコード

#sqlitedemo.py
import sqlite3
from employee import employee

# データベースに接続
c = conn.cursor()

# テーブルを作成
c.execute('''CREATE TABLE employee(first text, last text, pay integer)''')

# Employeeオブジェクトを生成
emp_1 = employee('John', 'Doe', 50000)
emp_2 = employee('Jane', 'Doe', 60000)
emp_3 = employee('James', 'Dell', 80000)

# 名前付きプレースホルダーを使ってデータを挿入
c.execute('''INSERT INTO employee VALUES(:first, :last, :pay)''',
          {'first': emp_1.first, 'last': emp_1.last, 'pay': emp_1.pay})
c.execute('''INSERT INTO employee VALUES(:first, :last, :pay)''',
          {'first': emp_2.first, 'last': emp_2.last, 'pay': emp_2.pay})
c.execute('''INSERT INTO employee VALUES(:first, :last, :pay)''',
          {'first': emp_3.first, 'last': emp_3.last, 'pay': emp_3.pay})

# 条件に一致するデータを取得
c.execute("SELECT * FROM employee WHERE last='Doe'")
print(c.fetchone())
print(c.fetchmany(2))

# 変更を保存して接続を閉じる
conn.commit()
conn.close()

コードのポイント

• sqlite3.connect()：指定したファイル名のSQLiteデータベースに接続します。ファイルが存在しない場合は自動的に新規作成されます。
• cursor()：SQL文を実行するためのカーソルオブジェクトを取得します。
• CREATE TABLE：従業員の名（first）、姓（last）、給与（pay）を格納するテーブルを定義しています。
• 名前付きプレースホルダー：:first や :last のように値を直接SQL文に埋め込まず辞書で渡すことで、SQLインジェクションを防ぎながら安全にデータを挿入できます。
• fetchone() / fetchmany()：fetchone() は条件に一致した最初の1行を、fetchmany(n) は最大 n 行をタプルとして返します。
• commit()：INSERTなどの変更を確定してデータベースに保存します。呼び出し忘れるとデータが反映されないため注意が必要です。

実行結果

('John', 'Doe', 50000)
[('Jane', 'Doe', 60000), ('James', 'Dell', 80000)]

このように、sqlite3モジュールを使えば、テーブルの作成からデータの挿入・取得までをPythonのコードだけで完結できます。実際の開発では with conn: のようなコンテキストマネージャーを利用すると、例外発生時にも自動的にコミットやロールバックが行われるため、より安全にデータベースを扱えます。