Merge pull request #9 from oncase/feat/add_new_aws_services

DynamoDB resource bug fixed

.github/workflows/publish.yml

@@ -35,8 +35,8 @@ jobs:
         uses: marvinpinto/[email protected]
         with:
           repo_token: ${{ secrets.GITHUB_TOKEN }}
-          title: 'v1.1.2 Ldap module added'
+          title: 'v1.1.3 DynamoDB resource bug fixed'
           prerelease: false
-          automatic_release_tag: v1.1.2
+          automatic_release_tag: v1.1.3
           files: |
             LICENSE

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyscora-wrangler"
-version = "1.1.2"
+version = "1.1.3"
 description = "Python lib for DE"
 authors = ["Oncase <[email protected]>"]
 maintainers = ["Guilherme Morone <[email protected]>"]

pyscora_wrangler/aws/dynamodb/__init__.py

@@ -98,11 +98,11 @@ def get_data_by_key(
     """
 
     session = get_boto3_session(boto3_session)
-    client = session.client(DYNAMODB_SERVICE_NAME)
+    resource = session.resource(DYNAMODB_SERVICE_NAME)
 
     data = None
 
-    dynamo_table = client.Table(table_name)
+    dynamo_table = resource.Table(table_name)
     response = dynamo_table.query(KeyConditionExpression=Key(key).eq(value))
     response = response.get("Items", [])
 
@@ -167,9 +167,9 @@ def put_item(
     """
 
     session = get_boto3_session(boto3_session)
-    client = session.client(DYNAMODB_SERVICE_NAME)
+    resource = session.resource(DYNAMODB_SERVICE_NAME)
 
-    table = client.Table(table_name)
+    table = resource.Table(table_name)
     encoded_data = get_data_encoded(data) if encode_data else data
     item = table.put_item(Item=encoded_data, *dynamodb_additional_args, **dynamodb_additional_kwargs)