r<SrSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSKJ r SS KJ r SS KJr SS KJr SS KJr SSKJr SS KJr SS KJ r SSKJr SSKJr \R4"S5rS5SjrSrS6SjrS6SjrS7Sjr S6Sjr!S8Sjr"S9Sjr#Sr$S6Sjr%Sr&S:Sjr'S:Sjr(Sr)S;Sjr*S6S jr+SS#jr.S>S$jr/S%r0S&r1S'r2S(r3S)r4S*r5S+r6S,r7S-r8S>S.jr9S/r:S?S0jr;S?S1jrS4r?g)@z3Functions for resource arguments in fleet commands.)absolute_import)division)unicode_literalsN)util) arg_parsers)base) exceptions)concepts)deps)api_util)concept_parsers) properties) console_ioz/locations/([a-z0-9-]+)/c[R"5(d&[R"USR U55eUcC[ R "5upU(a%[R"SR U55eU(d[R"S5e[R"[U5UUS9nUbX$S$)aPrompt the user for a membership from a list of memberships in the fleet. This method is referenced by fleet and feature commands as a fallthrough for getting the memberships when required. Args: memberships: List of memberships to prompt from flag: The name of the membership flag, used in the error message message: The message given to the user describing the prompt. cancel: Whether to include a "cancel" option. Returns: The membership specified by the user (str), or None if unable to prompt. Raises: OperationCancelledError if the prompt is cancelled by user RequiredArgumentException if the console is unable to prompt zvCannot prompt a console for membership. Membership is required. Please specify `{}` to select at least one membership.NLocations {} are currently unreachable. Please specify memberships using `--location` or the full resource name (projects/*/locations/*/memberships/*)z&No Memberships available in the fleet.)message cancel_option) r CanPromptcalliope_exceptionsRequiredArgumentExceptionformatr ListMembershipsFullr Error PromptChoiceMembershipPartialNames) membershipsflagrcancel unreachableidxs ;lib/googlecloudsdk/command_lib/container/fleet/resources.pyPromptForMembershipr"'s,       7 7 ''-vd| 66';;=K    45;VK5H JJ    C DD[) #!_ 6$6cZUVs/sHn[R"U5PM sn$s snf)z?Converts a list of full membership names to LOCATION/ID format.)rMembershipPartialName)rms r!rrUs%1< =A$ $ $Q ' == =s (c[R"S5[R"[RR R 5/n[R"SU(aUUS9$SUS9$)zCreate location attributes in resource argument. Args: help_text: If set, overrides default help text for `--location` Returns: Location resource argument parameter config z --locationlocationzLocation for the {resource}.)name help_text fallthroughs) r ArgFallthroughPropertyFallthroughrVALUESgkehubr(r ResourceParameterAttributeConfig)r*r+s r!_LocationAttributeConfigr1Zsi ,' z0077@@A,  2 2 &  !!-K !!r#cB[R"UU(aUS9$SS9$)zCreate basic attributes in resource argument. Args: attr_name: Name of the resource help_text: If set, overrides default help text Returns: Resource argument parameter config Name of the {resource}.)r)r*)r r0) attr_namer*s r!_BasicAttributeConfigr5ms0  2 2 &  II-F IIr#c >SnU(aUnOU(aSnO U(aSn[R"SUSSS[R[U5[ U(aSOSU5S9n [ R RUU U(aS OS UUS 9RU5 g ) =Add resource arg for projects/{}/locations/{}/memberships/{}. --membershipMEMBERSHIP_NAMEz --membershipsz%gkehub.projects.locations.memberships membershiprT) api_version resource_name plural_namedisable_auto_completers projectsId locationsId membershipsIdz8The group of arguments defining one or more memberships.z-The group of arguments defining a membership.pluralrequiredN r ResourceSpec DEFAULT_PROJECT_ATTRIBUTE_CONFIGr1r5r ConceptParser ForResource AddToParser) parserr; positionalrCmembership_required flag_overridemembership_help location_help flag_namespecs r!AddMembershipResourceArgrS}s)I!I I   - "::*=9)!-|_F G$++ AD " ,$ %0K$7r#cDURS5(agURS5(a"[RUR5bgURS5(a"[RUR 5bgURS5(aA[ URVs/sHn[RU5SLPM sn5(agURU5(a'[RURU55bggs snf)z;Returns whether a membership location is specified in args.r(Tr:Nr9rF) IsSpecifiedIsKnownAndSpecified _LOCATION_REsearchr:r9allrGetValue)argsrNr&s r!MembershipLocationSpecifiedr\s j!!  l++ 0C0C oo1"1#  /00\5H5H 6#'6(  m,,373C3CD3Ca|1T)3CD2F2F  m,,1D1D mmM"2$+/20   Es1!Dc[U5(a/[R"[U55(a [U5$UR U5(a [ X5nOGUR S5(a UR nO$UR S5(a URnOg[R"US9upEU(a%[R"SRU55eU(d[R"S5e/nUH0nU[R"U5:XdMURU5 M2 U(d%[R"SRU55e[U5S:a [!U5eUS $) a$Searches the fleet for an ambiguous membership provided in args. Only necessary if location is ambiguous, i.e. MembershipLocationSpecified(args) is False, or this behavior is necessary for backwards compatibility. If flag_override is unset, the argument must be called `MEMBERSHIP_NAME` if positional and `--membership` otherwise. Runs a ListMemberships API call to verify the membership exists. Args: args: arguments provided to a command, including a membership resource arg flag_override: a custom membership flag filter_cluster_missing: whether to filter out memberships that are missing a cluster. Returns: A membership resource name string (e.g. projects/x/locations/y/memberships/z) Raises: googlecloudsdk.core.exceptions.Error: unable to find the membership in the fleet r9r:Nfilter_cluster_missingr&No memberships available in the fleet.%Membership {} not found in the fleet.r)r\r GetMembershipMembershipResourceNamerVgetattrr9r:rr rrrMembershipShortnameappendlenAmbiguousMembershipError)r[rNr_arg_membershipall_memberships unavailablefoundexisting_memberships r!SearchMembershipResourcerosW2!&&8+A+AT",$,$ !$ '' m,,T1N  122))N  --__N !)!=!=3"5/    2396+3F HH    C DD %,112EFF ll&'-    /66~F HH 5zA~ "> 22 q/r#cFURS5(a URnOg[R"US9up4U(a%[R "SR U55eU(d[R "S5e/nUHn/nUH0nU[R"U5:XdMURU5 M2 U(d%[R "SR U55e[U5S:a [U5eURUS5 M U$) aSearches the fleet for the membership resources provided in args. Only necessary if location is ambiguous, i.e. MembershipLocationSpecified(args) is False. Assumes the argument is called `--membership`, `--memberships` if plural, or `MEMBERSHIP_NAME` if positional. Runs ListMemberships API call to verify the membership exists. Args: args: arguments provided to a command, including a membership resource arg filter_cluster_missing: whether to filter out memberships that are missing a cluster. Returns: A list of membership resource names (e.g. ["projects/x/locations/y/memberships/z"]) Raises: googlecloudsdk.core.exceptions.Error: unable to find a membership in the fleet rNr^zLocations [{}] are currently unreachable. Please specify memberships using `--location` or the full resource name (projects/*/locations/*/memberships/*)r`rarbr) rVrr rr rrrrfrgrhri) r[r_arg_membershipsrkrlrrjrmrns r!SearchMembershipResourcesPluralrrs, m,,&&O !)!=!=3"5/    2396+3F HH    C DD+'n E. 4334GH H () /     1 8 8 H JJ Ua $^ 44uQx ( r#cL[R"SRU55$)NzMultiple memberships named {} found in the fleet. Please use `--location` or full resource name (projects/*/locations/*/memberships/*) to specify.)r rr)r:s r!riri%s&   <=CVJ=O QQr#cURU5(a7URRU5R5R 5$URS5(a2URR R5R 5$URR R5R 5$)aGets a membership resource name from a membership resource argument. If flag_override is unset, the argument must be `MEMBERSHIP_NAME` if positional and `--membership` otherwise. Args: args: arguments provided to a command, including a membership resource arg flag_override: a custom membership flag name Returns: The membership resource name (e.g. projects/x/locations/y/memberships/z) r9)rVCONCEPTSrZParse RelativeNamemembership_namer:)r[rNs r!rdrd,s m,, == ! !- 0 6 6 8 E E GG /00 == ( ( . . 0 = = ??  ! ! ' ' ) 6 6 88r#cURRR5Vs/sHoR5PM sn$s snf)a Gets a list of membership resource names from a --memberships resource arg. Args: args: arguments provided to a command, including a plural memberships resource arg Returns: A list of membership resource names (e.g. projects/x/locations/y/memberships/z) )rurrvrw)r[r&s r!PluralMembershipsResourceNamesrz@s6%)MM$=$=$C$C$E F$Eq.. $E FF FsAcU[RRL=(a& [R"5[R :H$)aEReturns whether regional memberships should be included. This will be updated as regionalization is released, and eventually deleted when it is fully rolled out. Args: track: The release track of the command Returns: A bool, whether regional memberships are supported for the release track in the active environment ) calliope_base ReleaseTrackALPHAcmd_util APIEndpoint AUTOPUSH_API)tracks r!UseRegionalMembershipsrNs9 =--33 3 7 5 557r#cX/SQnU[RRL=(a X;$)alReturns whether project is allowlisted for regional memberships in Prod. This will be updated as regionalization is released, and eventually deleted when it is fully rolled out. Args: project: The parent project ID of the membership track: The release track of the command Returns: A bool, whether project is allowlisted for regional memberships in Prod )zgkeconnect-proberzgkeconnect-e2ezgkehub-cep-testzconnectgateway-gke-testingz xuebinz-gkezkolber-anthos-testingzanthonytong-hub2 wenjuntoy2zhub-regionalisation-testzhub-regionalisation-test-2za4vm-ui-tests-3m4a-ui-playground-1zanthos-cl-e2e-testsza4vm-ui-playgroundr)r|r}r~)projectrprod_regional_allowlists r!InProdRegionalAllowlistr_s0" -,,22 2 * (*r#c[5nUHfn[R"SU5nU(d%[R"SR U55eUR URS55 Mh [U5$)aReturns all unique project identifiers of the given membership names. ListMemberships should use the same identifier (all number or all ID) in membership names. Users can convert their own project identifiers for manually entering arguments. Args: memberships: A list of full membership resource names Returns: A list of project identifiers in the parents of the memberships Raises: googlecloudsdk.core.exceptions.Error if unable to parse any membership name z2projects\/(.*)\/locations\/(.*)\/memberships\/(.*)zMUnable to parse membership {} (expected projects/*/locations/*/memberships/*)rb) setrematchr rraddgrouplist)rprojectsr&rs r!GetMembershipProjectsrsm U( a HHJA NE    EEKVAY PP LLQ  hr#cURU5(a&[U5(a [U5$[U5$[R "US5e)aReturns a membership on which to run the command, given the arguments. This function is currently only used by the unregister command. This logic should be combined with the feature ParseMembership function in a later CL. Allows for `MEMBERSHIP_NAME` positional flag. Args: args: object containing arguments passed as flags with the command membership_flag: the membership flag used to pass in the memberhip resource Returns: membership: A membership resource name string Raises: exceptions.Error: no memberships were found or memberships are invalid calliope_exceptions.RequiredArgumentException: membership was not provided z(membership is required for this command.)rVr\rdrorr)r[membership_flags r!ParseMembershipArgrsQ( o.."4(( #D )) %d ++55A CCr#c[R"S[R"[R SS9/U(aUS9$SS9$)zCreate basic attributes that fallthrough location to global in resource argument. Args: help_text: If set, overrides default help text Returns: Resource argument parameter config r(z%global is the only supported location)functionhintr3)r)r+r*)r r0r FallthroughrDefaultToGlobal)r*s r!'_DefaultToGlobalLocationAttributeConfigrsP  2 2    //:  '   .G  r#c [R"SUSSS[R[5[ SU5S9n[ R RUUSSUUSS 0S 9RU5 g ) z8Add resource arg for projects/{}/locations/{}/scopes/{}.z gkehub.projects.locations.scopesscopescopesT)r;r<r=r>r?r@scopesIdz0The group of arguments defining the Fleet Scope.Fr()rCrDrflag_name_overridesN r rFrGrr5r rHrIrJ)rKrQr; scope_helprDrrRs r!AddScopeResourceArgrs}   ("::9;$Wj9 $++ 8  %r*, Kr#c[R"SUSSS[R[5[ SS5[ SU5S9 n[ R RUUSS US S 0S 9RU5 g )zFAdd resource arg for projects/{}/locations/{}/scopes/{}/namespaces/{}.z+gkehub.projects.locations.scopes.namespaces namespace namespacesTrthe)r;r<r=r>r?r@r namespacesIdz4The group of arguments defining the Fleet Namespace.Fr(r)rCrDrNr)rKrQr;namespace_helprDrRs r!AddScopeNamespaceResourceArgrs   3"::9;$We4(nE $++ < %r*,Kr#cSn[R"SUSSS[R[5[ SS5[ SU5S9 n[ R RUUS S SS 9RU5 g ) zLAdd resource arg for projects/{}/locations/{}/scopes/{}/rbacrolebindings/{}.NAMEz1gkehub.projects.locations.scopes.rbacrolebindingsrbacrolebindingrbacrolebindingsTrr)r;r<r=r>r?r@rrbacrolebindingsId3The group of arguments defining an RBACRoleBinding.FrBNrErKr; rbacrb_helprQrRs r!AddScopeRBACResourceArgr s)   9%$"::*,$Wb1./@+N $++ ;  , Kr#cSn[R"SUSSS[R[5[ SS5[ SU5S9 n[ R RUUS S SS 9RU5 g ) r7rz5gkehub.projects.locations.namespaces.rbacrolebindingsrrTrr)r;r<r=r>r?r@rrrFrBNrErs r!AddRBACResourceArgr#s)   =%$"::*,(b9./@+N P$++ ;  , ![(r#c[S5RU5 UR5n[S5RU5 [ S5RU5 g)zzAdds flags to an argparse parser for updating namespace labels. Args: parser: The argparse parser to add the flags to. rN)_GetUpdateNamespaceLabelsFlagrJadd_mutually_exclusive_group_GetClearNamespaceLabelsFlag_GetRemoveNamespaceLabelsFlag)rK remove_groups r!AddUpdateNamespaceLabelsFlagsr9sP  ,88@446,{+77 ,88Fr#c[R"SS9nUR[S55 UR[ S55 [ S5U/$)NT)mutexr)r| ArgumentGroup AddArgumentrrr)rs r!UpdateScopeLabelsFlagsrGsW,,48,"7+#G,$G, r#c8[S5RU5 g)zzAdds flags to an argparse parser for creating namespace labels. Args: parser: The argparse parser to add the flags to. rN)_GetCreateNamespaceLabelsFlagrJ)rKs r!AddCreateNamespaceLabelsFlagsrUs  ,88@r#c[S5/$)Nr)rr#r!CreateScopeLabelsFlagsr^s ' 0 11r#c jSn[R"SRU5SSRXS9S9$)Nnamespace-labelsz --clear-{} store_truea Remove all {resource_type}-level labels from the cluster namespace. If `--update-{labels}` is also specified then `--clear-{labels}` is applied first. For example, to remove all labels: $ {{command}} {resource_type}_name --clear-{labels} To remove all existing {resource_type}-level labels and create two new labels, ``foo'' and ``baz'': $ {{command}} {resource_type}_name --clear-{labels} --update-{labels} foo=bar,baz=qux labels resource_type)actionhelp)r|Argumentrr labels_names r!rrbs@"+   +&  fKfE GGr#c Sn[R"SRU5S[R"5[R SRXS9S9$)Nrz --remove-{}KEYz List of {resource_type}-level label keys to remove in the cluster namespace. If a label does not exist it is silently ignored. If `--update-{labels}` is also specified then `--update-{labels}` is applied first. rmetavartyperr)r|rrrArgList UpdateActionrs r!rrvsU"+   ;'     % %  & & A C Cr#c Sn[R"SRU5S[R"5[R SRUS9S9$)z?Makes a base.Argument for the `--update-namespace-labels` flag.rz --update-{} KEY=VALUEz List of {resource_type}-level label KEY=VALUE pairs to update in the cluster namespace. If a label exists, its value is modified. Otherwise, a new label is' created.rrr|rrrArgDictrrs r!rrsS"+   ;'     % % &}&5 77r#c Sn[R"SRU5S[R"5[R SRUS9S9$)Nrz--{}rzH List of {resource_type}-level label KEY=VALUE pairs to add. rrrrs r!rrsO"+    mmK      % % &}& - //r#cfURRR5R5$)a[Gets an RBACRoleBinding resource name from a resource argument. Assumes the argument is called NAME. Args: args: arguments provided to a command, including an rbacRB resource arg Returns: The rbacRB resource name (e.g. projects/x/locations/global/namespaces/y/rbacrolebindings/z projects/x/locations/global/scopes/y/rbacrolebindings/z) )rur)rvrwr[s r!RBACResourceNamers&    ! ! # 0 0 22r#cSn[R"SUSSS[R[5[ SS5[ SU5S9 n[ R RUUS S SS 9RU5 g ) zIAdd resource arg for projects/{}/locations/{}/memberships/{}/bindings/{}.BINDINGz.gkehub.projects.locations.memberships.bindingsbindingbindingsTr:r)r;r<r=r>r?r@rA bindingsIdz5The group of arguments defining a Membership Binding.FrBNrE)rKr; binding_helprQrRs r!AddMembershipBindingResourceArgrs)   6"::*,),;&y,? A$++ =  , ![(r#cfURRR5R5$)aGets a Membership-Binding resource name from a resource argument. Assumes the argument is called BINDING. Args: args: arguments provided to a command, including a Binding resource arg Returns: The Binding resource name (e.g. projects/x/locations/l/memberships/y/bindings/z) )rurrvrwrs r!MembershipBindingResourceNamers&    $ $ & 3 3 55r#c [R"SUSSS[R[5[ S5S9n[ R RSUSSSSS 0S 9RU5 g ) z:Add resource arg for projects/{}/locations/{}/rollouts/{}.z"gkehub.projects.locations.rolloutsrolloutrolloutsT)r;r<r=r>r?r@ rolloutsIdz0The group of arguments defining a Fleet Rollout.Fr(rr) resource_spec group_helprCrDrNrrKr;rRs r!AddRolloutResourceArgrsx   *"::9;&y1 $++ C %r*,Kr#c [R"SUSSS[R[5[ S5S9n[ R RSUSSSS S 0S 9RU5 g ) zBAdd resource arg for projects/{}/locations/{}/rolloutSequences/{}.z*gkehub.projects.locations.rolloutSequencesrolloutSequencerolloutSequencesTrollout_sequence)r;r<r=r>r?r@rolloutSequencesIdz3The group of arguments defining a Rollout Sequence.Fr(rrNrrs r!AddRolloutSequenceResourceArgrsy   2%$"::9;./AB $++ F %r*,Kr#cfURRR5R5$)a.Gets a RolloutSequence resource name from a resource argument. Assumes the argument is called ROLLOUTSEQUENCE. Args: args: arguments provided to a command, including a RolloutSequence resource arg Returns: The RolloutSequence resource name (e.g. projects/x/locations/l/rolloutSequences/z) )rurolloutsequencervrwrs r!RolloutSequenceResourceNamers&  & & , , . ; ; ==r#c Sn[R"SUSS[R[5[ S5S9n[ R RUUSSSS S 0S 9RU5 g ) zGAdd resource arg for projects/{}/locations/{}/workloadidentitypools/{}.WORKLOAD_IDENTITY_POOLz,iam.projects.locations.workloadIdentityPoolsworkloadidentitypoolworkloadidentitypools)r;r<r=r?r@workloadIdentityPoolsIdz9The group of arguments defining a Workload Identity Pool.FTr(rrNrE)rKr;rQrRs r!"AddWorkloadIdentityPoolResourceArgrs{')   4*)::*,34JK $++ L %r*,Kr#cfURRR5R5$)aGGets a WorkloadIdentityPool resource name from a resource argument. Assumes the argument is called WORKLOAD_IDENTITY_POOL. Args: args: arguments provided to a command, including a WorkloadIdentityPool resource arg Returns: The WorkloadIdentityPool resource name (e.g. projects/x/locations/l/workloadidentitypools/z) )ruworkload_identity_poolrvrwrs r! WorkloadIdentityPoolResourceNamer,s&  - - 3 3 5 B B DDr#)Nr8zPlease specify a membership: T)r)v1FFFrrr)rF)F)N)r9)rrrFN) NAMESPACErrF)rr)r)@__doc__ __future__rrrr&googlecloudsdk.api_lib.container.fleetrgooglecloudsdk.callioperrr|r r googlecloudsdk.calliope.conceptsr r *googlecloudsdk.command_lib.container.fleetr r(googlecloudsdk.command_lib.util.conceptsr googlecloudsdk.corergooglecloudsdk.core.consolercompilerWr"rr1r5rSr\rorrrirdrzrrrrrrrrrrrrrrrrrrrrrrrrrrr#r!rsX:&' 7/9E51?GD**2zz45 %)+ @#(7\> !& I"*.(-$)16+--/+-!8H(,.499x2jQ9( G7"*D4C<.   B  <0), G A2G( C 7 / 3 ), 600 > 2Er#