ZeroDownTime/streamlogs2fluentd
3
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

This time actually disable it

Browse Source
This commit is contained in:
Stefan Reimer 2019-06-28 15:36:23 +00:00
parent 71753107cc
commit 7a54650552
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
index.py
View File

@ -77,6 +77,7 @@ def get_source(region, account_id):
""" returns a new base source object """ returns a new base source object
resolves aws account_id to account alias and caches for lifetime of lambda function resolves aws account_id to account alias and caches for lifetime of lambda function
""" """
global RESOLVE_ACCOUNT
source = {'account': account_id, 'region': region} source = {'account': account_id, 'region': region}
if RESOLVE_ACCOUNT and not TEST: if RESOLVE_ACCOUNT and not TEST:
try: try:
@ -84,14 +85,11 @@ def get_source(region, account_id):
iam = boto3.client('iam') iam = boto3.client('iam')
account_aliases[account_id] = iam.list_account_aliases()['AccountAliases'][0] account_aliases[account_id] = iam.list_account_aliases()['AccountAliases'][0]
# if there is no alias disable further retries
if not account_aliases[account_id]:
RESOLVE_ACCOUNT = False
source['account_alias'] = account_aliases[account_id] source['account_alias'] = account_aliases[account_id]
except(KeyError, IndexError): except(KeyError, IndexError):
logger.warning("Could not resolve IAM account alias") logger.warning("Could not resolve IAM account alias")
RESOLVE_ACCOUNT = False
pass pass
return source return source