K*dZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Zddl mZdd lmZddlmZdd lmZdd lmZdd lmZdd lmZddlmZdd lmZddlmZddlmZddlZGddej@Z!Gdde!Z"Gdde"ejFZ$Gdde!ejJZ%Gdde!Z&Gdde&ejNZ'Gdde&ejPZ) d%dZ* d&d Z+d!Z,d"Z-d'd#Z.d$Z/y)(zIBase classes for [enable|disable|describe] commands for Feature resource.)absolute_import)division)unicode_literals) exceptions)util) enable_api)apis)base)api_util) resources)info)log) properties)retryNcPeZdZdZdZedZdfd ZddZddZ ddZ xZ S) FeatureCommandzFFeatureCommand is a mixin adding common utils to the Feature commands.c@tj|jS)z2The Feature info entry for this command's Feature.)r Get feature_name)selfs ?lib/googlecloudsdk/command_lib/container/fleet/features/base.pyfeaturezFeatureCommand.feature+s 88D%% &&c@tt| |j|S)zZBuilds the full resource name, using the core project property if no project is specified.)superrFeatureResourceNamer)rproject __class__s rrz"FeatureCommand.FeatureResourceName0s$  ++/+<+EE LL % %w 0 11rc |jj|j|S#tj$r|j |tj $r|j|wxYw)zBFetch this command's Feature from the API, handling common errors.) hubclient GetFeaturerapitools_exceptionsHttpNotFoundErrorr)HttpUnauthorizedErrorr+r(s rr.zFeatureCommand.GetFeatureBsl- ^^ & &t'?'?'H II  0 01  ' ' 00  4 4-  # #G ,,-s ),AA4N) __name__ __module__ __qualname____doc__rpropertyrrr)r+r. __classcell__)rs@rrr's2N, ' 'G - 1-rrceZdZdZdZdZy)EnableCommandMixinz.A mixin for functionality to enable a Feature.ctjjjj }|j j r*tj||j j tj|} tjdd}|j|jj||j |f|j"d}d j+|j j,}|j?|jj@|| S#tj$$r9t'j(dj+|j j,t.j0$rr}t3j4|}|j6dk7rt8j:j=dj+|j j,|Yd}~yd}~wwxYw) Ng?) max_retrialsexponential_sleep_multiplieri)argsshould_retry_ifsleep_msz-Retry limit exceeded waiting for {} to enableALREADY_EXISTSz.{} Feature for project [{}] is already enabledz$Waiting for Feature {} to be created)opmessage)!rr"r#rr$rapirEnableServiceIfDisabledrLocationResourceNamerRetryerRetryOnExceptionr- CreateFeaturer_FeatureAPINotEnabledMaxRetrialsExceptionrr%r&r'r/HttpConflictErrorcore_api_exceptionsHttpErrorPayloadstatus_descriptionrstatusPrint WaitForHubOpfeature_waiter) rrrparentretryerrCeerrormsgs rEnablezEnableCommandMixin.EnableOs$$,,668G ||(($,,2B2BC  & &w /F  14Pg  # # .. & &))7344 $ b" 1 7 7 !! #C   T^^::r3  OO  % %*    9 @ @ll'' ) **  0 0 !2215e  ! !%5 5  jjGNN ,, # #W./  s AD--AG< A(G77G<cF~~|jjsy|tjk7ryt j |}|j dk(r0|jj|jvrd|jvsytjjdy)NFFAILED_PRECONDITIONzis not enabledz/Waiting for service API enablement to finish...T) rrEr/HttpBadRequestErrorrNrOrPrDrrQrR)rexc_type exc_value tracebackstaterXs rrKz(EnableCommandMixin._FeatureAPINotEnabledms5 <<   &:::   0 0 ;E  $ $(= = LL   -  - JJFG rN)r3r4r5r6rZrKrrr:r:Ls6P< rr:ceZdZdZdZy) EnableCommandz2Base class for the command that enables a Feature.cT|j|jjSr2)rZmessagesFeaturer_s rRunzEnableCommand.Runs ;;t}},,. //rN)r3r4r5r6rjrbrrrdrd|s :0rrdceZdZdZdZdZy)DescribeCommandzBBase class for the command that describes the status of a Feature.c"|jSr2)r.rhs rrjzDescribeCommand.Runs ?? rc|Dcic]}tj||}}i}|jrb|jjDcic]C}tj|j|vr tj|j|E}}|j Dcgc] \}}||vr| }}}|r9t jdj|jj|i} |jrb|jjDcic]C}tj|j|vr tj|j|E} }|jjjt|j!|_|jjj#|Dcgc] }|| vr| | c}|_ ycc}wcc}wcc}}wcc}wcc}w)aLeave only specs and states of Memberships in the Feature. Respects the order of the Membership specs in the original Feature. Args: feature: Feature in the v1 API. memberships: List of resource names according to go/resource-names. Ideally, the existence of these Memberships will have been verified. Returns: None Raises: exceptions.Error: if any of Memberships does not have a spec in Feature. zcThe following requested memberships are not configured on the {} feature, under membershipSpecs: {})additionalPropertiesN)rMembershipPartialNamemembershipSpecsrokeyitemsrr%r&rr'membershipStatesrfrgMembershipSpecsValuelistvaluesMembershipStatesValue) rr membershipsm memberships_by_location_and_name%membership_specs_by_location_and_nameentry location_namemissing_memberships&membership_states_by_location_and_names rfilter_feature_for_membershipsz.DescribeCommand.filter_feature_for_membershipss"3>(2=Q""1%q(+%(-/)..CC/Ce  ' ' 2 -.  $ $UYY / 6C,/!A F F H H M1  E E H     1 F4<<,,.A B  .0*//DD0De  ' ' 2 23  $ $UYY / 6D-0#mm33HH! 1 8 8 : IG $}}44JJ; ::: 31 5:  K GK( /0 sG( AG-'G2AG8 G=N)r3r4r5r6rjrrbrrrlrlsJ ;rrlceZdZdZdZy)UpdateCommandMixinz.A mixin for functionality to update a Feature.cD |jj|j||}dj |jj}|j|jj||dS#tj$r|j wxYw)z>Update provides common API, display, and error handling logic.z$Waiting for Feature {} to be updatedFrDwarnings) r- UpdateFeaturerr/r0r)r&rr'rSrT)rmaskpatchrCrYs rUpdatezUpdateCommandMixin.Updates* >> ' '(@(@(BD% Pb 1 7 7 !! #C    %%r3    0 0*  ' ' ))*s +A;;$BN)r3r4r5r6rrbrrrrs 6 rrceZdZdZy) UpdateCommandzBase class for the command that updates a Feature. Because Features updates are often bespoke actions, there is no default `Run` override like some of the other classes. N)r3r4r5r6rbrrrrsrrceZdZdZej dddZej dddZdZe d Z d Z d Z d Z y )DisableCommandzLBase class for the command that disables an entire or parts of a Feature. z--force store_truezForce disablement. Bypasses any prompts for confirmation. When disabling the entire feature, proceeds even if the feature is in use. Might result in unexpected behavior.)actionhelpz--fleet-default-member-configzDisable the [fleet-default membership configuration](https://cloud.google.com/kubernetes-engine/fleet-management/docs/manage-features). Does not change existing membership configurations. Does nothing if the feature is disabled.Fc|jj||jr|jj|yyr2) FORCE_FLAG AddToParsersupport_fleet_default FLEET_DEFAULT_MEMBER_CONFIG_FLAG)clsparsers rArgszDisableCommand.Argss6NNv&    **66v>!rc|jr|jr|jy|j|jyr2)rfleet_default_member_configclear_fleet_defaultDisableforce)rr?s rrjzDisableCommand.Runs/ !!d&F&F   ll4::rc* |jj|j|}dj |j j}|j|jj||dy#tj$rYywxYw)N)rz$Waiting for Feature {} to be deletedFr) r- DeleteFeaturerr/r0r&rr'rSresourceless_waiter)rrrCrDs rrzDisableCommand.Disable s >> ' '(@(@(B% ' Pb5;; !!#G **B%Q  0 0   s+A<<BBcdg}|jjd} |j||S#tj$rE}t j |t j |jk7r|Yd}~yd}~wwxYw)Nr placeholder)name)rfrgrrr%six text_typer))rrrrWs rrz"DisableCommand.clear_fleet_defaults} ) *D MM ! !} ! 5E [[u %%    q S]]4+F+F+HI I Js3B ;BB N)r3r4r5r6 calliope_baseArgumentrrr classmethodrrjrrrbrrrrsr&}%%  2 *&<]%;%;% 6 &" ??  Q rrcl|jds"|jds|j|rFtj||s|stj||Stj||dS|s|s t ||t jd\}}|r$tjdj||rDt|dk(r6tjjdj|d|dS|rtj|}||St ||) a*Returns a membership on which to run the command, given the arguments. Allows for a `--membership` flag or a `MEMBERSHIP_NAME` positional flag. Args: args: object containing arguments passed as flags with the command prompt: whether to prompt in console for a membership when none are provided in args autoselect: if no membership is provided and only one exists, automatically use that one search: whether to search for the membership and error if it does not exist (not recommended) flag_override: to use a custom membership flag name 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 membershipMEMBERSHIP_NAMETfilter_cluster_missingLocations {} are currently unreachable. Please specify memberships using `--location` or the full resource name (projects/*/locations/*/memberships/*)Selecting membership [{}].r)IsKnownAndSpecifiedr MembershipLocationSpecifiedMembershipResourceNameSearchMembershipResourceMembershipRequiredErrorr ListMembershipsFullrr%r&lenrrQrRPromptForMembership)r?prompt autoselectsearch flag_overrideall_memberships unreachablers rParseMembershipr's18 l+t/G/G0 44]C,,T-:g}t|drd|jrXtjd\}}|r$t j dj ||st j d|S|jdrtj|r|tj|z }|r@|D];} tj| rt j dj | |sYttj|dkDr8ttj||tj|dz }|r|S|s |s t!|tjd\}}|r$t j d j ||rEt|dk(r7t"j$j'd j |d |d gS|r+tj(| } | r|j+| |St!|) aVParses a list of membership resources from args. Allows for a `--memberships` flag and a `--all-memberships` flag. Args: args: object containing arguments passed as flags with the command prompt: whether to prompt in console for a membership when none are provided in args prompt_cancel: whether to include a 'cancel' option in the prompt autoselect: if no memberships are provided and only one exists, automatically use that one allow_cross_project: whether to allow memberships from different projects search: whether to check that the membership exists in the fleet Returns: memberships: A list of membership resource name strings Raises: exceptions.Error if no memberships were found or memberships are invalid calliope_exceptions.RequiredArgumentException if membership was not provided rTrzaLocations {} are currently unreachable. Please try again or specify memberships for this command.z&No Memberships available in the fleet.ryz*Membership {} does not exist in the fleet.rrrr)cancel)hasattrrr rrr%r&rr rPluralMembershipsResourceNames GetMembershiprGetMembershipProjectsCrossProjectErrorSearchMembershipResourcesPluralrrrQrRrappend) r?r prompt_cancelrallow_cross_projectrryrrrs rParseMembershipsPluralras6+ T$%$*>*>#+#?#?#$% O[    228&2E GG    E FF  m,,,T2Y==dCCk %J'' 3""<CCJOQ Q& !S  ) )+ 6&8:;&< ? ? LMMY>> t--k   !$ ''!)!=!=!"#/;    2396+3F HHC(A-JJ1889KLM A   ..mDJ$ %%rcrtjdd}|jj|jj t jj}|jDcgc]6}t|jstj|j8c}Scc}w)z}Lists Membership IDs in the fleet for the current project. Returns: A list of Membership resource IDs in the fleet. gkehubv1beta)rU) core_apisGetClientInstanceprojects_locations_membershipsListMESSAGES_MODULE-GkehubProjectsLocationsMembershipsListRequesthub_base HubCommandrGr _ClusterMissingendpointrMembershipShortnamer)clientresponserzs rListMembershipsrs  & &x :&  2 2 7 7 JJ$$99;K=>( !! !! QZZ ( qvv&!  s6;B4cJtjdj|S)NzoMemberships for this command must belong to the same project and cannot mix project number and project ID ({}).)rr%r&)projectss rrrs$   --3VH-= ??rcd|r|}n|jdrd}nd}tj|dS)a Parses a list of membership resources from args. Assumes a `--memberships` flag or a `MEMBERSHIP_NAME` flag unless overridden. Args: args: argparse.Namespace arguments provided for the command flag_override: set to override the name of the membership flag Returns: memberships: A list of membership resource name strings Raises: exceptions.Error: if no memberships were found or memberships are invalid calliope_exceptions.RequiredArgumentException: if membership was not provided rryz5At least one membership is required for this command.)rcalliope_exceptionsRequiredArgumentException)r?rflags rrrs@" D  12 D D  6 6 C EErc\dD]'}t||stt||ddcSy)N) gkeClustermultiCloudCluster onPremClusterclusterMissingF)rgetattr)rzts rrrs, ?aq!} WQ]$4e << @r)FFFr)FTFFF)r)0r6 __future__rrrapitools.base.pyrr/&googlecloudsdk.api_lib.container.fleetrgooglecloudsdk.api_lib.servicesrgooglecloudsdk.api_lib.utilr rrNgooglecloudsdk.callioper rr*googlecloudsdk.command_lib.container.fleetr rr 3googlecloudsdk.command_lib.container.fleet.featuresr googlecloudsdk.corerrgooglecloudsdk.core.utilrrrrr: CreateCommandrdrlrr DeleteCommandrrrrrrrrbrrrs P&'>769I9E?G@D*#** "-X(("-J--`0& (C(C0Dnm&C&CDN"& (C(CA')D)DAJ!$ "$ 75v#()-&+/4"' Q&l$? E6=r