Vh`4ddlmZddlmZ ddlZddlmZddlm Z GddZ Gdd Z Gd d e Z y#e$rY3wxYw) deepcopywrapsN)camel_dict_to_snake_dict)boto3_tag_list_to_ansible_dictc:eZdZdZdZdZdZedZdZ dZ y)BaseWaiterFactoryag A helper class used for creating additional waiters. Unlike the waiters available directly from botocore these waiters will automatically retry on common (temporary) AWS failures. This class should be treated as an abstract class and subclassed before use. A subclass should: - create the necessary client to pass to BaseWaiterFactory.__init__ - override _BaseWaiterFactory._waiter_model_data to return the data defining the waiter Usage: waiter_factory = BaseWaiterFactory(module, client) waiter = waiters.get_waiter('my_waiter_name') waiter.wait(**params) Nc||_||_|j|j}tj j td||_y)N)versionwaiters) waiter_config) moduleclient_inject_ratelimit_retries_waiter_model_databotocorewaiter WaiterModeldict_model)selfrrdatas k/home/dcms/DCMS/lib/python3.12/site-packages/ansible_collections/community/aws/plugins/module_utils/base.py__init__zBaseWaiterFactory.__init__.sM  --d.E.EFoo11q$72  ctS)at Subclasses should override this method to return a dictionary mapping waiter names to the waiter definition. This data is similar to the data found in botocore's waiters-2.json files (for example: botocore/botocore/data/ec2/2016-11-15/waiters-2.json) with two differences: 1) Waiter names do not have transformations applied during lookup 2) Only the 'waiters' data is required, the data is assumed to be version 2 for example: @property def _waiter_model_data(self): return dict( tgw_attachment_deleted=dict( operation='DescribeTransitGatewayAttachments', delay=5, maxAttempts=120, acceptors=[ dict(state='retry', matcher='pathAll', expected='deleting', argument='TransitGatewayAttachments[].State'), dict(state='success', matcher='pathAll', expected='deleted', argument='TransitGatewayAttachments[].State'), dict(state='success', matcher='path', expected=True, argument='length(TransitGatewayAttachments[]) == `0`'), dict(state='success', matcher='error', expected='InvalidRouteTableID.NotFound'), ] ), ) or @property def _waiter_model_data(self): return { "instance_exists": { "delay": 5, "maxAttempts": 40, "operation": "DescribeInstances", "acceptors": [ { "matcher": "path", "expected": true, "argument": "length(Reservations[]) > `0`", "state": "success" }, { "matcher": "error", "expected": "InvalidInstanceID.NotFound", "state": "retry" } ] }, } rrs rrz$BaseWaiterFactory._waiter_model_data9s pv rcgd}g}|D]}|jtdd|!t|}|D]}||dj||S)N)RequestLimitExceeded UnavailableServiceUnavailableInternalFailure InternalErrorTooManyRequestsException Throttlingretryerror)statematcherexpected acceptors)appendrrextend)rmodel extra_retriesr.r*rrs rrz+BaseWaiterFactory._inject_ratelimit_retriesssp  " SE   T5Q R S% :F 6N; ' . .y 9 : rc|jj}||vr!|jjd|d|tj j ||j|jS)NzUnable to find waiter z. Available_waiters: )r waiter_namesr fail_jsonrrcreate_waiter_with_clientr)r waiter_namers r get_waiterzBaseWaiterFactory.get_waitersd++** g % KK ! !$:;-G]^e]f"g h88  KK KK  r) __name__ __module__ __qualname____doc__rrrpropertyrrr8rrr r s7"F F  77r* rr c*eZdZedZddZdZy) Boto3Mixincfd}|S)a A simple wrapper that handles the usual botocore exceptions and exits with module.fail_json_aws. Designed to be used with BaseResourceManager. Assumptions: 1) First argument (usually `self` of method being wrapped will have a 'module' attribute which is an AnsibleAWSModule 2) First argument of method being wrapped will have an _extra_error_output() method which takes no arguments and returns a dictionary of extra parameters to be returned in the event of a botocore exception. Parameters: description (string): In the event of a botocore exception the error message will be 'Failed to {DESCRIPTION}'. Example Usage: class ExampleClass(Boto3Mixin): def __init__(self, module) self.module = module self._get_client() @Boto3Mixin.aws_error_handler("connect to AWS") def _get_client(self): self.client = self.module.client('ec2') @Boto3Mixin.aws_error_handler("describe EC2 instances") def _do_something(**params): return self.client.describe_instances(**params) c2tfd}|S)Nc|j} |g|i|S#tjj$r,}|jj |fddi|Yd}~yd}~wtjj tjjf$r,}|jj |fddi|Yd}~yd}~wwxYw)NmsgzFailed waiting for z Failed to )_extra_error_outputr exceptions WaiterErrorr fail_json_aws ClientError BotoCoreError)_selfargskwargs extra_oupute descriptionfuncs rhandlerz>Boto3Mixin.aws_error_handler..wrapper..handlers#779 a7777**66j.ELL..qi8KK=6Yi]hi ++779L9L9Z9Z[a.ELL..q` ;-6P`T_`as C "A##:C "CC r)rQrRrPs` rwrapperz-Boto3Mixin.aws_error_handler..wrappers" 4[ a aNrr>)rPrSs` raws_error_handlerzBoto3Mixin.aws_error_handlers> rc||y|jdd}|r t|}n|s|i}t|}|||d<|S)a+ Performs common boto3 resource to Ansible resource conversion. `resource['Tags']` will by default be converted from the boto3 tag list format to a simple dictionary. Parameters: resource (dict): The boto3 style resource to convert to the normal Ansible format (snake_case). add_tags (bool): When `true`, if a resource does not have 'Tags' property the returned resource will have tags set to an empty dictionary. NTagstags)getrr)rresourceadd_tagsrWnormalized_resources r_normalize_boto3_resourcez$Boto3Mixin._normalize_boto3_resourcesY  ||FD) 1$7D )D6x@  *.  '""rctSNrr s rrEzBoto3Mixin._extra_error_outputs v rN)F)r9r:r; staticmethodrTr\rEr>rrr@r@s++Z#4rr@ceZdZfdZddZdZdZdZdZe dZ dZ d Z d Z d Zd Zd ZdZdZddZddZdZdZxZS)BaseResourceManagerc||_d|_t|_t|_t|_t|_d|_d|_tt|/y)zW Parameters: module (AnsibleAWSModule): An Ansible module. FTN) rchangedroriginal_resourceupdated_resource_resource_updates_preupdate_resource_wait _wait_timeoutsuperrar)rr __class__s rrzBaseResourceManager.__init__sX   !% $!%#'6  ! !413rct|j}|j|j|r|j |}|S)zz Merges the contents of the 'pre_update' resource and metadata variables with the pending updates )rrgupdaterf%_filter_immutable_resource_attributes)rfilter_immutablecreationrYs r_merge_resource_changesz+BaseResourceManager._merge_resource_changess> D445../ AA(KHrct|Sr^r)rrYs rrnz9BaseResourceManager._filter_immutable_resource_attributess !!rc yr^r>rparamss r_do_creation_waitz%BaseResourceManager._do_creation_wait rc yr^r>rts r_do_deletion_waitz%BaseResourceManager._do_deletion_waitrwrc yr^r>rts r_do_update_waitz#BaseResourceManager._do_update_waitrwrct}|jr7td|j}|j|z}t||}||d<|S)N)Delay MaxAttempts WaiterConfig)rrimin)rrudelay max_attemptsconfigs r_waiter_configz"BaseResourceManager._waiter_config sO   4--.E--6L<@F%+F> " rcZ|jsy|j}|jdi|yNr>)rhrryrts r_wait_for_deletionz&BaseResourceManager._wait_for_deletion*zz $$((rcZ|jsy|j}|jdi|yr)rhrrvrts r_wait_for_creationz&BaseResourceManager._wait_for_creationrrcZ|jsy|j}|jdi|yr)rhrr{rts r_wait_for_updatez$BaseResourceManager._wait_for_update!s*zz $$&v&rc&|jdS)z Merges all pending changes into self.updated_resource Used during check mode where it's not possible to get and refresh the resource F)ro)rqr s r_generate_updated_resourcez.BaseResourceManager._generate_updated_resource's ++U+CCrc4d}|jjsF|j}|j|j |j |_n$|j|j|_t|_ ||_ y)NT) r check_mode_do_create_resourcerrv get_resourcere_normalize_resourcerrrfrc)rrcs r _flush_createz!BaseResourceManager._flush_create1s{{{%%..0G  # # %  " " $$($5$5$7D !$($<$rra) copyr functoolsrr ImportError0ansible.module_utils.common.dict_transformationsr;ansible_collections.amazon.aws.plugins.module_utils.taggingrr r@rar>rrrs[ Vfw w tLL^\*\c  s>AA