%dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lZeeej2fZGd dZy )z;Provides a utility mixin for Poco-specific gcloud commands.)absolute_import)division)unicode_literals)Dict)messages)util)parser_extensions)base)convert) exceptionsNc eZdZdZddZdej deeeffdZ de fdZ ddej de d e de fd Z d ejdejfd Zdd e ddfdZde fdZdej de fdZy) PocoCommandz5A mixin for Policy Controller specific functionality.returnNcdg}|jjd}|!|jj||_ |j ||S#t j $rX}|j}tj|tj|k(r|j|cYd}~S|d}~wwxYw)z!Update the feature configuration.fleet_default_member_config notarealname)nameNpolicycontroller) rFeature"CommonFleetDefaultMemberConfigSpecfleetDefaultMemberConfigUpdategcloud_exceptionsErrorFeatureNotEnabledErrorsix text_typeEnable)self default_cfgmaskfeatureefnes Jlib/googlecloudsdk/command_lib/container/fleet/policycontroller/command.pyupdate_fleet_defaultz PocoCommand.update_fleet_default's ) *Dmm##  $G -- : :* ; &  [[w ''  " "  ' ' )c q S]]3/ /{{7## s%AC)A B<4C:B<<Cargsctj|dddDcic]}tj||c}Scc}w)z@Filtered map of short membership names to full membership paths.TF)prompt prompt_cancel autoselect) feature_baseParseMembershipsPlural fleet_utilMembershipPartialName)r r(paths r&_membership_pathszPocoCommand._membership_pathsAsQ !77 Ut   D ((.4   s<c8 |jj|jjS#tj $rQ}|j }tj|tj|k(rtcYd}~S|d}~wwxYw)zFetches the current specs from the server. If the feature is not enabled, this will return an empty dictionary. Returns: dictionary mapping from full path to membership spec. N) hubclientToPyDict GetFeaturemembershipSpecsrrrrrdictr r$r%s r& current_specszPocoCommand.current_specsLss ^^ $ $T__%6%F%F GG  " "  ' ' )c q S]]3/ /v  "25BAB BBBignore_missingignore_metadatac |j|}|jjDcic]4\}}tj||vrtj|||f6}}}|r9|jDcic]\}\}}|||j |f}}}}|rS|jD cgc]\} } | |vs | | f} } } | D]$\} } | |j jf|| <&nNd}|Dcgc]*}||vr$tj|j|,}}|rtj||jDcic]\}}|| c}}Scc}}wcc}}}wcc} } wcc}wcc}}w)aRetrieves memberships specified by the command that exist in the Feature. Args: args: The argparse object passed to the command. ignore_missing: Use this to return a mapping that includes an 'empty' spec for each specified path if it doesn't already exist. ignore_metadata: If true, remove the Hub-managed metadata (i.e. origin). If the spec is being retrieved for reporting (i.e. describe) then set to false to get the full current value. If it is being used to update the spec leave as True so that the return spec can be used in a patch. Returns: A dict mapping a path to the membership spec. Raises: exceptions.DisabledMembershipError: If the membership is invalid or not enabled. 2Policy Controller is not enabled for membership {}) r2r:itemsr/r0 _rebuild_specrMembershipFeatureSpecr InvalidPocoMembershipErrorformatvalues)r r(r<r=memberships_pathsr1specspecs partial_pathsfmissingshortfullmsgmissing_membershipss r& path_specszPocoCommand.path_specs]s2..t4 ,,.446 6JD$  + +D 15F F ((.t <6   -2KKM,9(lLT4 t11$78 8,9  $5$;$;$=P$=DAq%!Q$=gP +%dmmAACDe ! Ac('d    / / 40@ A' 334GHH,1<<> :> ::9 Q  ;s#9E1!E7? E> E>/F F rGcN|jj|jS)auRebuilds the spec to only include information from policycontroller. This is necessary so that feature-level values managed by Hub are not unintentionally overwritten (i.e. 'origin'). Args: spec: The spec found by querying the API. Returns: MembershipFeatureSpec with only policycontroller values, leaving off other top-level data. r)rrBr)r rGs r&rAzPocoCommand._rebuild_specs) == . ... / rHc.|jj|jj|}|sq |j |jjD]F}|j}|j}tj ||d|} |j#|dg| Hy |j%dg|S#t j $rV}|j}tj|tj|k(r|j|Yd}~d}~wwxYw#t j $rX}|j}tj|tj|k(r|j|cYd}~S|d}~wwxYw)aBMerges spec changes and sends and update to the API. Specs refer to PolicyControllerMembershipSpec objects defined here: third_party/py/googlecloudsdk/generated_clients/apis/gkehub/v1alpha/gkehub_v1alpha_messages.py (Note the above is for the ALPHA api track. Other tracks are found elsewhere.) Args: specs: Specs with updates. These are merged with the existing spec (new values overriding) and the merged result is sent to the Update api. use_default_cfg: If true, use the default config for the update. Returns: None )r7NrrGmembership_specs)rrr4ToMembershipSpecsr6rrrrrrr7additionalPropertieskeyvaluer ToV2MembershipFeatureUpdateV2r) r rHuse_default_cfgr#r$r%rGmembership_pathv1_specmembershipfeatures r& update_specszPocoCommand.update_specssW"mm##88?$G  ))>>$((**#99 /#5w  ox1BC ?{{./99 $ $))+ == s}}S1 1 ++g    $ $))+ == s}}S1 1W% %' s=B=*D)=D&A D!!D&)F?%   1 12E FF,2==? ;?r{sRB&'/E5TQR? 3((() g<g