fix: update example and fix issue in cases there is no messages

Author: danfimov

examples/example_broker.py

@@ -8,33 +8,51 @@
 
 import asyncio
 
-import boto3
 import dotenv
+from aiobotocore.session import get_session
 
 from taskiq_sqs import S3Bucket, S3ResultBackend, SQSBroker
 
 
 dotenv.load_dotenv()
 
 QUEUE_NAME = "my-queue"
-QUEUE_URL = f"http://localhost:4566/000000000000/{QUEUE_NAME}"
-
-
-boto3.client("sqs").create_queue(QueueName=QUEUE_NAME)
-
-broker = SQSBroker(QUEUE_URL, sqs_region_override="us-east-1").with_result_backend(
-    S3ResultBackend(bucket=S3Bucket(name="response-bucket"))
+ENDPOINT_URL = "http://localhost:4566"
+AWS_REGION = "us-east-1"
+
+
+broker = SQSBroker(
+    queue_name=QUEUE_NAME,
+    endpoint_url=ENDPOINT_URL,
+    aws_region_name=AWS_REGION,
+).with_result_backend(
+    S3ResultBackend(
+        bucket=S3Bucket(name="response-bucket"),
+        endpoint_url=ENDPOINT_URL,
+        aws_region_name=AWS_REGION,
+    ),
 )
 
 
 @broker.task()
 async def i_love_aws() -> None:
     """I hope my cloud bill doesn't get too high!"""
-    await asyncio.sleep(5.5)
+    await asyncio.sleep(2)
     print("Hello there!")
 
 
+async def ensure_queue_exists() -> None:
+    session = get_session()
+    async with session.create_client(
+        "sqs",
+        region_name=AWS_REGION,
+        endpoint_url=ENDPOINT_URL,
+    ) as sqs:
+        await sqs.create_queue(QueueName=QUEUE_NAME)
+
+
 async def main() -> None:
+    await ensure_queue_exists()
     await broker.startup()
     task = await i_love_aws.kiq()
     print(await task.wait_result())

src/taskiq_sqs/broker.py

@@ -216,7 +216,7 @@ async def listen(self) -> AsyncGenerator[AckableMessage, None]:
                 MaxNumberOfMessages=self._max_number_of_messages,
                 WaitTimeSeconds=self._wait_time_seconds,
             )
-            messages: list[MessageTypeDef] = results["Messages"]
+            messages: list[MessageTypeDef] = results.get("Messages", [])
 
             for message in messages:
                 if (body := message.get("Body")) and (receipt_handle := message.get("ReceiptHandle")):