m

Author: Lucas McDonald

DynamoDbEncryption/runtimes/python/src/aws_dbesdk_dynamodb/__init__.py

@@ -1,6 +1,8 @@
 # Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+"""AWS Database Encryption SDK for DynamoDB."""
+
 # Initialize generated Dafny
 from .internaldafny.generated import module_  # noqa: F401
 

DynamoDbEncryption/runtimes/python/src/aws_dbesdk_dynamodb/encrypted/boto3_interface.py

@@ -1,21 +1,25 @@
 # Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
+"""Interface for interacting with DynamoDB using boto3."""
 import abc
 from typing import Any, Dict
 from abc import abstractmethod
 
+
 class EncryptedBotoInterface(abc.ABC):
     """Interface for encrypted boto3 interfaces."""
 
-    def _copy_sdk_response_to_dbesdk_response(self, sdk_response: Dict[str, Any], dbesdk_response: Dict[str, Any]) -> Dict[str, Any]:
+    def _copy_sdk_response_to_dbesdk_response(
+        self, sdk_response: Dict[str, Any], dbesdk_response: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Copy any missing fields from the SDK response to the DBESDK response.
 
         Args:
-            sdk_response: The raw SDK response
-            dbesdk_response: The current DBESDK response
+            sdk_response: The raw SDK response.
+            dbesdk_response: The current DBESDK response.
 
         Returns:
-            dict: The DBESDK response with any missing fields copied from SDK response
+            dict: The DBESDK response with any missing fields copied from SDK response.
         """
         for sdk_response_key, sdk_response_value in sdk_response.items():
             if sdk_response_key not in dbesdk_response:
@@ -42,4 +46,6 @@ def __getattr__(self, name: str) -> Any:
         client_attr = getattr(self, self._boto_client_attr_name)
         if hasattr(client_attr, name):
             return getattr(client_attr, name)
-        raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'")
+        raise AttributeError(
+            f"'{self.__class__.__name__}' object has no attribute '{name}'"
+        )

DynamoDbEncryption/runtimes/python/src/aws_dbesdk_dynamodb/encrypted/client.py

@@ -20,6 +20,7 @@
     ddb_to_dict,
 )
 
+
 class ItemEncryptor:
     """Class providing item-level encryption for DynamoDB items / Python dictionaries."""
 
@@ -35,7 +36,7 @@ def __init__(
         Parameters:
         item_encryptor_config (DynamoDbItemEncryptorConfig): Encryption configuration object.
         """
-        self._internal_client = DynamoDbItemEncryptor(config = item_encryptor_config)
+        self._internal_client = DynamoDbItemEncryptor(config=item_encryptor_config)
 
     def encrypt_python_item(
         self,
@@ -60,15 +61,15 @@ def encrypt_python_item(
 
         Parameters:
         plaintext_dict_item (dict[str, Any]): A standard Python dictionary.
-        
+
         Returns:
         EncryptItemOutput: Structure containing the following fields:
             - `encrypted_item` (dict[str, Any]): The encrypted Python dictionary.
                 **Note:** The item was encrypted as DynamoDB JSON, then transformed to a Python dictionary.
             - `parsed_header` (Optional[ParsedHeader]): The encrypted DynamoDB item's header (parsed `aws_dbe_head` value).
 
         Example:
-        
+
         >>> plaintext_item = {
         ...     'some': 'data',
         ...     'more': 5
@@ -78,11 +79,13 @@ def encrypt_python_item(
         >>> header = encrypt_output.parsed_header
         """
         plaintext_ddb_item = dict_to_ddb(plaintext_dict_item)
-        encrypted_ddb_item: EncryptItemOutput = self.encrypt_dynamodb_item(plaintext_ddb_item)
+        encrypted_ddb_item: EncryptItemOutput = self.encrypt_dynamodb_item(
+            plaintext_ddb_item
+        )
         encrypted_dict_item = ddb_to_dict(encrypted_ddb_item.encrypted_item)
         return EncryptItemOutput(
-            encrypted_item = encrypted_dict_item,
-            parsed_header = encrypted_ddb_item.parsed_header
+            encrypted_item=encrypted_dict_item,
+            parsed_header=encrypted_ddb_item.parsed_header,
         )
 
     def encrypt_dynamodb_item(
@@ -101,15 +104,15 @@ def encrypt_dynamodb_item(
 
         Parameters:
         plaintext_dynamodb_item (dict[str, dict[str, Any]]): The item to encrypt formatted as DynamoDB JSON.
-        
+
         Returns:
         EncryptItemOutput: Structure containing the following fields:
             - `encrypted_item` (dict[str, Any]): A dictionary containing the encrypted DynamoDB item
                 formatted as DynamoDB JSON.
             - `parsed_header` (Optional[ParsedHeader]): The encrypted DynamoDB item's header (`aws_dbe_head` value).
 
         Example:
-        
+
         >>> plaintext_item = {
         ...     'some': {'S': 'data'},
         ...     'more': {'N': '5'}
@@ -119,9 +122,7 @@ def encrypt_dynamodb_item(
         >>> header = encrypt_output.parsed_header
         """
         return self.encrypt_item(
-            EncryptItemInput(
-                plaintext_item = plaintext_dynamodb_item
-            )
+            EncryptItemInput(plaintext_item=plaintext_dynamodb_item)
         )
 
     def encrypt_item(
@@ -137,14 +138,14 @@ def encrypt_item(
         Parameters:
         encrypt_item_input (EncryptItemInput): Structure containing the following field:
             - `plaintext_item` (dict[str, Any]): The item to encrypt formatted as DynamoDB JSON.
-        
+
         Returns:
         EncryptItemOutput: Structure containing the following fields:
             - `encrypted_item` (dict[str, Any]): The encrypted DynamoDB item formatted as DynamoDB JSON.
             - `parsed_header` (Optional[ParsedHeader]): The encrypted DynamoDB item's header (`aws_dbe_head` value).
 
         Example:
-        
+
         >>> plaintext_item = {
         ...     'some': {'S': 'data'},
         ...     'more': {'N': '5'}
@@ -182,15 +183,15 @@ def decrypt_python_item(
 
         Parameters:
         encrypted_dict_item (dict[str, Any]): A standard Python dictionary with encrypted values.
-        
+
         Returns:
         DecryptItemOutput: Structure containing the following fields:
             - `plaintext_item` (dict[str, Any]): The decrypted Python dictionary.
                 **Note:** The item was decrypted as DynamoDB JSON, then transformed to a Python dictionary.
             - `parsed_header` (Optional[ParsedHeader]): The encrypted DynamoDB item's header (parsed `aws_dbe_head` value).
 
         Example:
-        
+
         >>> encrypted_item = {
         ...     'some': b'ENCRYPTED_DATA',
         ...     'more': b'ENCRYPTED_DATA',
@@ -200,11 +201,13 @@ def decrypt_python_item(
         >>> header = decrypt_output.parsed_header
         """
         encrypted_ddb_item = dict_to_ddb(encrypted_dict_item)
-        plaintext_ddb_item: DecryptItemOutput = self.decrypt_dynamodb_item(encrypted_ddb_item)
+        plaintext_ddb_item: DecryptItemOutput = self.decrypt_dynamodb_item(
+            encrypted_ddb_item
+        )
         plaintext_dict_item = ddb_to_dict(plaintext_ddb_item.plaintext_item)
         return DecryptItemOutput(
-            plaintext_item = plaintext_dict_item,
-            parsed_header = plaintext_ddb_item.parsed_header
+            plaintext_item=plaintext_dict_item,
+            parsed_header=plaintext_ddb_item.parsed_header,
         )
 
     def decrypt_dynamodb_item(
@@ -223,14 +226,14 @@ def decrypt_dynamodb_item(
 
         Parameters:
         encrypted_ddb_item (dict[str, dict[str, Any]]): The item to decrypt formatted as DynamoDB JSON.
-        
+
         Returns:
         DecryptItemOutput: Structure containing the following fields:
             - `plaintext_item` (dict[str, Any]): The plaintext DynamoDB item formatted as DynamoDB JSON.
             - `parsed_header` (Optional[ParsedHeader]): The decrypted DynamoDB item's header (`aws_dbe_head` value).
 
         Example:
-        
+
         >>> encrypted_item = {
         ...     'some': {'B': b'ENCRYPTED_DATA'},
         ...     'more': {'B': b'ENCRYPTED_DATA'}
@@ -240,9 +243,7 @@ def decrypt_dynamodb_item(
         >>> header = decrypt_output.parsed_header
         """
         return self.decrypt_item(
-            DecryptItemInput(
-                encrypted_item = encrypted_dynamodb_item
-            )
+            DecryptItemInput(encrypted_item=encrypted_dynamodb_item)
         )
 
     def decrypt_item(
@@ -258,14 +259,14 @@ def decrypt_item(
         Parameters:
         decrypt_item_input (DecryptItemInput): Structure containing the following field:
             - `encrypted_item` (dict[str, Any]): The item to decrypt formatted as DynamoDB JSON.
-        
+
         Returns:
         DecryptItemOutput: Structure containing the following fields:
             - `plaintext_item` (dict[str, Any]): The decrypted DynamoDB item formatted as DynamoDB JSON.
             - `parsed_header` (Optional[ParsedHeader]): The decrypted DynamoDB item's header (`aws_dbe_head` value).
 
         Example:
-        
+
         >>> encrypted_item = {
         ...     'some': {'B': b'ENCRYPTED_DATA'},
         ...     'more': {'B': b'ENCRYPTED_DATA'}