99
1010import boto3
1111from asyncer import asyncify
12- from botocore .config import Config
1312from botocore .exceptions import ClientError
1413from taskiq import AsyncBroker
1514from taskiq .abc .result_backend import AsyncResultBackend
@@ -38,7 +37,6 @@ def __init__(
3837 result_backend : Optional [AsyncResultBackend ] = None ,
3938 task_id_generator : Optional [Callable [[], str ]] = None ,
4039 sqs_region_override : str | None = None ,
41- sqs_credentials_source_override : str | None = None ,
4240 ) -> None :
4341 super ().__init__ (result_backend , task_id_generator )
4442
@@ -47,10 +45,7 @@ def __init__(
4745
4846 self .sqs_queue_url = sqs_queue_url
4947 self ._sqs : SQSServiceResource = boto3 .resource (
50- "sqs" ,
51- config = Config (
52- region_name = sqs_region_override , credential_source = sqs_credentials_source_override
53- ),
48+ "sqs" , region_name = sqs_region_override
5449 )
5550 self ._sqs_queue : Optional [Queue ] = None
5651
@@ -64,7 +59,9 @@ async def _get_queue(self) -> Queue:
6459 queue_name = self .sqs_queue_url .split ("/" )[- 1 ]
6560
6661 if not self ._sqs_queue :
67- self ._sqs_queue = await asyncify (self ._sqs .get_queue_by_name )(QueueName = queue_name )
62+ self ._sqs_queue = await asyncify (self ._sqs .get_queue_by_name )(
63+ QueueName = queue_name
64+ )
6865
6966 if not self ._sqs_queue :
7067 raise Exception ("SQS Queue not found" )
0 commit comments