Python 스크립트 - Python의 AES 암호화

이 튜토리얼에서는 다음 Python 라이브러리가 필요합니다.

cryptography



설치하지 않은 경우 "PowerShell"또는 "터미널"(Windows)을 열고 다음 코드를 사용하여 설치하십시오.

pip install cryptography


일하고 있는



암호화는 특별한 AES 암호화 모듈입니다. 이는 파일을 암호화하고 동일한 파일을 해독하려면 파일을 암호화하는 데 사용된 동일한 키/토큰이 있어야 하는 특수 토큰을 생성해야 함을 의미합니다. 다른 사람이 토큰에 액세스할 수 없기 때문에 다른 암호화보다 안전합니다.

모듈 가져오기




from cryptography.fernet import Fernet


고유 키 생성 및 저장




def write_key():
    key = Fernet.generate_key()
    with open("key.key", "wb") as key_file:
        key_file.write(key)


키 로드




def load_key():
    return open("key.key", "rb").read()


암호화 기능




def encrypt(filename, key):
    f = Fernet(key)
    with open(filename, "rb") as file:
        file_data = file.read()
    encrypted_data = f.encrypt(file_data)
    with open(f'{filename}.enc', "wb") as file:
        file.write(encrypted_data)


복호화 기능




def decrypt(filename, key):
    f = Fernet(key)
    with open(f'{filename}.enc', "rb") as file:
        # read the encrypted data
        encrypted_data = file.read()
    # decrypt data
    decrypted_data = f.decrypt(encrypted_data)
    with open(f'{filename}.dec', "wb") as file:
        file.write(decrypted_data)


실제 작업




write_key()

# AFTER RUNNING THE FILE FOR THE FIRST TIME COMMENT 
# OUT THE ABOVE LINE TO PREVENT OVERWRITING OF TKE 'KEY.KEY'
# FILE 

#LOADING THE KEY
key = load_key()

file = 'YOUR FILENAME HERE'

#THIS FUNCTION FOR ENCRYPTION  
encrypt(file,key)

#THIS FUNCTION FOR DECRYPTION
decrypt(file,key)

좋은 웹페이지 즐겨찾기