[Python] cryptographyの使い方 – ファイルの暗号化/復号化

pip install cryptography

pip show cryptography

pip install --upgrade cryptography

from cryptography.fernet import Fernet
# Fernetキーの生成
key = Fernet.generate_key()
# キーをファイルに保存
with open('secret.key', 'wb') as key_file:
    key_file.write(key)
print("Fernetキーが生成され、'secret.key'に保存されました。")

# ファイルの読み込み
file_path = 'example.txt'  # 読み込むファイルのパス
with open(file_path, 'rb') as file:
    file_data = file.read()
print("ファイルが正常に読み込まれました。")

# Fernetオブジェクトの作成
fernet = Fernet(key)
# ファイルデータの暗号化
encrypted_data = fernet.encrypt(file_data)
print("ファイルが正常に暗号化されました。")

# 暗号化データの保存
with open('encrypted_file.enc', 'wb') as encrypted_file:
    encrypted_file.write(encrypted_data)
print("暗号化データが'encrypted_file.enc'に保存されました。")

# 暗号化ファイルの読み込み
with open('encrypted_file.enc', 'rb') as encrypted_file:
    encrypted_data = encrypted_file.read()
print("暗号化ファイルが正常に読み込まれました。")

from cryptography.fernet import Fernet
# キーの読み込み
with open('secret.key', 'rb') as key_file:
    key = key_file.read()
# Fernetオブジェクトの作成
fernet = Fernet(key)
# ファイルデータの復号化
decrypted_data = fernet.decrypt(encrypted_data)
print("ファイルが正常に復号化されました。")

# 復号化データの保存
with open('decrypted_file.txt', 'wb') as decrypted_file:
    decrypted_file.write(decrypted_data)
print("復号化データが'decrypted_file.txt'に保存されました。")

from cryptography.fernet import Fernet, InvalidToken
# 無効なキーの例
invalid_key = b'invalid_key'
# Fernetオブジェクトの作成
fernet = Fernet(invalid_key)
try:
    # 復号化処理
    decrypted_data = fernet.decrypt(encrypted_data)
except InvalidToken:
    print("エラー: 無効なキーです。復号化に失敗しました。")

file_path = 'non_existent_file.txt'
try:
    with open(file_path, 'rb') as file:
        file_data = file.read()
except FileNotFoundError:
    print(f"エラー: '{file_path}' が見つかりません。")

try:
    # 復号化処理
    decrypted_data = fernet.decrypt(encrypted_data)
except InvalidToken:
    print("エラー: 復号化に失敗しました。暗号化されたデータが無効です。")

import os
from cryptography.fernet import Fernet
# Fernetキーの読み込み
with open('secret.key', 'rb') as key_file:
    key = key_file.read()
fernet = Fernet(key)
# 暗号化するファイルのディレクトリ
directory = 'files_to_encrypt'
# ディレクトリ内のファイルを一括暗号化
for filename in os.listdir(directory):
    if filename.endswith('.txt'):
        with open(os.path.join(directory, filename), 'rb') as file:
            file_data = file.read()
        encrypted_data = fernet.encrypt(file_data)
        with open(os.path.join(directory, f'encrypted_{filename}'), 'wb') as encrypted_file:
            encrypted_file.write(encrypted_data)
print("すべてのファイルが正常に暗号化されました。")

import os
from cryptography.fernet import Fernet
# 環境変数からキーを取得
key = os.environ.get('FERNET_KEY').encode()
fernet = Fernet(key)

from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
from cryptography.fernet import Fernet
import os
class FileHandler(FileSystemEventHandler):
    def on_created(self, event):
        if event.is_directory:
            return
        with open('secret.key', 'rb') as key_file:
            key = key_file.read()
        fernet = Fernet(key)
        with open(event.src_path, 'rb') as file:
            file_data = file.read()
        encrypted_data = fernet.encrypt(file_data)
        with open(event.src_path + '.enc', 'wb') as encrypted_file:
            encrypted_file.write(encrypted_data)
        print(f"{event.src_path} が暗号化されました。")
# 監視するディレクトリ
path = 'directory_to_watch'
event_handler = FileHandler()
observer = Observer()
observer.schedule(event_handler, path, recursive=False)
observer.start()
try:
    while True:
        pass
except KeyboardInterrupt:
    observer.stop()
observer.join()

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import rsa, padding
from cryptography.hazmat.primitives import serialization, hashes
# RSA鍵ペアの生成
private_key = rsa.generate_private_key(public_exponent=65537, key_size=2048, backend=default_backend())
public_key = private_key.public_key()
# 公開鍵の保存
with open('public_key.pem', 'wb') as pub_file:
    pub_file.write(public_key.public_bytes(encoding=serialization.Encoding.PEM, format=serialization.PublicFormat.SubjectPublicKeyInfo))
# ファイルの暗号化
file_path = 'example.txt'
with open(file_path, 'rb') as file:
    file_data = file.read()
encrypted_data = public_key.encrypt(
    file_data,
    padding.OAEP(
        mgf=padding.MGF1(algorithm=hashes.SHA256()),
        algorithm=hashes.SHA256(),
        label=None
    )
)
# 暗号化データの保存
with open('encrypted_file_rsa.enc', 'wb') as enc_file:
    enc_file.write(encrypted_data)
print("ファイルがRSAで暗号化されました。")