$dZddlmZddlmZddlmZddlmZddlmZddl mZ ddl m Z ddl m Z dZd Zd Zd Zd Zd ZdZdZdZdZdZdZ ddZdZ ddZdZdZdZy)z0Utilities for the cloud deploy rollout resource.) exceptions) client_util)release)rollout)log) resourceszBclouddeploy.projects.locations.deliveryPipelines.releases.rolloutszMapprovalState="NEEDS_APPROVAL" AND state="PENDING_APPROVAL" AND targetId="{}"zg(approvalState!="REJECTED" AND approvalState!="NEEDS_APPROVAL") AND state="SUCCEEDED" AND targetId="{}"z targetId="{}"z{}-to-{}-{:04d}z {}/releases/-zDeployEndTime desczCreateTime desczEnqueueTime desccLtjj|tS)zReturns a rollout reference object from a rollout message. Args: rollout_name: str, full canonical resource name of the rollout Returns: Rollout reference object  collection)rREGISTRYParseRelativeName_ROLLOUT_COLLECTION) rollout_names 5lib/googlecloudsdk/command_lib/deploy/rollout_util.pyRolloutReferenceFromNamer.s(    - -2 . cx|}d|vr3tjj|tj }|S)zqReturns rollout ID. Args: rollout_name_or_id: str, rollout full name or ID. Returns: Rollout ID. z projects/r )rr r rName)rollout_name_or_id rollout_ids r RolloutIdr<sE"*&&##55':6 df rctj|j}tj|j }t j j||tS)aLists the rollouts in PENDING_APPROVAL state for the releases associated with the specified target. The rollouts must be approvalState=NEEDS_APPROVAL and state=PENDING_APPROVAL. The returned list is sorted by rollout's create time. Args: target_ref: protorpc.messages.Message, target object. pipeline_ref: protorpc.messages.Message, pipeline object. Returns: a sorted list of rollouts. ) release_name filter_strorder_by) PENDING_APPROVAL_FILTER_TEMPLATEformatrWILDCARD_RELEASE_NAME_TEMPLATE RelativeNamer RolloutClientListPENDING_ROLLOUT_ORDERBY) target_ref pipeline_refrparents rListPendingRolloutsr&Ns^066z7HI* ) 0 01J1J1L M&    % %& & rNctj|j}tjj ||j|j |||S)a/Gets successfully deployed rollouts for the releases associated with the specified target and index. Args: target_ref: protorpc.messages.Message, target object. pipeline_ref: protorpc.messages.Message, pipeline object. filter_str: Filter string to use when listing rollouts. order_by: order_by field to use when listing rollouts. page_size: int, the maximum number of objects to return per page. limit: int, the maximum number of `Rollout` objects to return. Returns: a rollout object or None if no rollouts in the target. )rrr page_sizelimit)rrrrr r!r)r#r$rrr(r)r%s rGetFilteredRolloutsr*fs_ * 0 01J1J1L M&    % %"":??#45  & rc<tj|} tjj |j |}t |j||S#tj$r$tj|j wxYwN) !ROLLOUT_IN_TARGET_FILTER_TEMPLATErrr r!rComputeRolloutIDrapitools_exceptions HttpError cd_exceptionsListRolloutsError) to_target release_refrrolloutss rGenerateRolloutIdr6s077 B*F$$&++  "JH K,,. 8 DD  & &F  ) )+*B*B*D EEFs A A$$7Bc B|}|s t||}|j} tjj |t | j d| j d| j d|jd} tjj| j||} tjjdj| j|t!j"j%| | ||||} tjj'| j(d } tj*j-| | d t!j"j/| jS) aCreates a rollout by calling the rollout create API and waits for the operation to finish. Args: release_ref: protorpc.messages.Message, release resource object. to_target: str, the target to create create the rollout in. rollout_id: str, rollout ID. annotations: dict[str,str], a dict of annotation (key,value) pairs that allow clients to store small amounts of arbitrary data in cloud deploy resources. labels: dict[str,str], a dict of label (key,value) pairs that can be used to select cloud deploy resources and to find collections of cloud deploy resources that satisfy certain conditions. description: str, rollout description. starting_phase_id: str, rollout starting phase. override_deploy_policies: List of Deploy Policies to override. Raises: ListRolloutsError: an error occurred calling rollout list API. Returns: The rollout resource created. projectsId locationsIddeliveryPipelinesId)r8r9r: releasesId)r params)nametargetId descriptionz Creating rollout {} in target {}z)clouddeploy.projects.locations.operationsr z2Waiting for rollout creation operation to complete)r6AsDictrr ParsergetrrGetMessagesModuleRolloutrrstatusPrintrrr Creater r=OperationsClientWaitForOperationGet)r4r3r annotationslabelsr?starting_phase_idoverride_deploy_policiesfinal_rollout_id resource_dict rollout_ref rollout_obj operation operation_refs r CreateRolloutrUs@  (K@$$&-""(($%)),7&**=9!.!2!23H!I#((* ) +--/77  # # %8+ **(//  " " $i ##%,,  )$$66nn!L7- 11:    $ $[%=%=%? @@rct||ttdd}g}|D]C}|s|j|nt |s|j|t |dk\sB|S|S)aGets the currently deployed release and the next valid release that can be rolled back to. Args: target_ref: protorpc.messages.Message, target resource object. pipeline_ref: protorpc.messages.Message, pipeline resource object. Raises: HttpException: an error occurred fetching a resource. Returns: An list containg the currently deployed release and the next valid deployable release. N )r#r$rrr)r()r* DEPLOYED_ROLLOUT_FILTER_TEMPLATESUCCEED_ROLLOUT_ORDERBYappend_RolloutIsFromAbandonedReleaselen)r#r$iterabler5rRs rGetValidRollBackCandidater_so!1&  ((k ook" +K 8ook" 8} / /rc(t|j}|j} tjj |j }|jS#tj$r}tj|d}~wwxYwr,) rr=Parentr ReleaseClientrJrr/r0r HttpException abandoned)rRrQr4 release_objerrors rr\r\s~()9)9:+""$+*'')--k.F.F.HIK     & &*  " "5 ))*s1A$$B7B  Bc|Dchc]}t|j}}tddD]!}tj |||}||vs|cSt j |cc}w)arGenerates a rollout ID. Args: release_id: str, release ID. target_id: str, target ID. rollouts: [apitools.base.protorpclite.messages.Message], list of rollout messages. Returns: rollout ID. Raises: googlecloudsdk.command_lib.deploy.exceptions.RolloutIdExhaustedError: if there are more than 1000 rollouts with auto-generated ID. i)rr=rangeROLLOUT_ID_TEMPLATErr1RolloutIDExhaustedError) release_id target_idr5r rollout_idsirs rr.r.sm -55Hq166"H+5 D>a%++J 1EJ$   --j996sA()NN)NNNNNN) __doc__apitools.base.pyrr/"googlecloudsdk.api_lib.clouddeployrrrgooglecloudsdk.calliope!googlecloudsdk.command_lib.deployr1googlecloudsdk.corerrrrrYr-rjrrZr"ENQUEUETIME_ROLLOUT_ORDERBYrrr&r*r6rUr_r\r.rrrys7?:66.I#)I1! O!%4!'!0.+0 $2KO6F !KA\B:r