Skip to content

Commit 49f3477

Browse files
committed
simplify create_data_key() call
1 parent 7404286 commit 49f3477

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

django_mongodb_backend/schema.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -501,7 +501,8 @@ def _get_encrypted_fields(self, model, key_alt_name=None, path_prefix=None):
501501

502502
kms_provider = router.kms_provider(model)
503503
kms_providers = auto_encryption_opts._kms_providers
504-
master_key = kms_providers.get(kms_provider, {})
504+
# Providing master_key raises an error for the local provider.
505+
master_key = kms_providers[kms_provider] if kms_provider != "local" else None
505506
client_encryption = self.connection.client_encryption
506507

507508
field_list = []
@@ -526,10 +527,11 @@ def _get_encrypted_fields(self, model, key_alt_name=None, path_prefix=None):
526527
if data_key:
527528
data_key = data_key["_id"]
528529
else:
529-
kwargs = {"kms_provider": kms_provider, "key_alt_names": [new_key_alt_name]}
530-
if kms_provider != "local":
531-
kwargs["master_key"] = master_key
532-
data_key = client_encryption.create_data_key(**kwargs)
530+
data_key = client_encryption.create_data_key(
531+
kms_provider=kms_provider,
532+
key_alt_names=[new_key_alt_name],
533+
master_key=master_key,
534+
)
533535
field_dict = {
534536
"bsonType": bson_type,
535537
"path": path,

0 commit comments

Comments
 (0)