From c57568e23c1f8c042ad471ae852d79870f784254 Mon Sep 17 00:00:00 2001 From: Janusz Gregorczyk Date: Thu, 29 Aug 2024 21:23:01 +0200 Subject: [PATCH] chore: Improve types. --- sqs_workers/queue.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/sqs_workers/queue.py b/sqs_workers/queue.py index bc10d1f..d3f5590 100644 --- a/sqs_workers/queue.py +++ b/sqs_workers/queue.py @@ -98,11 +98,13 @@ def process_batch(self, wait_seconds: int = 0) -> BatchProcessingResult: wait_seconds, self.batching_policy.batch_size ) success = self.process_messages(messages) - messages_with_success = ((m, success) for m in messages) + messages_with_success: Iterable[tuple[Any, bool]] = ( + (m, success) for m in messages + ) else: messages = self.get_raw_messages(wait_seconds) - success = [self.process_message(message) for message in messages] - messages_with_success = zip(messages, success) + successes = [self.process_message(message) for message in messages] + messages_with_success = zip(messages, successes) return self._handle_processed(messages_with_success) @@ -202,7 +204,7 @@ def get_raw_messages(self, wait_seconds: int, max_messages: int = 10) -> List[An return queue.receive_messages(**kwargs) messages_left_on_queue = True - received_messages = [] + received_messages: list[Any] = [] # receive_messages will only return a maximum batch of 10 messages per call # if the client requests more than that we must poll multiple times