Skip to content

Commit bf0871a

Browse files
author
Mike Shultz
authored
Merge pull request #5 from ApeWorX/fix/url-required
fix: require URL to broker
2 parents 4c0f411 + 8082165 commit bf0871a

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

taskiq_sqs/broker.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@ def __init__(
3131
task_id_generator: Optional[Callable[[], str]] = None,
3232
) -> None:
3333
super().__init__(result_backend, task_id_generator)
34+
35+
if not sqs_queue_url or not sqs_queue_url.startswith("http"):
36+
raise BrokerError("A valid SQS Queue URL is required")
37+
3438
self.sqs_queue_url = sqs_queue_url
3539
self._sqs = boto3.resource("sqs")
3640
self._sqs_queue: Optional[Queue] = None
@@ -45,9 +49,7 @@ async def _get_queue(self) -> Queue:
4549
queue_name = self.sqs_queue_url.split("/")[-1]
4650

4751
if not self._sqs_queue:
48-
self._sqs_queue = await asyncify(self._sqs.get_queue_by_name)(
49-
QueueName=queue_name
50-
)
52+
self._sqs_queue = await asyncify(self._sqs.get_queue_by_name)(QueueName=queue_name)
5153

5254
if not self._sqs_queue:
5355
raise Exception("SQS Queue not found")

0 commit comments

Comments
 (0)