W0dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l mZddlZGd d eZy) z7Base class for Anthos GKE On-Prem API client resources.)absolute_import)division)unicode_literalsN)apis) arg_parsers)parser_extensions)gkeonprem_v1_client)gkeonprem_v1_messagesceZdZdZd&dZdej fdZd&dejfdZ dZ d Z dejfd Z dejfd Zdejfd Zdejfd ZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdejfdZdZdejfdZdejfdZdejfdZ dejfdZ!dejfdZ"dejfd Z#dejfd!Z$dejddfd"Z%dejfd#Z&dejfd$Z'dejfd%Z(y)' ClientBasez.Base class for Anthos GKE On-Prem API clients.Nc<|j|_||_yN)_get_client_instance_client_service)selfservices 8lib/googlecloudsdk/api_lib/container/gkeonprem/client.py__init__zClientBase.__init__#s,,.DLDMreturnc.tjddS)zReturns the client instance. Function created for IDE type hint only, inline type annotation is not supported due to gcloud using a low Python3 version. gkeonpremv1)rGetClientInstance)rs rrzClientBase._get_client_instance's  ! !+t 44rargscB||jvr t||S|S)aReturns the flag value if it's set, otherwise returns None. Args: args: An argparser Namespace class instance. flag: A string type flag name. default: The default value to return if not found in the argparser namespace. Returns: The flag value if it is set by the user. If the flag is not added to the interface, or it is added by not specified by the user, returns the default value. )GetSpecifiedArgsDictgetattr)rrflagdefaults rGetFlagzClientBase.GetFlag/s' t((** T4  nrcZ|jD]}t|try|syy)aReturns True if any of the kwargs is set to not None value. The added condition handles the case when user specified boolean False for the given args, and it should return True, which does not work with normal Python identity comparison. Args: kwargs: dict, a mapping from proto field to its respective constructor function. Returns: True if there exists a field that contains a user specified argument. TF)values isinstancebool)rkwargsvalues rIsSetzClientBase.IsSetBs, E4   ! rc|jjd|j}|jj|S)zGets a gkeonprem API resource.Get)name)rGetRequestType RelativeNamer+)r resource_refreqs rDescribezClientBase.DescribeWs; -$-- & &u -<3L3L3N OC ==  S !!rczt|jddr$|jjjSy)z@Parses user cluster resource argument and returns its reference.clusterN)rCONCEPTSr3Parserrs r_user_cluster_refzClientBase._user_cluster_ref\s.t}}i. ]] " " ( ( ** rczt|jddr$|jjjSy)AParses admin cluster resource argument and returns its reference. admin_clusterN)rr4r:r5r6s r_admin_cluster_refzClientBase._admin_cluster_refbs.t}}ot4 ]] ( ( . . 00 rczt|jddr$|jjjSy)zr.)rr location_refs r_location_namezClientBase._location_namens'%%d+L  & & (( rcJ|j|}|r|jSy)z3Parses user cluster from args and returns its name.N)r7r.rruser_cluster_refs r_user_cluster_namezClientBase._user_cluster_nameus(--d3  * * ,, rcf|j|}|r|jjSy)z:Parses user cluster from args and returns its parent name.N)r7Parentr.rCs r_user_cluster_parentzClientBase._user_cluster_parent|s1--d3  $ $ & 3 3 55 rcJ|j|}|r|jSy)z1Parses user cluster from args and returns its ID.N)r7NamerCs r_user_cluster_idzClientBase._user_cluster_ids(--d3  " " $$ rcJ|j|}|r|jSyz4Parses admin cluster from args and returns its name.N)r;r.rradmin_cluster_refs r_admin_cluster_namezClientBase._admin_cluster_names(//5  + + -- rcf|j|}|r|jjSy)z;Parses admin cluster from args and returns its parent name.N)r;rGr.rNs r_admin_cluster_parentz ClientBase._admin_cluster_parents1//5  % % ' 4 4 66 rcJ|j|}|r|jSy)z2Parses admin cluster from args and returns its ID.N)r;rJrNs r_admin_cluster_idzClientBase._admin_cluster_ids(//5  # # %% rczt|jddr$|jjjSy)r9admin_cluster_membershipN)rr4rVr5r6s r_admin_cluster_membership_refz(ClientBase._admin_cluster_membership_refs/t}}8$? ]] 3 3 9 9 ;; rcJ|j|}|r|jSyrM)rWr.rNs r_admin_cluster_membership_namez)ClientBase._admin_cluster_membership_names(::4@  + + -- rczt|jddr$|jjjSyz=Parses node pool resource argument and returns its reference. node_poolNrr4r\r5r6s r_node_pool_refzClientBase._node_pool_refs.t}}k40 ]] $ $ * * ,, rcJ|j|}|r|jSyz0Parses node pool from args and returns its name.N)r^r.rr node_pool_refs r_node_pool_namezClientBase._node_pool_names'''-M  ' ' )) rcJ|j|}|r|jSyz.Parses node pool from args and returns its ID.N)r^rJras r _node_pool_idzClientBase._node_pool_ids)''-M    !!rcf|j|}|r|jjSyz7Parses node pool from args and returns its parent name.N)r^rGr.ras r_node_pool_parentzClientBase._node_pool_parents0''-M  ! ! # 0 0 22 rc"tjj}|j|j|j d}dj |}dj dt|jD}tjd}|j|}|s$tjdj||j\}} } | |vr%tjdj|||| || dS) aValidates and parses a node taint object. Args: node_taint: tuple, of format (TAINT_KEY, value), where value is a string of format TAINT_VALUE:EFFECT. Returns: If taint is valid, returns a dict mapping message NodeTaint to its value; otherwise, raise ArgumentTypeError. For example, { 'key': TAINT_KEY 'value': TAINT_VALUE 'effect': EFFECT } ) NoSchedulePreferNoSchedule NoExecute=z, c3FK|]}tj|ywr)six text_type).0keys r z/ClientBase._parse_node_taint..s)&Is c&Is!z2([a-zA-Z0-9-_]*)=([a-zA-Z0-9-_]*):([a-zA-Z0-9-_]*)z?Node taint [{}] not in correct format, expect KEY=VALUE:EFFECT.z1Invalid taint effect in [{}] , expect one of [{}])rsr(effect)messages NodeTaintEffectValueValuesEnum NO_SCHEDULEPREFER_NO_SCHEDULE NO_EXECUTEjoinsortedkeysrecompile fullmatchrArgumentTypeErrorformatgroups) r node_tainttaint_effect_enumtaint_effect_mappinginput_node_taintvalid_node_taint_effects taint_pattern taint_match taint_key taint_value taint_effects r_parse_node_taintzClientBase._parse_node_taints "!**@@'33-@@&11 xx +#yy)&,-A-F-F-H&I) JJ=M ))*:;K   ) ) K 6" # ,7+=+=+?(I{L//  ) ) = D D 8  &|4 rczt|jddr$|jjjSy)FParses standalone cluster resource argument and returns its reference.standalone_clusterN)rr4rr5r6s r_standalone_cluster_refz"ClientBase._standalone_cluster_refs/t}}2D9 ]] - - 3 3 55 rcJ|j|}|r|jSyz9Parses standalone cluster from args and returns its name.N)rr.rrstandalone_cluster_refs r_standalone_cluster_namez#ClientBase._standalone_cluster_names(!99$? # 0 0 22 rcf|j|}|r|jjSy)z@Parses standalone cluster from args and returns its parent name.N)rrGr.rs r_standalone_cluster_parentz%ClientBase._standalone_cluster_parents1!99$? # * * , 9 9 ;; rcJ|j|}|r|jSy)zAParses standalone cluster from the given args and returns its ID.N)rrJrs r_standalone_cluster_idz!ClientBase._standalone_cluster_ids(!99$? # ( ( ** rczt|jddr$|jjjSy)r membershipN)rr4rr5r6s r"_standalone_cluster_membership_refz-ClientBase._standalone_cluster_membership_refs0t}}lD1 ]] % % + + -- rcJ|j|}|r|jSyr)rr.rs r#_standalone_cluster_membership_namez.ClientBase._standalone_cluster_membership_names*"DDTJ # 0 0 22 rczt|jddr$|jjjSyr[r]r6s r_standalone_node_pool_refz$ClientBase._standalone_node_pool_ref&s0t}}k40 ]] $ $ * * ,,1rcJ|j|}|r|jSyr`)rr.ras r_standalone_node_pool_namez%ClientBase._standalone_node_pool_name+s)2248M  ' ' )) rcJ|j|}|r|jSyre)rrJras r_standalone_node_pool_idz#ClientBase._standalone_node_pool_id4s)2248M    !!rcf|j|}|r|jjSyrh)rrGr.ras r_standalone_node_pool_parentz'ClientBase._standalone_node_pool_parent:s02248M  ! ! # 0 0 22 rc|j|d}|ytjj|}tj|S)zrArErHrKrPrRrTrWrYr^rcrfrirrrrrrrrrrrrrrr r sH65F$6$65+55&*" $5$?$? %6%@%@  1 ; ; !2!#>&7&A&A(9(C(C$5$?$?0A0K0K 1B1L1L!2!-H-H):)D)D#--#---,=,G,G- #-- "+<+F+F" /@/J/J=(9(C(C=rr )r __future__rrrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioperr2googlecloudsdk.generated_clients.apis.gkeonprem.v1r rr rvrpobjectr rrrrs6>&' ,/5\` i=i=r