Skip to content

Commit e3aef75

Browse files
committed
fix: raise error if URL is not provided
1 parent 4c0f411 commit e3aef75

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:
36+
raise BrokerError("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)