Formatting.

Author: Harinath

pydanticrud/backends/dynamodb.py

@@ -96,7 +96,7 @@ def _to_epoch_float(dt):
 
 
 def chunk_list(lst, size):
-    return [lst[i: i + size] for i in range(0, len(lst), size)]
+    return [lst[i : i + size] for i in range(0, len(lst), size)]
 
 
 def index_definition(index_name, keys, gsi=False):
@@ -429,11 +429,17 @@ def batch_save(self, items: list) -> dict:
         # chunk list for size limit of 25 items to write using this batch_write operation refer below.
         # https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/dynamodb/client/batch_write_item.html#:~:text=The%20BatchWriteItem%20operation,Data%20Types.
         for chunk in chunk_list(items, 20):
-            serialized_items = [self.serializer.serialize_record(item.dict(by_alias=True)) for item in chunk]
+            serialized_items = [
+                self.serializer.serialize_record(item.dict(by_alias=True)) for item in chunk
+            ]
             for serialized_item in serialized_items:
-                request_items[self.table_name].append({"PutRequest":
-                                                           {"Item": serialized_item }#self.serializer.serialize_item(serialized_item)}
-                                                       })
+                request_items[self.table_name].append(
+                    {
+                        "PutRequest": {
+                            "Item": serialized_item
+                        }  # self.serializer.serialize_item(serialized_item)}
+                    }
+                )
         try:
             response = self.dynamodb.batch_write_item(RequestItems=request_items)
         except ClientError as e:

pydanticrud/dynamo_type_serializer.py

@@ -3,6 +3,7 @@
 
 NoneType = type(None)
 
+
 class DynamoTypeSerializer:
     def __init__(self):
         self._methods = {
@@ -20,12 +21,10 @@ def serialize_item(self, item) -> dict:
         return {k: self.serialize(v) for k, v in item.items()}
 
     def serialize_mapping(self, value: Mapping) -> dict:
-        return {
-            'M': {k: self.serialize(v) for k, v in value.items()}
-        }
+        return {"M": {k: self.serialize(v) for k, v in value.items()}}
 
     def serialize_list(self, value: Union[list, tuple]) -> dict:
-        return {'L': [self.serialize(item) for item in value]}
+        return {"L": [self.serialize(item) for item in value]}
 
     def serialize(self, value: Any):
         value_type = type(value)
@@ -38,17 +37,20 @@ def serialize(self, value: Any):
                     return route(value)
         else:
             return route(value)
-        raise TypeError(f'The value {value} is not Dynamodb serializable type.')
+        raise TypeError(f"The value {value} is not Dynamodb serializable type.")
+
 
 def serialize_bool(value: bool) -> dict:
-    return {'BOOL': value}
+    return {"BOOL": value}
+
 
 def serialize_number(value: Union[int, float, decimal.Decimal]) -> dict:
-    return {'N': str(value)}
+    return {"N": str(value)}
+
 
 def serialize_none(value: None) -> dict:
-    return {'NULL': True}
+    return {"NULL": True}
 
-def serialize_str(value: str) -> dict:
-    return {'S': value}
 
+def serialize_str(value: str) -> dict:
+    return {"S": value}

tests/random_values.py

@@ -15,7 +15,7 @@ def random_datetime():
         random.randint(1, 28),
         random.randint(1, 12),
         random.randint(1, 59),
-        0
+        0,
     )
 
 

tests/test_dynamo_type_serializer.py

@@ -3,54 +3,54 @@
 from pydanticrud.backends.dynamodb import DynamoSerializer
 
 serializer = DynamoSerializer(schema={})
+
+
 def test_int():
-    assert serializer.serialize(0) == {'N': '0'}
-    assert serializer.serialize(34) == {'N': '34'}
+    assert serializer.serialize(0) == {"N": "0"}
+    assert serializer.serialize(34) == {"N": "34"}
 
 
 def test_decimal():
-    assert serializer.serialize(Decimal('1.2345')) == {'N': '1.2345'}
-    assert serializer.serialize(Decimal('0.00')) == {'N': '0.00'}
+    assert serializer.serialize(Decimal("1.2345")) == {"N": "1.2345"}
+    assert serializer.serialize(Decimal("0.00")) == {"N": "0.00"}
+
 
 def test_str():
-    assert serializer.serialize('') == {'S': ''}
-    assert serializer.serialize('hello') == {'S': 'hello'}
-    assert serializer.serialize('hello world') == {'S': 'hello world'}
+    assert serializer.serialize("") == {"S": ""}
+    assert serializer.serialize("hello") == {"S": "hello"}
+    assert serializer.serialize("hello world") == {"S": "hello world"}
+
 
 def test_none():
-    return serializer.serialize(None) == {'NULL': True}
+    return serializer.serialize(None) == {"NULL": True}
+
 
 def test_bools():
-    assert serializer.serialize(True) == {'BOOL': True}
-    assert serializer.serialize(False) == {'BOOL': False}
+    assert serializer.serialize(True) == {"BOOL": True}
+    assert serializer.serialize(False) == {"BOOL": False}
 
 
 def test_list():
-    assert serializer.serialize([1, 'abc', None, None, True]) == {
-        'L': [{'N': '1'}, {'S': 'abc'}, {'NULL': True}, {'NULL': True},
-              {'BOOL': True}]
+    assert serializer.serialize([1, "abc", None, None, True]) == {
+        "L": [{"N": "1"}, {"S": "abc"}, {"NULL": True}, {"NULL": True}, {"BOOL": True}]
     }
 
+
 def test_maps():
-    assert serializer.serialize({
-        'hello': 'world',
-        'another': 123,
-        'loop': {'Working': True}
-    }) == {
-        'M': {
-            'hello': {'S': 'world'},
-            'another': {'N': '123'},
-            'loop': {'M': {'Working': {'BOOL': True}}}
+    assert serializer.serialize({"hello": "world", "another": 123, "loop": {"Working": True}}) == {
+        "M": {
+            "hello": {"S": "world"},
+            "another": {"N": "123"},
+            "loop": {"M": {"Working": {"BOOL": True}}},
         }
     }
 
+
 def test_item():
-    assert serializer.serialize_item({
-        'Hello': 'World',
-        'another': 123,
-        'loop': {'Working': True}
-    }) == {
-        'Hello': {'S': 'World'},
-        'another': {'N': '123'},
-        'loop': {'M': {'Working': {'BOOL': True}}}
+    assert serializer.serialize_item(
+        {"Hello": "World", "another": 123, "loop": {"Working": True}}
+    ) == {
+        "Hello": {"S": "World"},
+        "another": {"N": "123"},
+        "loop": {"M": {"Working": {"BOOL": True}}},
     }