m

Author: Lucas McDonald

DynamoDbEncryption/runtimes/python/test/integ/encrypted/test_client.py

@@ -7,8 +7,6 @@
 from aws_dbesdk_dynamodb.encrypted.client import EncryptedClient
 from aws_dbesdk_dynamodb.encrypted.paginator import EncryptedPaginator
 
-from . import sort_dynamodb_json_lists
-
 from ...constants import (
     INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME,
     INTEG_TEST_DEFAULT_TABLE_CONFIGS,
@@ -45,10 +43,12 @@
     basic_transact_write_item_put_request_ddb,
     basic_transact_write_item_put_request_dict,
 )
+from . import sort_dynamodb_json_lists
 
 serializer = TypeSerializer()
 deserializer = TypeDeserializer()
 
+
 # Creates a matrix of tests for each value in param,
 # with a user-friendly string for test output:
 # expect_standard_dictionaries = True -> "standard_dicts"
@@ -73,6 +73,7 @@ def plaintext_client(expect_standard_dictionaries):
         client = boto3.client("dynamodb")
     return client
 
+
 # Creates a matrix of tests for each value in param,
 # with a user-friendly string for test output:
 # encrypted = True -> "encrypted"
@@ -89,6 +90,7 @@ def client(encrypted, expect_standard_dictionaries):
     else:
         return plaintext_client(expect_standard_dictionaries)
 
+
 # Creates a matrix of tests for each value in param,
 # with a user-friendly string for test output:
 # use_complex_item = True -> "complex_item"
@@ -155,6 +157,7 @@ def get_item_request(expect_standard_dictionaries, test_item):
         return {**basic_get_item_request_dict(test_item), "TableName": INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME}
     return basic_get_item_request_ddb(test_item)
 
+
 def test_GIVEN_valid_put_and_get_requests_WHEN_put_and_get_THEN_round_trip_passes(
     client, put_item_request, get_item_request
 ):

Examples/runtimes/python/DynamoDBEncryption/src/basic_put_get_example/with_encrypted_client.py

@@ -20,11 +20,11 @@
 )
 from aws_cryptographic_material_providers.mpl.references import IKeyring
 from aws_dbesdk_dynamodb.encrypted.client import EncryptedClient
-from aws_dbesdk_dynamodb.models.dynamodb import (
+from aws_dbesdk_dynamodb.structures.dynamodb import (
     DynamoDbTableEncryptionConfig,
     DynamoDbTablesEncryptionConfig,
 )
-from aws_dbesdk_dynamodb.models.structured_encryption import (
+from aws_dbesdk_dynamodb.structures.structured_encryption import (
     CryptoAction,
 )
 

Examples/runtimes/python/DynamoDBEncryption/src/item_encryptor/encrypt_decrypt_example.py

@@ -32,7 +32,9 @@
     DBEAlgorithmSuiteId,
 )
 from aws_cryptographic_material_providers.mpl.references import IKeyring
-
+from aws_dbesdk_dynamodb.encrypted.item import (
+    ItemEncryptor,
+)
 from aws_dbesdk_dynamodb.structures.item_encryptor import (
     DecryptItemInput,
     DynamoDbItemEncryptorConfig,
@@ -41,9 +43,6 @@
 from aws_dbesdk_dynamodb.structures.structured_encryption import (
     CryptoAction,
 )
-from aws_dbesdk_dynamodb.encrypted.item import (
-    ItemEncryptor,
-)
 
 
 def encrypt_decrypt_example(kms_key_id: str, ddb_table_name: str) -> None: