Skip to content

Commit d1907ca

Browse files
author
Mike Shultz
authored
Merge pull request #9 from ApeWorX/feat/region-override
feat: adds optional sqs_region_override arg to broker
2 parents dec6aa7 + e9b3594 commit d1907ca

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

taskiq_sqs/broker.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,17 @@ def __init__(
3636
max_number_of_messages: int = 1, # size of batch to receive from the queue
3737
result_backend: Optional[AsyncResultBackend] = None,
3838
task_id_generator: Optional[Callable[[], str]] = None,
39+
sqs_region_override: str | None = None,
3940
) -> None:
4041
super().__init__(result_backend, task_id_generator)
4142

4243
if not sqs_queue_url or not sqs_queue_url.startswith("http"):
4344
raise BrokerError("A valid SQS Queue URL is required")
4445

4546
self.sqs_queue_url = sqs_queue_url
46-
self._sqs: SQSServiceResource = boto3.resource("sqs")
47+
self._sqs: SQSServiceResource = boto3.resource(
48+
"sqs", region_name=sqs_region_override
49+
)
4750
self._sqs_queue: Optional[Queue] = None
4851

4952
if max_number_of_messages > 10:

0 commit comments

Comments
 (0)