[python] 반복문으로 파일 암호화하기

05 Dec 2023

python

이번에는 Python을 사용하여 파일을 암호화하는 방법을 알아보겠습니다. 반복문을 활용하여 간단하게 파일을 암호화하는 과정을 구현할 것입니다.

필요한 라이브러리 설치하기

파일 암호화를 위해서는 cryptography 라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 라이브러리를 설치합니다.

pip install cryptography

파일 암호화하기

암호화할 파일을 선택하고, 암호화 키를 생성한 후 암호화를 수행하는 코드를 작성해보겠습니다.

from cryptography.fernet import Fernet

# 암호화 키 생성
def generate_key():
    key = Fernet.generate_key()
    with open("encryption.key", "wb") as key_file:
        key_file.write(key)
        
# 파일 암호화
def encrypt_file(file_path, key):
    with open(file_path, "rb") as file:
        file_data = file.read()
        
    f = Fernet(key)
    encrypted_data = f.encrypt(file_data)
    
    with open(file_path, "wb") as file:
        file.write(encrypted_data)

# 암호화할 파일 경로와 암호화 키 생성
file_path = "test.txt"
generate_key()

# 암호화 키 불러오기
with open("encryption.key", "rb") as key_file:
    key = key_file.read()

# 파일 암호화 수행
encrypt_file(file_path, key)

위의 코드에서 generate_key() 함수는 암호화에 사용할 키를 생성하고, encrypt_file() 함수는 파일을 암호화하는 역할을 합니다.

암호화할 파일은 test.txt 파일로 가정하고, 암호화 키는 encryption.key 파일에 저장합니다. 이렇게 생성된 암호화 키를 encrypt_file() 함수에 전달하여 파일을 암호화합니다.

결론

Python을 사용하여 파일을 반복문을 이용해서 암호화하는 방법을 알아보았습니다. cryptography 라이브러리를 활용하여 간단하게 파일을 암호화할 수 있습니다. 파일 암호화는 보안을 강화하는데 도움이 되며, 개인정보와 같은 중요한 데이터를 안전하게 보호할 수 있습니다.