wip

Author: kauelima21

dynolayer/dynolayer.py

@@ -26,10 +26,10 @@ def __init__(
         self._offset = False
         self._order_by = {}
         self._count = 0
-        self._secondary_index = None
-        self._attributes_to_get = None
-        self._filter_expression = None
-        self._filter_params = None
+        self._secondary_index = ''
+        self._attributes_to_get = ''
+        self._filter_expression = ''
+        self._filter_params = {}
         self._filter_params_name = {}
         self._last_evaluated_key = {}
         self._error = ''
@@ -150,13 +150,13 @@ def find_by(
         if isinstance(attribute_value, dict) or isinstance(attribute_value, list):
             attribute_value = json.dumps(attribute_value)
 
-        self._filter_expression = f'#{attribute} {comparator} :{attribute}'
-        self._filter_params = {
+        self._filter_expression += f'#{attribute} {comparator} :{attribute}'
+        self._filter_params.update({
             f':{attribute}': attribute_value
-        }
-        self._filter_params_name = {
+        })
+        self._filter_params_name.update({
             f'#{attribute}': attribute
-        }
+        })
 
         return self
 
@@ -266,6 +266,7 @@ def _fetch(self, paginate_through_results: bool = False, just_count=False, objec
             if self._secondary_index:
                 scan_params.update({'IndexName': self._secondary_index})
 
+            self._is_find_by = False  # return to default value
             if self._is_query_operation:
                 return self._order_response(self._fetch_query(scan_params, paginate_through_results), object=object)
 

tests/unit/test_find.py

@@ -134,6 +134,9 @@ def test_it_should_find_a_collection_of_records_by_filter():
     assert response_find_by[0].data().get('last_name', None)
     assert response_find_by[0].data().get('stars', None)
 
+    response_by_id = user.find_by_id('123456').attributes_to_get('name').fetch(object=True)
+    assert response_find_by
+
 
 @mock_aws
 def test_it_should_paginate():