rename params
This commit is contained in:
@@ -27,30 +27,31 @@ serializer = TypeSerializer()
|
||||
deserializer = TypeDeserializer()
|
||||
|
||||
|
||||
def _serialize_python_type(value: Any) -> str | dict | list:
|
||||
match value:
|
||||
def _serialize_to_primitive_types(data: Any) -> str | dict | list:
|
||||
match data:
|
||||
case datetime():
|
||||
return value.isoformat()
|
||||
return data.isoformat()
|
||||
case UUID():
|
||||
return str(value)
|
||||
return str(data)
|
||||
case IPv4Address():
|
||||
return str(value)
|
||||
return str(data)
|
||||
case list() | tuple():
|
||||
return [_serialize_python_type(v) for v in value]
|
||||
return [_serialize_to_primitive_types(v) for v in data]
|
||||
case dict():
|
||||
return {k: _serialize_python_type(v) for k, v in value.items()}
|
||||
return {k: _serialize_to_primitive_types(v) for k, v in data.items()}
|
||||
case _:
|
||||
return value
|
||||
return data
|
||||
|
||||
|
||||
def serialize(value: dict) -> dict:
|
||||
def serialize(data: dict) -> dict:
|
||||
return {
|
||||
k: serializer.serialize(_serialize_python_type(v)) for k, v in value.items()
|
||||
k: serializer.serialize(_serialize_to_primitive_types(v))
|
||||
for k, v in data.items()
|
||||
}
|
||||
|
||||
|
||||
def deserialize(value: dict) -> dict:
|
||||
return {k: deserializer.deserialize(v) for k, v in value.items()}
|
||||
def deserialize(data: dict) -> dict:
|
||||
return {k: deserializer.deserialize(v) for k, v in data.items()}
|
||||
|
||||
|
||||
if TYPE_CHECKING:
|
||||
@@ -555,16 +556,16 @@ class DynamoDBCollection:
|
||||
self,
|
||||
persistence_layer: DynamoDBPersistenceLayer,
|
||||
/,
|
||||
exc_cls: Type[ValueError] = MissingError,
|
||||
exception_cls: Type[ValueError] = MissingError,
|
||||
tz: str = TZ,
|
||||
) -> None:
|
||||
if not issubclass(exc_cls, ValueError):
|
||||
if not issubclass(exception_cls, ValueError):
|
||||
raise TypeError(
|
||||
f'exception_cls must be a subclass of ValueError, got {exc_cls}'
|
||||
f'exception_cls must be a subclass of ValueError, got {exception_cls}'
|
||||
)
|
||||
|
||||
self.persistence_layer = persistence_layer
|
||||
self.exc_cls = exc_cls
|
||||
self.exception_cls = exception_cls
|
||||
self.tz = tz
|
||||
|
||||
def get_item(
|
||||
@@ -575,7 +576,7 @@ class DynamoDBCollection:
|
||||
default: Any = None,
|
||||
delimiter: str = '#',
|
||||
) -> Any:
|
||||
exc_cls = self.exc_cls
|
||||
exc_cls = self.exception_cls
|
||||
data = self.persistence_layer.get_item(key)
|
||||
|
||||
if raise_on_error and not data:
|
||||
|
||||
Reference in New Issue
Block a user