P dZddlmZddlmZddlmZddlZddlmZddlmZddl m Z dd l m Z d Z e jjd e jj d iZe jjfd Ze jjfdZe jjfdZde jjej*dfdZy)z/Utility functions to call Security Posture API.)absolute_import)division)unicode_literalsN)apis)waiter)base) resourcessecurityposturev1alphav1c`tj|}tjt|SN) VERSION_MAPgetrGetMessagesModuleAPI_NAME release_track api_versions =lib/googlecloudsdk/api_lib/scc/remediation_intents/sps_api.pyrr$# .+  + 66c`tj|}tjt|Sr)rrrGetClientInstancerrs rrr)rrcltjj|dtj |S)aOperations to Resource used for `waiter.WaitFor`. Args: operation_id: The operation ID for which resource reference is required. release_track: The release track to use for the API version. Returns: The resource reference for the operation. z2securityposture.organizations.locations.operations) collectionr)r REGISTRYParseRelativeNamerr) operation_idrs rGetOperationsRefr .s4    - -E//-0 . rFiX)secondsct|}|j}|j}|rtj||}ntj |}tj ||||jdz} | S)aWaits for an operation to complete. Polls the Security Posture Operations service until the operation completes, fails, or max_wait_seconds elapses. Args: operation_ref: A Resource created by GetOperationRef describing the Operation. message: The message to display to the user while they wait. has_result: If True, the function will return the target of the operation when it completes. If False, nothing will be returned. release_track: The release track to use for the API version. max_wait: The time to wait for the operation to succeed before timing out. Returns: if has_result = True, a RemediationIntent entity. Otherwise, None. i) max_wait_ms)r*organizations_locations_remediationIntents"organizations_locations_operationsrCloudOperationPollerCloudOperationPollerNoResourcesWaitForr!) operation_refmessage has_resultrmax_waitclientresource_clientoperations_clientpollerresponses rWaitForOperationr2?sz2 ] +&EE/??  ( (*F  3 34E FF ^^ mW(2B2BT2I( /r)__doc__ __future__rrrdatetimegooglecloudsdk.api_lib.utilrrgooglecloudsdk.calliopergooglecloudsdk.corer r ReleaseTrackALPHAGArrrr timedeltar2rrr>s6&',.() Y$ %)$5$5$;$;7 %)$5$5$;$;7 261B1B1H1H(##)) X   , 'r