VhM ddlmZ ddlmZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z dd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddlmZddlmZddlmZddlmZddlmZddlmZdeefdZ d&dedeede eeeefde efdZ! d'dede eeefde ede eeeffdZ"Gd d!Z#Gd"d#Z$Gd$d%Z%y#e$rYwxYw)()deepcopy) WaiterError)Any)Dict)List)Optional)Tuple)camel_dict_to_snake_dict)AnsibleEC2Error)%create_transit_gateway_vpc_attachment)%delete_transit_gateway_vpc_attachment)describe_subnets)(describe_transit_gateway_vpc_attachments)ensure_ec2_tags)%modify_transit_gateway_vpc_attachment)AnsibleAWSModule)boto3_tag_list_to_ansible_dict)boto3_tag_specifications)!ansible_dict_to_boto3_filter_list)boto3_resource_to_ansible_dict) get_waiterreturnc gdS)N) availabledeletingfailedfailinginitiatingRequest modifyingpendingAcceptancepending rollingBackrejected rejectingr%r/home/dcms/DCMS/lib/python3.12/site-packages/ansible_collections/amazon/aws/plugins/module_utils/transitgateway.py get_statesr($s   r&Nmodulesubnetssubnet_detailscR|sy| t|t|}|Dcgc]}|j d}}t t |dkDr&|jdtt |||dS#t$r}|j|Yd}~d}~wwxYwcc}w)N SubnetIdsVpcIdz>Attachment subnets may only be in one VPC, multiple VPCs found)msgvpcsr*r)rlistr fail_json_aws_errorgetlenset fail_json)clientr)r*r+esr2s r'subnets_to_vpcr<4s  *-fW NN%3 3qAEE'N 3D 3 3t9~Pc$i"  7N *  & &q ) ) * 4sA=B$= B!BB!filters attachment_idci}|r|g|d<n|rt||d< t|fi|}t dkDr t d|r|dSdS#t$r}|j|Yd}~Ad}~wwxYw)aVFind an existing transit gateway attachment based on filters or attachment ID. Args: client: The AWS client used to interact with the EC2 service. module: The Ansible module instance used for error handling. filters (Optional[Dict[str, Any]]): A dictionary of filters to apply when searching for attachments. attachment_id (Optional[str]): The ID of a specific attachment to find. Returns: Optional[Dict[str, Any]]: The found attachment details or None if not found. Raises: ValueError: If multiple attachments match the criteria. TransitGatewayAttachmentIdsFiltersNr0z3Multiple matching attachments found, provide an ID.r)rrr r4r6 ValueError)r9r)r=r>params attachmentsr:s r'find_existing_attachmentrEKs$F1>,- =gFy&>vPP  ;!NOO(;q>2d2 &""1%%&s A A/A**A/cveZdZdededdfdZedeeeffdZ deeefdefdZ de fd Z d eddfd Z y) $TransitGatewayAttachmentStateManagerr)r>rNc.||_||_||_yN)r9r)r>)selfr9r)r>s r'__init__z-TransitGatewayAttachmentStateManager.__init__ps  *r&ci}td|jjjd}|jjjd|z}t ||}||d<|S)N wait_timeout)Delay MaxAttempts WaiterConfig)minr)rCr5dict)rJrCdelay max_attemptsconfigs r' waiter_configz2TransitGatewayAttachmentStateManager.waiter_configusb!#At{{))--n=>{{))--n=F E|<!'~ r&rCc |jd}|rt|dg|d< t|jfi|}d|_|dS#t$rd}YFwxYw#t $r%}|j j|Yd}~Id}~wwxYw)a Create a new transit gateway attachment. Args: params (Dict[str, Any]): A dictionary containing the parameters needed to create the transit gateway attachment. Returns: str: The ID of the newly created transit gateway attachment. Raises: AnsibleEC2Error: If there is an error while creating the VPC attachment, it will fail the module and provide an error message. TagsNtransit-gateway-attachment)typesTagSpecificationsTransitGatewayAttachmentId) popKeyErrorrr r9r r)r4r>)rJrCtagsresponser:s r'create_attachmentz6TransitGatewayAttachmentStateManager.create_attachments ::f%D *B4PlOm*nF& ' /r) check_moder r9r r4)rJr:s r'delete_attachmentz6TransitGatewayAttachmentStateManager.delete_attachmentse!!{{%% 35dkk4CUCUV# 3 //22 3s A A5A00A5 desired_statecd|jgi}|j|jt|jd|} |j di|y#t $r&}|jj|dYd}~yd}~wwxYw)Nr@transit_gateway_vpc_attachment_z Timeout waiting for State changer%) r>updaterWrr9waitrr) fail_json_aws)rJrfrCwaiterr:s r'wait_for_state_changez:TransitGatewayAttachmentStateManager.wait_for_state_changes/$2D2D1EF d(()DKK+J=/)Z[ M FKK !& ! M KK % %a)K L L MsA B!BB)__name__ __module__ __qualname__rstrrKpropertyrrrWrbboolrermr%r&r'rGrGos+'7++PT+ tCH~6S#X636@ 4  M3 M4 Mr&rGc:eZdZdededeeefddfdZedeeeffdZ ede efdZ ede efd Z dd e e ed eddfd Zd ZdZdZdefdZdefdZdZddZdede edefdZd de ededefdZdeeefdeeeffdZy)!AttachmentConfigurationManagerr)r>existingrNcn||_||_||_|xsi|_i|_g|_g|_yrI)r9r)r>rv_resource_updates_subnets_to_add_subnets_to_remove)rJr9r)r>rvs r'rKz'AttachmentConfigurationManager.__init__s;  * B !#!"$r&c|jSrI)rxrJs r'resource_updatesz/AttachmentConfigurationManager.resource_updatess%%%r&c|jSrI)ryr|s r'subnets_to_addz-AttachmentConfigurationManager.subnets_to_adds###r&c|jSrI)rzr|s r'subnets_to_removez0AttachmentConfigurationManager.subnets_to_removes&&&r&r*purgec|yt|jjdg}t|}|s|j|} t |j t |}t|j |j|}|jd|d|Dcgc]}|jd} }t| tt| k7r|jjd| | t |j||_t |j||_|jdt |y#t$r&}|jj|Yd}~d}~wwxYwcc}w) aQ Set or update the subnets associated with the transit gateway attachment. Args: subnets (Optional[List[str]]): A list of subnet IDs to associate with the attachment. purge (bool): If True, the existing subnets will be replaced with the specified subnets. Nr.r-r/T) immutableAvailabilityZoneIdzZ   &D A5CCqquu)*CC s8s3s8} $ KK ! !R#&& "  $O$>$>$OP"&'A'A/'R"S   d?.CD! / KK + +A . . / Ds E"F FE>>Fc&|jd|S)N DnsSupport _set_optionrJvalues r'set_dns_supportz.AttachmentConfigurationManager.set_dns_supports e44r&c&|jd|S)N Ipv6Supportrrs r'set_ipv6_supportz/AttachmentConfigurationManager.set_ipv6_supports u55r&c&|jd|S)NApplianceModeSupportrrs r'set_appliance_mode_supportz9AttachmentConfigurationManager.set_appliance_mode_supports 6>>r&tgw_idc&|jd|S)NTransitGatewayIdr)rJrs r'set_transit_gatewayz2AttachmentConfigurationManager.set_transit_gateways''(:FCCr&rc&|jd|S)Nr/r)rJrs r'set_vpcz&AttachmentConfigurationManager.set_vpcs''88r&ct|jjdd}|r t|}ni||}|j d|y)NrY)rrvr5rr)rJr` purge_tags current_tags desired_tagss r'set_tagsz'AttachmentConfigurationManager.set_tags sI5dmm6G6GPT6UV #D>L3l3d3L   6r&cr|jj||}|jj||SrI)rvr5rx)rJkeydefault default_values r'_get_resource_valuez2AttachmentConfigurationManager._get_resource_values1 ))#w7 %%))#}==r&namerc|y|rdnd}t|jjdt}|j |j jdt|||<|j d|S)a Set a VPC attachment option to either enable or disable. Args: name (str): The name of the option to be updated. value (Optional[bool]): A boolean indicating whether to enable (True) or disable (False) the specified option. If None, no action is taken. Returns: bool: Returns True if the option was successfully set, or False if no update was made (because the value was None). FenabledisableOptions)rrvr5rSrirxr)rJrroptionss r'rz*AttachmentConfigurationManager._set_optionsq ="y4==,,Y?@t--11)TVDE '' 7;;r& descriptionrc|||j|k(ry|r1|jr%|xs|}|jj|d||j|<y)a Set a value for a resource attribute and track changes. Args: key (str): The attribute key to be updated. value (Any): The new value to set for the specified key. description (Optional[str], optional): A human-readable description of the resource attribute. immutable (bool, optional): A flag indicating whether the attribute is immutable. If True, and the resource exists, an error will be raised if attempting to change the value. Defaults to False. Returns: bool: Returns True if the value was successfully set, or False if no update was made. Fz" can not be updated after creation)r1T)rrvr)r8r})rJrrrrs r'rz2AttachmentConfigurationManager._set_resource_value1sa" =ET%=%=c%BB %,K KK ! ! 5W&X ! Y%*c"r&resourcechgd}|jDcic] \}}||vs ||c}}Scc}}w)a Filter out immutable resource attributes from the given resource dictionary. Args: resource (Dict[str, Any]): A dictionary representing the resource, which may contain various attributes, including both mutable and immutable ones. Returns: Dict[str, Any]: A new dictionary containing only the mutable attributes of the resource. )rr/ VpcOwnerIdStater. CreationTimerY)items)rJrimmutable_optionsrrs r'$filter_immutable_resource_attributeszCAttachmentConfigurationManager.filter_immutable_resource_attributesMs6v-5^^-=^zsEL]A]U ^^^s ..)NTrINF)rnrorprrqrrrKrrr}rrrrrsrrrrrrrrrrrr%r&r'rurusX%'7%%W[\_ad\dWe%jn%&$sCx.&&$S $$'49'''E8DI#6'Ed'EVZ'ER56?D#D9c97><rc||_||_||_|xsi|_i|_d|_t ||||_t|||||_ yr) r9r)r>rvupdatedchangedrG state_managerruconfig_manager)rJr9r)rvr>s r'rKz+TransitGatewayVpcAttachmentManager.__init__^sY  * B   A&&R_`rrr}rrr)rdrr9r r4)rJupdatesrrr:s r'apply_configurationz6TransitGatewayVpcAttachmentManager.apply_configuration|s!!%%JJ4K^K^KoKop,,;; //AA~6G &4GN # ):G% &040B0B,-{{%% 35dkkMWM# 3 //22 3s$B<< C*C%%C*c|jj|jjj d|jj |jjd|jjj dd|jj |jjj d|jj|jjj d|jj|jjj dy) z@Set configuration parameters for the transit gateway attachment.transit_gatewayr* purge_subnetsT dns_support ipv6_supportappliance_mode_supportN) rrr)rCr5rrrrr|s r'_set_configuration_parametersz@TransitGatewayVpcAttachmentManager._set_configuration_parameterss // 0B0B0F0FGX0YZ '' (:(:9(Et{{GYGYG]G]^mosGtu ++DKK,>,>,B,B=,QR ,,T[[-?-?-C-CN-ST 66t{{7I7I7M7MNf7ghr&cZ|jjjd}|jjjd}|jjjdr3d|jjdi}|d}n|j||}|i|fS||fS)zPrepare and return the tags and purge flag. Returns: Tuple[Optional[Dict[str, str]], bool]: A tuple containing the tags dictionary and the purge flag. r`rrNameF)r)rCr5ri)rJr`rnew_tagss r' _prepare_tagsz0TransitGatewayVpcAttachmentManager._prepare_tagss {{!!%%f-[[''++L9 ;;   ! !& ) 2 26 :;H|" %D\rZ77tZ77r&c"|jjsr|jd}|jj ||_|jj jdr|jjdd|_ y)z(Create a new transit gateway attachment.FrrjrTN) r)rdrrrbr>rCr5rmr)rJrCs r'_create_attachmentz5TransitGatewayVpcAttachmentManager._create_attachmentsn{{%%00%0HF!%!3!3!E!Ef!MD {{!!%%f-""88E r&r`rc H|jjddk(r|jjdnO|jjddk(r1|jj dt |jg|jrGd|_|jjjdr|jjd|xjt|j|j|jd || zc_y ) z.Update an existing transit gateway attachment.rr!rrz&Deletion in progress, unable to update)r1rDTrjrZ) resource_typer`rN) rvr5rrmr)r8rrrrCrr9r>rJr`rs r'_update_attachmentz5TransitGatewayVpcAttachmentManager._update_attachments ==  W % 2    4 4[ A ]]  w ': 5 KK ! !<;DMMJK "   # # %DL{{!!%%f-""88E  KK KK   6!   r&c|j|j\}}|jj|||js|j n|j |||jjr$t|jddg|_ ytt|j|j|j|_ y)zOCreate or modify a transit gateway attachment based on the provided parameters.FrrY) ignore_list)r>N)rrrrrvrrr)rdr rrrrEr9r>rs r'create_or_modify_attachmentz>TransitGatewayVpcAttachmentManager.create_or_modify_attachments **, --/j $$T:6}}  # # %  # #D* 5 ;; ! !3++U+CRXQYDL:(dkkQUQcQcdDLr&c|jjddk(rH|jjjdr|jj dd|_y|xj|jjzc_|jjjdr|jj dyy)zDelete attachmentrrrjdeletedFN) rvr5r)rCrrmchangerrer|s r'rez4TransitGatewayVpcAttachmentManager.delete_attachments ==  W % 3{{!!%%f-""88CDK LLD..@@B BL{{!!%%f-""88C.r&rI)T)rN)rnrorprrrqrrrKrsrrrr rrrrrer%r&r'rr]sim f. f:>sCx. fYabeYf f  fttCQTH~"Bi8uXd38n%=t%CD8&  tCH~ 4 D 66 Dr&rrI)NN)&copyrbotocore.exceptionsr ImportErrortypingrrrrr 0ansible.module_utils.common.dict_transformationsr 7ansible_collections.amazon.aws.plugins.module_utils.ec2r r r rrrr;ansible_collections.amazon.aws.plugins.module_utils.modulesr;ansible_collections.amazon.aws.plugins.module_utils.taggingrrBansible_collections.amazon.aws.plugins.module_utils.transformationrr;ansible_collections.amazon.aws.plugins.module_utils.waitersrrqr(r<rErGrurr%r&r'rs7  /USiiTlSiXf`pmR DI "lp$/3CyJRSWX\]`be]eXfSgJh c]0pt!3$!3/7S#X/G!3_ghk_l!3 d38n!3HHMHMV`_`_FeDeDg   sC((C0/C0