Skip to content

Commit 46cb156

Browse files
committed
MESH-2092 Run ruff
1 parent f07c782 commit 46cb156

File tree

2 files changed

+7
-13
lines changed

2 files changed

+7
-13
lines changed

nhs_aws_helpers/__init__.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -837,10 +837,7 @@ def _should_retry(self, err: ClientError, retries: int):
837837
if reasons:
838838
return True
839839

840-
if err.response["Error"]["Code"] in _RETRY_EXCEPTIONS:
841-
return True
842-
843-
return False
840+
return err.response["Error"]["Code"] in _RETRY_EXCEPTIONS
844841

845842
def _sleep_for(self, retries: int):
846843
backoff = pow(2, retries) * self._backoff

nhs_aws_helpers/dynamodb_model_store/base_model_store.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def deserialise_value(cls, value_type: type, value: Any, **kwargs) -> Any: # no
140140

141141
if value and value_type in (bytes, bytearray) and hasattr(value, "value"):
142142
value = value.value
143-
if value_type == bytearray:
143+
if isinstance(value_type, bytearray):
144144
value = bytearray(value)
145145

146146
if value_type in (str, bool, bytes, bytearray):
@@ -160,18 +160,18 @@ def deserialise_value(cls, value_type: type, value: Any, **kwargs) -> Any: # no
160160

161161
origin_type = get_origin(value_type)
162162

163-
if origin_type == list:
163+
if isinstance(origin_type, list):
164164
item_type = get_args(value_type)[0]
165165
return [cls.deserialise_value(item_type, val, **kwargs) for val in value]
166166

167-
if origin_type == dict:
167+
if isinstance(origin_type, dict):
168168
val_type = get_args(value_type)[1]
169169
return {key: cls.deserialise_value(val_type, val, **kwargs) for key, val in value.items()}
170170

171-
if origin_type == frozenset:
171+
if isinstance(origin_type, frozenset):
172172
return frozenset(val for val in value)
173173

174-
if origin_type == set:
174+
if isinstance(origin_type, set):
175175
return set(value)
176176

177177
return value
@@ -413,10 +413,7 @@ async def item_exists(self, key: TModelKey, consistent_read: bool = False, **kwa
413413
item = await self.get_item(
414414
key, ConsistentRead=consistent_read, ProjectionExpression=self._partition_key, **kwargs
415415
)
416-
if not item or self._partition_key not in item:
417-
return False
418-
419-
return True
416+
return item and self._partition_key in item
420417

421418
@dynamodb_retry_backoff()
422419
async def delete_item(self, key: TModelKey, **kwargs):

0 commit comments

Comments
 (0)