Allow SQS to use environment variables

This commit is contained in:
Sander van de Graaf
2018-01-09 13:03:51 +01:00
parent e22546672e
commit 17ea1e7a7b
2 changed files with 10 additions and 7 deletions
+6 -3
View File
@@ -51,9 +51,12 @@ class Sqs(Broker):
@staticmethod
def get_connection(list_key=Conf.PREFIX):
return Session(aws_access_key_id=Conf.SQS['aws_access_key_id'],
aws_secret_access_key=Conf.SQS['aws_secret_access_key'],
region_name=Conf.SQS['aws_region'])
config = Conf.SQS
if 'aws_region' in config:
config['region_name'] = config['aws_region']
del(config['aws_region'])
return Session(**config)
def get_queue(self):
self.sqs = self.connection.resource('sqs')