Flask入門｜SQLiteで基本的なデータ操作を学ぼう【チャプター5-03】

import os      # ファイルやディレクトリ操作で使用する標準ライブラリ「os」をインポート
import sqlite3 # 組み込み型データベース「SQLite」を操作するためのsqlite3モジュールをインポート

# ====================================
# データベースのパス（保存場所）を設定
# ====================================
base_dir = os.path.dirname(__file__)             # このスクリプトファイルがあるフォルダのパスを取得して、変数base_dirに代入
database = os.path.join(base_dir, 'data.sqlite') # base_dir内のdata.sqliteというファイルへのパスを作成し、変数databaseに代入

# ====================================
# データベース接続とカーソルの初期化
# ====================================
## 接続オブジェクト
conn = sqlite3.connect(database) # databaseパスにSQLiteのデータベースを作成or接続し、接続オブジェクトconnに代入
print('▼▼▼ 接続開始 ▼▼▼')    # データベース接続開始を示すログ出力
print()                          # 空行を出力してログの見やすさを確保

## カーソルオブジェクト
cur = conn.cursor() # SQL文を実行するためのカーソルオブジェクトを作成（インスタンス生成）し、変数curに代入

# ====================================
# SQL操作の実行
# ====================================
## テーブルを初期化
drop_sql = """
    DROP TABLE IF ExISTS items; 
""" # items というテーブル（表）が存在する場合は削除するSQL文を、変数drop_sqlに代入
cur.execute(drop_sql) # .executeメソッドでdrop_sqlを実行 ⇒ 対象テーブルを削除
print('対象テーブルがあれば削除')

## テーブルを作成
create_sql = """
    CREATE TABLE items(
        item_id INTEGER PRIMARY KEY AUTOINCREMENT,
        item_name STRING UNIQUE NOT NULL,
        age INTEGER NOT NULL
    );
""" # items というテーブルを作るSQL文を、変数create_sqlに代入
cur.execute(create_sql) # .executeメソッドでcreate_sqlを実行 ⇒ テーブルを作成
print('テーブル作成')

# ★★★★★★★★★★★★★★★★★★★★★
# ここにChapter5-3のコードを追記していこう
# ★★★★★★★★★★★★★★★★★★★★★

# ====================================
# データベースとの接続を閉じる
# ====================================
conn.close() # 処理が終わったら明示的に接続を閉じる
print()
print('▲▲▲ 接続終了 ▲▲▲')

## データを追加
insert_sql = """
    INSERT INTO items(item_name, age)VALUES(?, ?)
""" # 「itemsテーブルにデータを追加する」というSQL文を、変数insert_sqlに代入
    # (?, ?) はプレースホルダー：後で実データを代入する
insert_data_list = [
    ('私',20),('父',50),('母',45)
] # 登録するタプルデータを変数insert_data_listに代入
cur.executemany(insert_sql, insert_data_list) # .executemanyメソッドで各タプルを insert_sql の ?, ? に順番に代入
conn.commit() # 登録内容を確定（コミット）
print('データ追加：実行')

## 全てのデータを参照
select_all_sql = """
    SELECT * FROM items
""" # 全データを取得するSQL文を変数select_all_sqlに代入
cur.execute(select_all_sql) # .executeメソッドでselect_all_sqlを実行 ⇒ 全データ取得
print('--- 全件取得：実行 ---')
data = cur.fetchall() # .fetchall()メソッドで取得データを受け取り、変数dataに代入
print(data)

## 1件のデータを参照
select_one_sql = """
    SELECT * FROM items WHERE item_id = ?
""" # item_id が一致する1件だけを取得するSQL文を、変数select_one_sqlに代入
    # ? はプレースホルダー：後で実データを代入する
id = 3
cur.execute(select_one_sql, (id,)) # .executeメソッドでselect_one_sqlを実行 ⇒ データ取得
print('--- 1件取得：実行 ---')
data = cur.fetchone() # .fetchone()メソッドで取得データの最初の1行を受け取り、変数dataに代入
print(data)

## データを更新
update_sql = """
    UPDATE items SET age=? WHERE item_id= ?
""" # item_idが指定する行のageを変更するSQL文を、変数update_sqlに代入
    # ? はプレースホルダー：後で実データを代入する
age = 21
id = 1
cur.execute(update_sql, (age, id)) # .executeメソッドでupdate_sqlを実行 ⇒ データ更新
print('--- データ更新：実行 ---')
conn.commit()                      # 更新内容を確定（コミット）
cur.execute(select_one_sql, (id,)) # 以下、「1件のデータを参照」部分と同じ
data = cur.fetchone()
print('確認のため1件取得：実行',data)

## データを削除
delete_sql = """
    DELETE FROM items WHERE item_id= ?
""" # item_id が一致するレコードだけを削除するSQL文を、変数delete_sqlに代入
    # ? はプレースホルダー：後で実データを代入する
id =3
cur.execute(delete_sql, (id,)) # .executeメソッドでdelete_sqlを実行 ⇒ データ削除
conn.commit()                  # 更新内容を確定（コミット）
print('--- データ削除：実行 ---')
cur.execute(select_all_sql)    # 以下、「全てのデータを参照」部分と同じ
data = cur.fetchall()
print('確認のため全件取得：実行', data)