[ec2_vpc_net] Add retries to describe_vpc_attribute call (#39256)

* [ec2_vpc_net] Add retries to describe_vpc_attribute call

* Use new AnsibleAWSModule client-based waiters
This commit is contained in:
Sloane Hertel 2018-04-30 15:27:22 -04:00 committed by Ryan Brown
parent cc06f4cba1
commit ec9c59f52b
2 changed files with 19 additions and 10 deletions

View file

@ -195,6 +195,11 @@ class AnsibleAWSModule(object):
class _RetryingBotoClientWrapper(object):
__never_wait = (
'get_paginator', 'can_paginate',
'get_waiter', 'generate_presigned_url',
)
def __init__(self, client, retry):
self.client = client
self.retry = retry
@ -212,7 +217,9 @@ class _RetryingBotoClientWrapper(object):
def __getattr__(self, name):
unwrapped = getattr(self.client, name)
if callable(unwrapped):
if name in self.__never_wait:
return unwrapped
elif callable(unwrapped):
wrapped = self._create_optional_retry_wrapper_function(unwrapped)
setattr(self, name, wrapped)
return wrapped