@@ -35,7 +35,7 @@ def from_client(cls) -> "KeyEncrypt":
35
35
# snippet-end:[python.example_code.kms.KeyEncrypt.decl]
36
36
37
37
# snippet-start:[python.example_code.kms.Encrypt]
38
- def encrypt (self , key_id : str , text : str ) -> str :
38
+ def encrypt (self , key_id : str , text : str ) -> bytes :
39
39
"""
40
40
Encrypts text by using the specified key.
41
41
@@ -64,7 +64,7 @@ def encrypt(self, key_id: str, text: str) -> str:
64
64
# snippet-end:[python.example_code.kms.Encrypt]
65
65
66
66
# snippet-start:[python.example_code.kms.Decrypt]
67
- def decrypt (self , key_id : str , cipher_text : str ) -> bytes :
67
+ def decrypt (self , key_id : str , cipher_text : bytes ) -> str :
68
68
"""
69
69
Decrypts text previously encrypted with a key.
70
70
@@ -75,7 +75,7 @@ def decrypt(self, key_id: str, cipher_text: str) -> bytes:
75
75
try :
76
76
return self .kms_client .decrypt (KeyId = key_id , CiphertextBlob = cipher_text )[
77
77
"Plaintext"
78
- ]
78
+ ]. decode ()
79
79
except ClientError as err :
80
80
logger .error (
81
81
"Couldn't decrypt your ciphertext. Here's why: %s" ,
@@ -198,7 +198,7 @@ def key_encryption(kms_client):
198
198
answer = input ("Ready to decrypt your ciphertext (y/n)? " )
199
199
if answer .lower () == "y" :
200
200
decrypted_text = key_encrypt .decrypt (key_id , cipher_text )
201
- print (f"Your plaintext is { decrypted_text . decode () } " )
201
+ print (f"Your plaintext is { decrypted_text } " )
202
202
print ("-" * 88 )
203
203
key_encrypt .re_encrypt (key_id , cipher_text )
204
204
else :
0 commit comments