PythonでPDFセキュリティを強化：PyPDF2でパスワード設定・解除（暗号化・復号）

pip install PyPDF2

from PyPDF2 import PdfWriter, PdfReader
import os

# --- テスト用のPDFファイルを仮作成 (中身は空でも可) ---
# この部分はPyPDF2では直接できませんが、説明のために存在すると仮定
# 例: 'original.pdf' という空のPDFファイルが事前に用意されているとする
# または、簡単なPDFを作成する他の方法 (例: reportlab) を使う

# 実際に処理するPDFファイル名
original_pdf_file = "original.pdf"
encrypted_pdf_file = "encrypted_document.pdf"

# テスト用に簡単なPDFを準備 (既存のPDFがない場合)
# 通常は手動で作成したPDFや、他のプログラムで生成したPDFを使います
# ここではダミーとして空のPDFを作成する（もしファイルが存在しない場合のみ）
if not os.path.exists(original_pdf_file):
    temp_writer = PdfWriter()
    temp_writer.add_blank_page(500, 700) # 空白ページを追加
    with open(original_pdf_file, "wb") as f:
        temp_writer.write(f)
    print(f"テスト用に '{original_pdf_file}' を作成しました。\n")
# ----------------------------------------------------


try:
    # 1. 元のPDFを読み込む
    reader = PdfReader(original_pdf_file)
    
    # 2. 新しいPdfWriterオブジェクトを作成
    writer = PdfWriter()

    # 3. 元のPDFの全てのページを新しいWriterにコピー
    for page in reader.pages:
        writer.add_page(page)

    # 4. パスワードを設定して暗号化
    # ユーザーパスワードとオーナーパスワードを設定できる
    # ここではユーザーパスワードのみを設定
    password = "mysecretpassword"
    writer.encrypt(password) # ユーザーパスワードを設定

    # オーナーパスワードも設定する場合 (閲覧制限と別に編集制限をかけたい場合など)
    # writer.encrypt(user_pwd=password, owner_pwd="ownerpassword")

    # 5. パスワード付きPDFとして保存
    with open(encrypted_pdf_file, "wb") as f:
        writer.write(f)

    print(f"'{original_pdf_file}' をパスワード '{password}' で暗号化し、'{encrypted_pdf_file}' として保存しました。")

except FileNotFoundError:
    print(f"エラー: '{original_pdf_file}' が見つかりません。")
except Exception as e:
    print(f"処理中にエラーが発生しました: {e}")

# 出力例:
# テスト用に 'original.pdf' を作成しました。
# 'original.pdf' をパスワード 'mysecretpassword' で暗号化し、'encrypted_document.pdf' として保存しました。

from PyPDF2 import PdfWriter, PdfReader
import os

# 暗号化されたPDFファイル名（上記で作成したファイルを使用）
encrypted_pdf_file = "encrypted_document.pdf"
decrypted_pdf_file = "decrypted_document.pdf"
correct_password = "mysecretpassword"
wrong_password = "wrongpassword"

# ファイルが存在しない場合はエラー
if not os.path.exists(encrypted_pdf_file):
    print(f"エラー: '{encrypted_pdf_file}' が見つかりません。パスワード設定のステップを実行してください。")
else:
    # 正しいパスワードで復号
    try:
        reader = PdfReader(encrypted_pdf_file)

        if reader.is_encrypted:
            # 正しいパスワードで復号を試みる
            reader.decrypt(correct_password)
            
            # 新しいPdfWriterオブジェクトを作成
            writer = PdfWriter()

            # 復号されたページをWriterにコピー
            for page in reader.pages:
                writer.add_page(page)

            # パスワードが解除されたPDFとして保存
            with open(decrypted_pdf_file, "wb") as f:
                writer.write(f)
            
            print(f"'{encrypted_pdf_file}' をパスワード解除し、'{decrypted_pdf_file}' として保存しました。")
        else:
            print(f"'{encrypted_pdf_file}' は暗号化されていません。")

    except Exception as e:
        print(f"正しいパスワードでの復号中にエラーが発生しました: {e}")

    # 間違ったパスワードで復号を試みる (エラー発生の確認)
    try:
        reader_wrong = PdfReader(encrypted_pdf_file)
        if reader_wrong.is_encrypted:
            print(f"\n--- 間違ったパスワード '{wrong_password}' で試行 ---")
            reader_wrong.decrypt(wrong_password) # ここでエラーが発生するはず
            print("間違ったパスワードで復号に成功してしまいました（これは想定外です）。")
        else:
            print(f"'{encrypted_pdf_file}' は暗号化されていません。")

    except Exception as e:
        print(f"間違ったパスワードでの復号は失敗しました: {e}")
        # PyPDF2 3.xでは、IncorrectPasswordError が発生する
        # PyPDF2 2.x以前では、PdfReadError が発生することがある

# 出力例:
# 'encrypted_document.pdf' をパスワード解除し、'decrypted_document.pdf' として保存しました。
#
# --- 間違ったパスワード 'wrongpassword' で試行 ---
# 間違ったパスワードでの復号は失敗しました: IncorrectPasswordError('Incorrect password')