@@ -288,9 +288,9 @@ def on_encrypt(self, encryption_materials):
288
288
289
289
# EncryptedData to EncryptedDataKey
290
290
encrypted_data_key = serialize_wrapped_key (
291
- key_provider = key_provider ,
292
- wrapping_algorithm = wrapping_algorithm ,
293
- wrapping_key_id = key_name ,
291
+ key_provider = self . _key_provider ,
292
+ wrapping_algorithm = self . _wrapping_algorithm ,
293
+ wrapping_key_id = self . _key_name ,
294
294
encrypted_wrapped_key = encrypted_wrapped_key ,
295
295
)
296
296
@@ -338,12 +338,14 @@ def on_decrypt(self, decryption_materials, encrypted_data_keys):
338
338
if plaintext_data_key :
339
339
# Create a keyring trace
340
340
keyring_trace = KeyringTrace (
341
- wrapping_key = key_provider , flags = {KeyringTraceFlag .WRAPPING_KEY_DECRYPTED_DATA_KEY }
341
+ wrapping_key = self . _key_provider , flags = {KeyringTraceFlag .WRAPPING_KEY_DECRYPTED_DATA_KEY }
342
342
)
343
343
344
344
# Update decryption materials
345
345
data_encryption_key = RawDataKey (
346
- key_provider = MasterKeyInfo (provider_id = key_provider .provider_id , key_info = key_name ),
346
+ key_provider = MasterKeyInfo (
347
+ provider_id = self ._key_provider .provider_id , key_info = self .key_name
348
+ ),
347
349
data_key = plaintext_data_key ,
348
350
)
349
351
decryption_materials .add_data_encryption_key (data_encryption_key , keyring_trace )
0 commit comments