dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z d Z d Z d Zd Zd ZGddeZy)zUtils for GKE Hub commands.)absolute_import)division)print_function)unicode_literals)waiter) kube_util) exceptionsaapiVersion: apiextensions.k8s.io/v1beta1 kind: CustomResourceDefinition metadata: name: memberships.hub.gke.io spec: group: hub.gke.io scope: Cluster names: plural: memberships singular: membership kind: Membership versions: - name: v1beta1 served: true storage: true validation: openAPIV3Schema: required: - spec properties: metadata: type: object properties: name: type: string pattern: '^(membership|test-.*)$' spec: type: object properties: owner: type: object properties: id: type: string description: Membership owner ID. Should be immutable.zykind: Membership apiVersion: hub.gke.io/v1beta1 metadata: name: membership spec: owner: id: projects/{project_id}c|j}|yd}|j|s$tjdj ||t |dS)aReturns the project id of the fleet the cluster is a member of. The Membership Custom Resource stores the project id of the fleet the cluster is registered to in the `.spec.owner.id` field. Args: kube_client: A KubernetesClient. Returns: a string, the project id None, if the Membership CRD or CR do not exist on the cluster. Raises: exceptions.Error: if the Membership resource does not have a valid owner id Nz projects/z(Membership .spec.owner.id is invalid: {})GetMembershipOwnerID startswithr Errorformatlen) kube_clientowner_id id_prefixs Blib/googlecloudsdk/command_lib/container/fleet/exclusivity_util.pyGetMembershipCROwnerIDrNsb" - - /(  )   Y '   299(C EE #i./ ""c\tj|}|jt|y)a Creates or updates the Membership CRD and CR with the hub project id. Args: kube_client: A KubernetesClient. project: The project id of the hub the cluster is a member of. Raises: exceptions.Error: if the Membership CR or CRD couldn't be applied. ) project_idN)MEMBERSHIP_CR_TEMPLATErApplyMembershipMEMBERSHIP_CRD_MANIFEST)rprojectmembership_cr_manifests rApplyMembershipResourcesris*288G8L 57MNrc  tjtjt |dtj tj tjtj\}}|s$tjdj|y#tj$rtjdwxYw)zDeletes the Membership CRD. Due to garbage collection all Membership resources will also be deleted. Args: kube_client: A KubernetesClient. z%Deleting membership CR in the cluster)pre_start_sleep_ms max_wait_mswait_ceiling_mssleep_msz,Timeout deleting membership CR from cluster.z6Could not delete membership CR from cluster. Error: {}N) rWaitForrKubernetesPollerMembershipCRDeleteOperation"NAMESPACE_DELETION_INITIAL_WAIT_MSNAMESPACE_DELETION_TIMEOUT_MS'NAMESPACE_DELETION_MAX_POLL_INTERVAL_MS+NAMESPACE_DELETION_INITIAL_POLL_INTERVAL_MS TimeoutErrorr r r)r succeedederrors rDeleteMembershipResourcesr-xs K~~""$#K0/$GG;;!IIFFHIu    @GGN PP    K   I JJKs A2B)Cc"eZdZdZdZdZdZy)r%z:An operation that waits for a membership CR to be deleted.c<||_d|_d|_d|_y)NF)rdoner+r,)selfrs r__init__z$MembershipCRDeleteOperation.__init__s"DDIDNDJrcy)Nz)r1s r__str__z#MembershipCRDeleteOperation.__str__s %rcr|jj}|syd|vrd|_d|_y||_y)zEUpdates this operation with the latest membership CR deletion status.NNotFoundT)rDeleteMembershipr0r+r,)r1errs rUpdatez"MembershipCRDeleteOperation.Updates=    + + -C  SdidndjrN)__name__ __module__ __qualname____doc__r2r5r:r4rrr%r%sB &rr%N)r> __future__rrrrgooglecloudsdk.api_lib.utilr*googlecloudsdk.command_lib.container.fleetrgooglecloudsdk.corer rrrrr-objectr%r4rrrDsN"&%'.@*#LP!#6 OP8&r