Psycopg2とPostgreSQL徹底解説: PythonからDB操作をマスター！

-- データベースの作成
CREATE DATABASE mydatabase;

-- ユーザーの作成とパスワード設定
CREATE USER myuser WITH PASSWORD 'mypassword';

-- データベースへの権限付与
GRANT ALL PRIVILEGES ON DATABASE mydatabase TO myuser;

pip install psycopg2-binary

import psycopg2

# データベース接続情報
DB_HOST = "localhost"
DB_NAME = "mydatabase"
DB_USER = "myuser"
DB_PASS = "mypassword"

conn = None # コネクションオブジェクトの初期化
cursor = None # カーソルオブジェクトの初期化

try:
    # 1. データベースへの接続
    conn = psycopg2.connect(host=DB_HOST, database=DB_NAME, user=DB_USER, password=DB_PASS)
    print("データベースに接続しました！")

    # 2. カーソルの作成
    cursor = conn.cursor()

    # 3. SQLクエリの実行（テーブル作成）
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS users (
            id SERIAL PRIMARY KEY,
            name VARCHAR(100) NOT NULL,
            email VARCHAR(100) UNIQUE
        );
    """)
    conn.commit() # 変更をコミット
    print("usersテーブルを作成または確認しました。")

    # 3. SQLクエリの実行（データ挿入）
    cursor.execute("INSERT INTO users (name, email) VALUES (%s, %s)", ("Alice", "alice@example.com"))
    cursor.execute("INSERT INTO users (name, email) VALUES (%s, %s)", ("Bob", "bob@example.com"))
    conn.commit()
    print("データを挿入しました。")

    # 3. SQLクエリの実行（データ取得）
    cursor.execute("SELECT id, name, email FROM users ORDER BY id DESC LIMIT 1;")
    user = cursor.fetchone() # 1件取得
    if user:
        print(f"取得データ: ID={user[0]}, Name={user[1]}, Email={user[2]}")

except psycopg2.Error as e:
    print(f"データベースエラー: {e}")
    if conn:
        conn.rollback() # エラー時はロールバック
finally:
    # 6. 接続のクローズ
    if cursor:
        cursor.close()
    if conn:
        conn.close()
    print("データベース接続を閉じました。")

import psycopg2

# データベース接続情報 (上記と同じ)

try:
    with psycopg2.connect(host=DB_HOST, database=DB_NAME, user=DB_USER, password=DB_PASS) as conn:
        with conn.cursor() as cursor:
            cursor.execute("SELECT version();")
            db_version = cursor.fetchone()
            print(f"PostgreSQLバージョン: {db_version[0]}")
except psycopg2.Error as e:
    print(f"エラー: {e}")

# 安全な方法: プレースホルダを使用
user_name = "Charlie"
user_email = "charlie@example.com"
cursor.execute("INSERT INTO users (name, email) VALUES (%s, %s)", (user_name, user_email))

# 危険な方法: 文字列結合（絶対に行わない）
# sql_query = f"INSERT INTO users (name, email) VALUES ('{user_name}', '{user_email}');"
# cursor.execute(sql_query)