Skip to content

Commit 55ee0d8

Browse files
author
Harinath
committed
Revert "Formatting."
This reverts commit 8abd275.
1 parent 8abd275 commit 55ee0d8

File tree

5 files changed

+133
-198
lines changed

5 files changed

+133
-198
lines changed

pydanticrud/backends/dynamodb.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def _to_epoch_float(dt):
9696

9797

9898
def chunk_list(lst, size):
99-
return [lst[i : i + size] for i in range(0, len(lst), size)]
99+
return [lst[i: i + size] for i in range(0, len(lst), size)]
100100

101101

102102
def index_definition(index_name, keys, gsi=False):
@@ -429,17 +429,11 @@ def batch_save(self, items: list) -> dict:
429429
# chunk list for size limit of 25 items to write using this batch_write operation refer below.
430430
# https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/dynamodb/client/batch_write_item.html#:~:text=The%20BatchWriteItem%20operation,Data%20Types.
431431
for chunk in chunk_list(items, 20):
432-
serialized_items = [
433-
self.serializer.serialize_record(item.dict(by_alias=True)) for item in chunk
434-
]
432+
serialized_items = [self.serializer.serialize_record(item.dict(by_alias=True)) for item in chunk]
435433
for serialized_item in serialized_items:
436-
request_items[self.table_name].append(
437-
{
438-
"PutRequest": {
439-
"Item": serialized_item
440-
} # self.serializer.serialize_item(serialized_item)}
441-
}
442-
)
434+
request_items[self.table_name].append({"PutRequest":
435+
{"Item": serialized_item }#self.serializer.serialize_item(serialized_item)}
436+
})
443437
try:
444438
response = self.dynamodb.batch_write_item(RequestItems=request_items)
445439
except ClientError as e:

pydanticrud/dynamo_type_serializer.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33

44
NoneType = type(None)
55

6-
76
class DynamoTypeSerializer:
87
def __init__(self):
98
self._methods = {
@@ -21,10 +20,12 @@ def serialize_item(self, item) -> dict:
2120
return {k: self.serialize(v) for k, v in item.items()}
2221

2322
def serialize_mapping(self, value: Mapping) -> dict:
24-
return {"M": {k: self.serialize(v) for k, v in value.items()}}
23+
return {
24+
'M': {k: self.serialize(v) for k, v in value.items()}
25+
}
2526

2627
def serialize_list(self, value: Union[list, tuple]) -> dict:
27-
return {"L": [self.serialize(item) for item in value]}
28+
return {'L': [self.serialize(item) for item in value]}
2829

2930
def serialize(self, value: Any):
3031
value_type = type(value)
@@ -37,20 +38,17 @@ def serialize(self, value: Any):
3738
return route(value)
3839
else:
3940
return route(value)
40-
raise TypeError(f"The value {value} is not Dynamodb serializable type.")
41-
41+
raise TypeError(f'The value {value} is not Dynamodb serializable type.')
4242

4343
def serialize_bool(value: bool) -> dict:
44-
return {"BOOL": value}
45-
44+
return {'BOOL': value}
4645

4746
def serialize_number(value: Union[int, float, decimal.Decimal]) -> dict:
48-
return {"N": str(value)}
49-
47+
return {'N': str(value)}
5048

5149
def serialize_none(value: None) -> dict:
52-
return {"NULL": True}
53-
50+
return {'NULL': True}
5451

5552
def serialize_str(value: str) -> dict:
56-
return {"S": value}
53+
return {'S': value}
54+

tests/random_values.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def random_datetime():
1515
random.randint(1, 28),
1616
random.randint(1, 12),
1717
random.randint(1, 59),
18-
0,
18+
0
1919
)
2020

2121

tests/test_dynamo_type_serializer.py

Lines changed: 30 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -3,54 +3,54 @@
33
from pydanticrud.backends.dynamodb import DynamoSerializer
44

55
serializer = DynamoSerializer(schema={})
6-
7-
86
def test_int():
9-
assert serializer.serialize(0) == {"N": "0"}
10-
assert serializer.serialize(34) == {"N": "34"}
7+
assert serializer.serialize(0) == {'N': '0'}
8+
assert serializer.serialize(34) == {'N': '34'}
119

1210

1311
def test_decimal():
14-
assert serializer.serialize(Decimal("1.2345")) == {"N": "1.2345"}
15-
assert serializer.serialize(Decimal("0.00")) == {"N": "0.00"}
16-
12+
assert serializer.serialize(Decimal('1.2345')) == {'N': '1.2345'}
13+
assert serializer.serialize(Decimal('0.00')) == {'N': '0.00'}
1714

1815
def test_str():
19-
assert serializer.serialize("") == {"S": ""}
20-
assert serializer.serialize("hello") == {"S": "hello"}
21-
assert serializer.serialize("hello world") == {"S": "hello world"}
22-
16+
assert serializer.serialize('') == {'S': ''}
17+
assert serializer.serialize('hello') == {'S': 'hello'}
18+
assert serializer.serialize('hello world') == {'S': 'hello world'}
2319

2420
def test_none():
25-
return serializer.serialize(None) == {"NULL": True}
26-
21+
return serializer.serialize(None) == {'NULL': True}
2722

2823
def test_bools():
29-
assert serializer.serialize(True) == {"BOOL": True}
30-
assert serializer.serialize(False) == {"BOOL": False}
24+
assert serializer.serialize(True) == {'BOOL': True}
25+
assert serializer.serialize(False) == {'BOOL': False}
3126

3227

3328
def test_list():
34-
assert serializer.serialize([1, "abc", None, None, True]) == {
35-
"L": [{"N": "1"}, {"S": "abc"}, {"NULL": True}, {"NULL": True}, {"BOOL": True}]
29+
assert serializer.serialize([1, 'abc', None, None, True]) == {
30+
'L': [{'N': '1'}, {'S': 'abc'}, {'NULL': True}, {'NULL': True},
31+
{'BOOL': True}]
3632
}
3733

38-
3934
def test_maps():
40-
assert serializer.serialize({"hello": "world", "another": 123, "loop": {"Working": True}}) == {
41-
"M": {
42-
"hello": {"S": "world"},
43-
"another": {"N": "123"},
44-
"loop": {"M": {"Working": {"BOOL": True}}},
35+
assert serializer.serialize({
36+
'hello': 'world',
37+
'another': 123,
38+
'loop': {'Working': True}
39+
}) == {
40+
'M': {
41+
'hello': {'S': 'world'},
42+
'another': {'N': '123'},
43+
'loop': {'M': {'Working': {'BOOL': True}}}
4544
}
4645
}
4746

48-
4947
def test_item():
50-
assert serializer.serialize_item(
51-
{"Hello": "World", "another": 123, "loop": {"Working": True}}
52-
) == {
53-
"Hello": {"S": "World"},
54-
"another": {"N": "123"},
55-
"loop": {"M": {"Working": {"BOOL": True}}},
48+
assert serializer.serialize_item({
49+
'Hello': 'World',
50+
'another': 123,
51+
'loop': {'Working': True}
52+
}) == {
53+
'Hello': {'S': 'World'},
54+
'another': {'N': '123'},
55+
'loop': {'M': {'Working': {'BOOL': True}}}
5656
}

0 commit comments

Comments
 (0)