From 173214593f617d19594abff71d530167c34dd3cf Mon Sep 17 00:00:00 2001 From: Jessie Nadler Date: Thu, 31 Oct 2019 11:16:11 -0400 Subject: [PATCH] Add way to disable using Trusted Advisor for usage --- awslimits/settings.py | 1 + awslimits/support.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/awslimits/settings.py b/awslimits/settings.py index 1131af4..97b6fdd 100644 --- a/awslimits/settings.py +++ b/awslimits/settings.py @@ -29,3 +29,4 @@ SNOOZE = os.environ.get("SNOOZE", '').replace('\'', '').replace('"', '') SUPPORT_REGION = os.environ.get("SUPPORT_REGION", "us-east-1") +USE_TRUSTED_ADVISOR_FOR_USAGE = int(os.environ.get("USE_TRUSTED_ADVISOR_FOR_USAGE", 1)) diff --git a/awslimits/support.py b/awslimits/support.py index a675811..d12f79d 100644 --- a/awslimits/support.py +++ b/awslimits/support.py @@ -227,7 +227,7 @@ def load_default_limits(): existing_limit_names = [limit['limit_name'] for limit in table.scan()['Items']] checker = get_aws_limit_checker() - checker.find_usage() + checker.find_usage(use_ta=settings.USE_TRUSTED_ADVISOR_FOR_USAGE) limits = checker.get_limits(use_ta=settings.PREMIUM_ACCOUNT)