Vh_ddlmZddlmZddlmZddlmZddlmZddlmZddlm Z  ddl m Z dd l m Z dd l m Z dd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddl#m$Z$edgdZ%gdZ&gdZ'gdZ(gdZ)GddeZ*Gd d!eZ+e+jYd"gejZd#ed$eee.effd%Z/e+jYd&gejZd#ed$eee.effd'Z0e+jYd(gejZd#ed$eefd)Z1e+jYd*gejZd+e.d$eee.e.ffd,Z2d-e.d.ed$e%fd/Z3d-e.d.ed$e.fd0Z4d.ed1ed-e.d2ee.efd$e5f d3Z6d.ed-e.d2ee.efd$e ee5ffd4Z7d.ed5e.d6e.d$d7fd8Z8d.ed9e.d6e.d$d7fd:Z9d.ed;e.d6e.d$d7fd<Z:d.ed;e.d6e.d$d7fd=Z;d.ed>e.d-e.d$d7fd?ZdEee.efd$ee.effdFZ?d.ed2ee.efd-e.dGe5d$ee.eff dHZ@d.ed+e.dIee.e.fdJeee.e.fdKe5d$e5f dLZAdMeee.e.fdNeee.e.fdOe5d$e eee.e.feee.e.fffdPZBd.edQe.dReee.e.fdSeee.e.fd$e5f dTZCejZd.edUedVee.d$eee.effdWZDejZ d]dUed.edXe.d$eeEfdYZFejZ d^d.edUedVe.dZe.d$eee.eff d[ZGy7#e$rYwxYw)_) namedtuple)sleep)Any)Dict)List)Optional)Tuple) BotoCoreError) ClientError) WaiterError)to_text)camel_dict_to_snake_dict)snake_dict_to_camel_dict)"get_boto3_client_method_parametersis_boto3_error_code)AnsibleAWSModule)AWSErrorHandler)AnsibleAWSError)AWSRetry)ansible_dict_to_boto3_tag_list)boto3_tag_list_to_ansible_dict)compare_aws_tags) get_waiterBoto3ClientMethodnamewaiteroperation_descriptionresource retry_codes) create_db_cluster restore_db_cluster_from_snapshotrestore_db_cluster_from_s3#restore_db_cluster_to_point_in_timemodify_db_clusterdelete_db_clusteradd_tags_to_resourceremove_tags_from_resourcelist_tags_for_resourcepromote_read_replica_db_clusterstop_db_clusterstart_db_cluster)create_db_instance$restore_db_instance_to_point_in_timerestore_db_instance_from_s3$restore_db_instance_from_db_snapshotcreate_db_instance_read_replicamodify_db_instancedelete_db_instancer(r)r*promote_read_replicastop_db_instancestart_db_instancereboot_db_instanceadd_role_to_db_instanceremove_role_from_db_instance)create_db_cluster_snapshotdelete_db_cluster_snapshotr(r)r*copy_db_cluster_snapshot)create_db_snapshotdelete_db_snapshotr(r)copy_db_snapshotr*c eZdZy)AnsibleRDSErrorN)__name__ __module__ __qualname__g/home/dcms/DCMS/lib/python3.12/site-packages/ansible_collections/amazon/aws/plugins/module_utils/rds.pyrBrB\srGrBc eZdZeZedZy)RDSErrorHandlerctgdS)N)DBInstanceNotFoundDBSnapshotNotFoundDBClusterNotFoundDBClusterSnapshotNotFoundFaultr)clss rH _is_missingzRDSErrorHandler._is_missingcs" o  rGN)rCrDrErB_CUSTOM_EXCEPTION classmethodrQrFrGrHrJrJ`s'  rGrJzdescribe db cluster snapshotsparamsreturnc j|jd}|jdi|jdS)Ndescribe_db_cluster_snapshotsDBClusterSnapshotsrF get_paginatorpaginatebuild_full_resultclientrT paginators rHrWrWjs:$$%DEI 9   ' ' 9 9 ;db_snapshot_availablewaitmethod zY hasn't been added to the list of accepted methods to use a waiter in module_utils/rds.pyr) replacecluster_method_namesrTinstance_method_namescluster_snapshot_method_namesinstance_snapshot_method_namesgetNotImplementedErrorr)rlrmr readable_opr r!s rHget_rds_method_attributers(F%%c3/77dCKHK**/Jfmm/[ - -&F(F < <:;K23K - -2NRXR_R_2_ . .*F . .*F 5 5&F : :)F 2 2,F 2 2+F,F @ @34K35RSK 5 5:Z^d^k^k:k% 6 62F:;K 8 84F23K5F:;K 6 6;SW]WdWd;d& . .*F34K 0 0,F34K-F34K ==  V $%+&A   {U]kv rGcd}|jjd}t||j}|dk(r|jd}|jd}n]|dk(r|jd}|jd}n9|d k(r|jd }n$|d k(r|jd }nt d |d|j s|r|r|}|S)al Returns the final identifier for the resource to which the specified method applies. Parameters: method_name (str): RDS method whose target resource final identifier is returned module: AnsibleAWSModule Returns: updated_identifier (str): The new resource identifier from module params if not in check mode, there is a new identifier in module params, and apply_immediately is True; otherwise returns the original resource identifier from module params Raises: NotImplementedError if the provided method is not supported Napply_immediatelyrudb_cluster_identifierrtr{db_instance_identifierrzrrrrrzI hasn't been added to the list of accepted methods in module_utils/rds.py)rTrrr r check_mode)rlrmupdated_identifierrr identifiers rHget_final_identifierrs ))*=>' V<EEH9]]#:; #]]+FG Z ]]#;< #]]+GH ( (]]#;< ' ']]#CD !k]"k l     !38I' rG exception parametersct|ts|j|d|d|d}|jdd}|dvrd|dk(r_d t |vrd }|Sd t |vr|j|d |S|j|d t ||j |S|dk(rB|dk(r=dt |vrd }|S|j|d t ||j |S|dk(rB|dk(r=dt |vrd }|S|j|d t ||j |S|dk(rr|dk(rmgd}|jd|vr*|j|d|jdd||S|j|d t ||j |S|j|d t ||j |S)a} Fails the module with an appropriate error message given the provided exception. Parameters: module: AnsibleAWSModule exception: Botocore exception to be handled method_name (str): Name of boto3 rds client method parameters (dict): Parameters provided to boto3 client method Returns: changed (bool): False if provided exception indicates that no modifications were requested or a read replica promotion was attempted on an instance/cluseter that is not a read replica; should never return True (the module should always fail instead) zUnexpected failure for method z with parameters msgTErrorCode)r3r&InvalidParameterCombinationzNo modifications were requestedFzModifyDbCluster APIzlIt appears you are trying to modify attributes that are managed at the cluster level. Please see rds_clusterz Unable to r5rz!DB Instance is not a read replicar+InvalidDBClusterStateFaultz%DB Cluster that is not a read replicar"InvalidParameterValue)auroraz aurora-mysqlzaurora-postgresqlmysqlpostgresEnginez DB engine z should be one of ) isinstancer fail_json_awsresponser rrr)rmrrlrchanged error_codeaccepted_enginess rH handle_errorsrs i -Y.L[MYjkujv,wxG##G,V4JAAjTqFq , 0B BGX NW#gi&8 8  C ! T NK   !9+v!N!d!d ef ! J NC . .:AY3Y .')2D DG> N;   !9+v!N!d!d ef ! : N3 9 9jLh>h 2gi6H HG. N+   !9+v!N!d!d ef ! * N# + + >U0U_ >>( #+; ;  JNN8,D+EEWXhWij !  N   !9+v!N!d!d ef !  N  5k6J``ab  NrGczi}d}|js~|jjd}t||j}t ||} t j||di|}|r|rt||} t||| |||fS#ttf$r} t|| ||}Yd} ~ Dd} ~ wwxYw)a7Calls the provided boto3 rds client method with the provided parameters. Handles check mode determination, whether or not to wait for resource status, and method-specific retry codes. Parameters: client: boto3 rds client module: Ansible AWS module method_name (str): Name of the boto3 rds client method to call parameters (dict): Parameters to pass to the boto3 client method; these must already match expected parameters for the method and be formatted correctly (CamelCase, Tags and other attributes converted to lists of dicts as needed) Returns: tuple (any, bool): result (any): Result value from method call changed (bool): True if changes were made to the resource, False otherwise Tr)catch_extra_error_codesNrF) rrTrrr!getattrrjittered_backoffr r rrwait_for_status) r^rmrlrresultrrr!methoders rH call_methodrIs"FG   }}  (.{FCOO - H[SX..{STZ[i^hiF G-k6BJ FFJ D 7? {+ H#FA{JGG Hs"BB:"B55B:db_instance_id waiter_nameNc d}ddd}|j|d}tddD]} ||||y y #t$ra}|jjdijd d k(rt dYd }~Y|j |d |d |Yd }~wd }~wt tf$r#}|j |d|d |Yd }~d }~wwxYw)a Waits until provided instance has reached the expected status for provided waiter. Fails the module if an exception is raised while waiting. Parameters: client: boto3 rds client module: AnsibleAWSModule db_instance_id (str): DB instance identifier waiter_name (str): Name of either a boto3 rds client waiter or an RDS waiter defined in module_utils/waiters.py c |j|}|jddd|y#t$rt||}Y.wxYw)N<)Delay MaxAttempts) WaiterConfigDBInstanceIdentifier)r ValueErrorr)r^rrrs rHrz&wait_for_instance_status..waitxsL 5&&{3F  2b!AXf g 5 4F 5s*AAdeletedstopped)r|r} availabler rrrLNz$Error while waiting for DB instance z to be rz/Unexpected error while waiting for DB instance )rranger last_responserrr r ) r^rmrrrwaiter_expected_statusexpected_status_wait_attemptsrs rHwait_for_instance_statusrksh )(-00kJO2,    5   y""7B/33F;?SSb   *N~N^^efuev(w x x{+   HHXX_`o_pq !   s' 9 C9BBC0CC db_cluster_idc t||j|y#t$r/}|dk(rd|d}nd|d}|j||Yd}~yd}~wtt f$r }|j|d|Yd}~yd}~wwxYw) a Waits until provided cluster has reached the expected status for provided waiter. Fails the module if an exception is raised while waiting. Parameters: client: boto3 rds client module: AnsibleAWSModule db_cluster_id (str): DB cluster identifier waiter_name (str): Name of either a boto3 rds client waiter or an RDS waiter defined in module_utils/waiters.py )DBClusterIdentifierrvzFailed to wait for DB cluster  to be deleted to be availablerNzAFailed with an unexpected error while waiting for the DB cluster rrr rr r )r^rmrrrrs rHwait_for_cluster_statusrs y6;',,,O ) + +2=/PC2=/AQRCQC(( ; 'yQ&ghugv$wxxys  B%AB$A??Bdb_snapshot_idc |j|j|y#t$r/}|dk(rd|d}nd|d}|j||Yd}~yd}~wtt f$r }|j|d|Yd}~yd}~wwxYw) a Waits until provided instance snapshot has reached the expected status for provided waiter. Fails the module if an exception is raised while waiting. Parameters: client: boto3 rds client module: AnsibleAWSModule db_snapshot_id (str): DB instance snapshot identifier waiter_name (str): Name of a boto3 rds client waiter DBSnapshotIdentifierrzFailed to wait for DB snapshot rrrNzBFailed with an unexpected error while waiting for the DB snapshot rr^rmrrrrs rH!wait_for_instance_snapshot_statusrs  +&+++P ) / /3N3C>RC3N3CCSTCQC(( ; '  WXfWgh    !$ B %AB )BB c |j|j|y#t$r/}|dk(rd|d}nd|d}|j||Yd}~yd}~wtt f$r }|j|d|Yd}~yd}~wwxYw) a Waits until provided cluster snapshot has reached the expected status for provided waiter. Fails the module if an exception is raised while waiting. Parameters: client: boto3 rds client module: AnsibleAWSModule db_snapshot_id (str): DB cluster snapshot identifier waiter_name (str): Name of a boto3 rds client waiter DBClusterSnapshotIdentifierrz'Failed to wait for DB cluster snapshot rrrNzJFailed with an unexpected error while waiting for the DB cluster snapshot rrs rH wait_for_cluster_snapshot_statusrs  +&+++W ) 7 7;N;K>ZC;N;KK[\CQC(( ; '  \]k\lm    rrct||}|j}|j}|dk(rt||||y|dk(rt ||||y|dk(rt ||||y|dk(rt ||||yy)aP Waits until provided resource has reached the expected final status for provided method. Parameters: client: boto3 rds client module: AnsibleAWSModule identifier (str): resource identifier method_name (str): Name of boto3 rds client method on whose final status to wait rur{rrN)rrr rrrr)r^rmrrlrds_method_attributesrr s rHrrs5[&I'..K$--H9 KH Z  [I ( ()&&*kR ' '([Q (rGsnapshot_identifier snapshot_type convert_tagscd}||vrtd|i}|dk(rt||}n|dk(r t||}r|d}|r |rt|j dd |d <|S) a Returns instance or cluster snapshot attributes given the snapshot identifier. Parameters: client: boto3 rds client snapshot_identifier (str): Unique snapshot identifier snapshot_type (str): Which type of snapshot to get, one of: cluster, instance convert_tags (bool): Whether to convert the snapshot tags from boto3 list of dicts to Ansible dict; defaults to True Returns: snapshot (dict): Snapshot attributes. If snapshot with provided id is not found, returns an empty dict Raises: ValueError if an invalid snapshot_type is passed )rur{z(Invalid snapshot_type. Expected one of: rurr{rrrkNTags)rrWrerpop)r^rrr valid_typessnapshot snapshotss rH get_snapshotrs *KK'CK=QRRH !1&Vij * $)&GZ[ Q<L9(,,yRV:WX OrGc t||}t|S#t$r*}|j|d|Yd}~tSd}~wwxYw)a Returns tags for provided RDS resource, formatted as an Ansible dict. Fails the module if an error is raised while retrieving resource tags. Parameters: client: boto3 rds client module: AnsibleAWSModule resource_arn (str): AWS resource ARN Returns: tags (dict): Tags for resource, formatted as an Ansible dict. An empty list is returned if the resource has no tags. z!Unable to list tags for resource rN)r*rBrr)r^rmrgtagsrs rHget_tagsrs^X%fl; *$ // XQ&G ~$VWW )$ //Xs  A AA  options_dictc2|jd}d}d|vrd}|jd}t|d}d}t|jD]5}|D].\}}||vs |j|||j ||<07||d<|r|d<|S) a Converts snake_cased rds module options to CamelCased parameter formats expected by boto3 rds client. Does not alter case for keys or values in the following attributes: tags, processor_features. Includes special handling of certain boto3 params that do not follow standard CamelCase. Parameters: options_dict (dict): Snake-cased options for a boto3 rds client method Returns: camel_options (dct): Options formatted for boto3 rds client rFprocessor_featuresT)capitalize_first))Dbrs)IamIAM)AzAZ)CaCA)PerformanceInsightsKmsKeyIdPerformanceInsightsKMSKeyIdrProcessorFeatures)rrlistkeysr) rrhas_processor_featuresr camel_optionsaws_replace_keyskeyoldnews rHarg_spec_to_rds_paramsr 1s   F #D"|+!%)--.BC,\DQMM&&()N( NHCcz7D7H7H7M ckk#s34 NN!M&-? )* rG format_tagscNt||d}tfd|Dr.t||j}|j d|d|t||t fdj D|r"jdrtdd<S) a Returns a dict of parameters validated and formatted for the provided boto3 client method. Performs the following parameters checks and updates: - Converts parameters supplied as snake_cased module options to CamelCase - Ensures that all required parameters for the provided method are present - Ensures that only parameters allowed for the provided method are present, removing any that are not relevant - Removes parameters with None values - If format_tags is True, converts "Tags" param from an Ansible dict to boto3 list of dicts Parameters: client: boto3 rds client module: AnsibleAWSModule parameters (dict): Parameter options as provided to module method_name (str): boto3 client method for which to validate parameters format_tags (bool): Whether to convert tags from an Ansible dict to boto3 list of dicts Returns: Dict of client parameters formatted for the provided method Raises: Fails the module if any parameters required by the provided method are not provided in module options T)requiredc3DK|]}j|duywN)r).0krs rH z6format_rds_client_method_parameters..ps ?:>>!  $ ?s zTo z requires the parameters: rc3<K|]\}}|vs |||fywrrF)rrvoptionss rHrz6format_rds_client_method_parameters..ts%^AqG|PQP]q!f^s  r) ranyrr fail_jsondictitemsrr)r^rmrrlr required_optionsmethod_descriptionrs ` @rH#format_rds_client_method_parametersrUs4:&+X\] ?.> ??5k6J``s#5"66PQaPbcd0EG^)9)9);^^Jz~~f-;Jv\]h>ij    3(4P  NrGexisting_roles target_roles purge_rolesc|Dcgc]"}td|jD$}}|Dcgc] }||vs| }}|r|Dcgc] }||vs| c}ng}||fScc}wcc}wcc}w)a Returns differences between target and existing IAM roles. Parameters: existing_roles (list): Existing IAM roles as a list of snake-cased dicts target_roles (list): Target IAM roles as a list of snake-cased dicts purge_roles (bool): Remove roles not in target_roles if True Returns: roles_to_add (list): List of IAM roles to add roles_to_delete (list): List of IAM roles to delete c36K|]\}}|dk7s ||fyw)statusNrF)rrrs rHrz$compare_iam_roles..sLdaa8mAq6Ls  )rr)r&r'r(role roles_to_addroles_to_removes rHcompare_iam_rolesr/sxZhhQUdLdjjlLLhNh%1PTT5ODPLPWbS$l:RtShjO  ((iPSs'A A$A$ A)A) instance_idr-r.c|D] }||d|dd}t||d|\}}"|D] }||d|dd}t||d|\}}"S)a Update a DB instance's associated IAM roles Parameters: client: RDS client module: AnsibleAWSModule instance_id (str): DB's instance ID roles_to_add (list): List of IAM roles to add in snake-cased dict format roles_to_delete (list): List of IAM roles to delete in snake-cased dict format Returns: changed (bool): True if changes were successfully made to DB instance's IAM roles; False if not role_arn feature_name)rRoleArn FeatureNamer:r r9)r) r^rmr0r-r.r,rT_resultrs rHupdate_iam_rolesr7s( v*5$zBRcghvcwx&vvCantuvq*5$zBRcghvcwx&vvC\iopq NrG connection group_namecg} i}|||d<|jd}|jdi|jd}|S#td$rY|St$r}|j |dYd}~|Sd}~wwxYw)NDBClusterParameterGroupName$describe_db_cluster_parameter_groupsDBClusterParameterGroupsDBParameterGroupNotFoundz,Couldn't access parameter groups informationrrFrZr[r\rr r)rmr8r9rrTr_rs rHr<r<sF T  !4>F0 1,,-ST ###-f-??AB\] M 9 :   M TQ$RSS MTs=AA:A:A55A:db_parameter_group_namect |r|j|d}n|jd}|D](}|j|dd}t||d<*|r|Dcgc]}t|dgc}SgScc}w#t d$rgcYSt $r}|j |d Yd}~Sd}~wwxYw) N)DBParameterGroupNameDBParameterGroupsDBParameterGroupArnrirkr) ignore_listr>z+Couldn't access parameter group informationr)describe_db_parameter_groupsr*rrrr r)r8rmr@rparameter_grouprgrouprs rH%describe_db_instance_parameter_groupsrIsS "<?RSF & TO&==?[pKq=rM'E]&SOF #  T _eSYZ%(VHEZljllZ 9 : SQ$QRR MSs6AA? A:5A?8A?:A??B7B7B22B7sourcecg} |jd}d|i}|dk7r||d<|jd i|jd}|S#td$rY|St$r}|j |dYd}~|Sd}~wwxYw) Ndescribe_db_cluster_parametersr;allSource Parametersr>z2Couldn't access RDS cluster parameters informationrrFr?)rmr8r9rJrr_rTrs rHrLrLsF Z,,-MN /< U?%F8 ###-f-??A,O M 9 :   M ZQ$XYY MZsAAA?A?!A::A?)Tr)rM)H collectionsrtimertypingrrrrr botocore.exceptionsr r r ImportErroransible.module_utils._textr 0ansible.module_utils.common.dict_transformationsrransible_collections.amazon.aws.plugins.module_utils.exceptionsr;ansible_collections.amazon.aws.plugins.module_utils.retriesr;ansible_collections.amazon.aws.plugins.module_utils.taggingrrr;ansible_collections.amazon.aws.plugins.module_utils.waitersrrrrrrrBrJlist_error_handlerrstrrWrarer*rrr"rrrrrrrrrr rr%r/r7r<rrIrLrFrGrHr`s # 1///UUk\UVZPffXR_ &!" o  o ##$CRHRDRT$sCx.=QRIR ##$;R@KDKT$sCx.5IKAK ##$;R@KDKT$ZKAK ##$ * 4S> " *  * *Z)c3h()8HK $sCx.m  sM MM