Creating skeleton to receive DynamoDB features

Author: igorborgest

awswrangler/__init__.py

@@ -13,6 +13,7 @@
 from awswrangler.aurora import Aurora  # noqa
 from awswrangler.emr import EMR  # noqa
 from awswrangler.sagemaker import SageMaker  # noqa
+from awswrangler.dynamodb import DynamoDB  # noqa
 import awswrangler.utils  # noqa
 import awswrangler.data_types  # noqa
 
@@ -51,6 +52,7 @@ def __getattr__(self, name):
 athena: Athena = DynamicInstantiate("athena", Athena)  # type: ignore
 aurora: Aurora = DynamicInstantiate("aurora", Aurora)  # type: ignore
 redshift: Redshift = DynamicInstantiate("redshift", Redshift)  # type: ignore
+dynamodb: DynamoDB = DynamicInstantiate("dynamodb", DynamoDB)  # type: ignore
 sagemaker: SageMaker = DynamicInstantiate("sagemaker", SageMaker)  # type: ignore
 cloudwatchlogs: CloudWatchLogs = DynamicInstantiate("cloudwatchlogs", CloudWatchLogs)  # type: ignore
 

awswrangler/dynamodb.py

@@ -0,0 +1,17 @@
+from typing import TYPE_CHECKING
+from logging import getLogger, Logger
+
+from boto3 import client  # type: ignore
+
+if TYPE_CHECKING:
+    from awswrangler.session import Session
+
+logger: Logger = getLogger(__name__)
+
+
+class DynamoDB:
+    def __init__(self, session: "Session"):
+        self._session: "Session" = session
+        self._client_dynamodb: client = session.boto3_session.client(service_name="dynamodb",
+                                                                     use_ssl=True,
+                                                                     config=session.botocore_config)

awswrangler/session.py

@@ -15,6 +15,7 @@
 from awswrangler.aurora import Aurora
 from awswrangler.emr import EMR
 from awswrangler.sagemaker import SageMaker
+from awswrangler.dynamodb import DynamoDB
 from awswrangler.exceptions import AWSCredentialsNotFound
 
 PYSPARK_INSTALLED: bool = False
@@ -120,6 +121,7 @@ def __init__(self,
         self._aurora: Optional[Aurora] = None
         self._spark: Optional[Spark] = None
         self._sagemaker: Optional[SageMaker] = None
+        self._dynamodb: Optional[DynamoDB] = None
 
     def _load_new_boto3_session(self) -> None:
         """
@@ -310,6 +312,12 @@ def sagemaker(self) -> SageMaker:
             self._sagemaker = SageMaker(session=self)
         return self._sagemaker
 
+    @property
+    def dynamodb(self) -> DynamoDB:
+        if self._dynamodb is None:
+            self._dynamodb = DynamoDB(session=self)
+        return self._dynamodb
+
     @property
     def spark(self):
         if PYSPARK_INSTALLED is False:

requirements-dev.txt

@@ -5,7 +5,7 @@ pytest-cov~=2.8.1
 scikit-learn~=0.22.1
 cfn-lint~=0.27.2
 twine~=3.1.1
-wheel~=0.33.6
+wheel~=0.34.0
 sphinx~=2.3.1
 pyspark~=2.4.4
 pyspark-stubs~=2.4.0.post7