\ dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZddlZdZdZdZGddej6ZGddeZeedeieedZGddej>dgdZ ee ddieeiZ!Gdde jDZ#d Z$d!Z%Gd"d#eZ&d$Z'y)%z completers for resource library.)absolute_import)division)unicode_literalsN)messages)resource) completers) arg_utils)registry)resource_parameter_info) exceptions)log) properties) resourcesnamecloudresourcemanager.projects projectIdceZdZdZy)Errorz!Base error class for this module.N__name__ __module__ __qualname____doc__:lib/googlecloudsdk/command_lib/util/concepts/completers.pyrr*s)rrc*eZdZdZddZdZdZdZy)ParentTranslatora>Translates parent collections for completers. Attributes: collection: str, the collection name. param_translation: {str: str}, lookup from the params of the child collection to the params of the special parent collection. If None, then the collections match and translate methods are a no-op. Nc(||_|xsi|_yN) collectionparam_translation)selfr!r"s r__init__zParentTranslator.__init__8s DO.4"Drc|jr>tj|jD]\}}|j|||<||=|S)zETranslate from original parent params to params that match the child.)r"six iteritemsget)r#params orig_param new_params r ToChildParamszParentTranslator.ToChildParams<sL #&==1G1G#H *i#ZZ 2z 9 $I Mrci}tj|jD]'\}} |j|t ||d||<)|S#t$rY7wxYw)zCGet dict for translating parent params into the given message type.N)r&r'r" field_by_nameKeyErrorgetattr)r#messagerefmessage_resource_mapr* special_params rMessageResourceMapz#ParentTranslator.MessageResourceMapDsn&)]]43I3I%J! Mj) *1mT)J:&&K sA AAcr|Dcic].}|jj|||j|0}}tj|D]<\}}|jj||}|s%|j|dr8|||<> t j t jt jj|jd|dScc}w#t j$r@}tjtj|jYd}~yd}~wwxYw)aParse the parent resource from parameter info and aggregations. Args: parent_params: [str], a list of params in the current collection's parent collection. parameter_info: the runtime ResourceParameterInfo object. aggregations_dict: {str: str}, a dict of params to values that are being aggregated from earlier updates. Returns: resources.Resource | None, the parsed parent reference or None if there is not enough information to parse. Ncollection_info subcollection param_values endpoint_url)r"r(GetValuer&r'rResourceREGISTRYGetCollectionInfor!rr info text_typerstrip) r# parent_paramsparameter_infoaggregations_dictpr;valuetranslated_namees rParsezParentTranslator.ParseSs  A ""1a(.*A*A!*DD MM"345..221a8o |''>(- _%5       #,,>>tO#  " ?? hhs}}Q&&() s3CAC##D666D11D6r )rrrrr$r,r5rKrrrrr.s5  !rr projectsId)rLrceZdZdZy)CollectionConfigzAStores data about special collections for configuring completion.NrrrrrNrNsJrrN) static_paramsid_field param_namefilterzlifecycleState:ACTIVEceZdZdZ dfd ZedZdZdZdZ dZ dZ dd Z dd Z d Z dd Zdd ZddZdZ ddZdZxZS)ResourceArgumentCompleterz?A completer for an argument that's part of a resource argument.c ||_||_|xsi|_|xst|_|j }|j } tt|&d|| |dd|y)z Initializes.T)r! api_versionparam parse_allNr) resource_spec_method_static_paramsDEFAULT_ID_FIELDrP full_namerVsuperrTr$) r#rYr9methodrOrPrWkwargscollection_namerV __class__s rr$z"ResourceArgumentCompleter.__init__so'DDL'-2D0 0DM%//O!--K #T3"   rc|jS)zGets the list method for the collection. Returns: googlecloudsdk.command_lib.util.apis.registry.APIMethod, the method. )rZr#s rr_z ResourceArgumentCompleter.methods <<rc>|jjdddS)z(Get the parent params of the collection.r7N)r9 GetParamsrds r _ParentParamsz'ResourceArgumentCompleter._ParentParamss    ) )" -cr 22rc.|jjdd}t|jjD]1\}}|jj |j |k(s/|}nd}i}t|jjd|D]z\}}t|j|j }|r+|df||jj |j <Sd||jj |j <||S)z*Helper function to build dict of updaters.r7rfrNT)NF)r9rg enumeraterY attributes ParamNamerCompleterForAttribute)r# final_parami attribute attribute_idxupdaters completers r _GetUpdatersz&ResourceArgumentCompleter._GetUpdaterss&&004R8K!$"4"4"?"?@ 9   % %inn 5 D  A mH! %%n}57 9'(:(:INNKi BKBFBH##--inn=>BI##--inn=>7 Orc|jj|j}|j}t j |||||j S)zBuilds a ResourceParameterInfo object. Args: parsed_args: the namespace. argument: unused. Returns: ResourceParameterInfo, the parameter info for runtime information. )rrr!)CONCEPTSArgNameToConceptInfodestrtr ResourceParameterInfor!)r# parsed_argsargument resource_inforrs r ParameterInfoz'ResourceArgumentCompleter.ParameterInfosO ((==hmmLM  "H " 8 8{Hx?? $$rct|jddDcgc]}|jc}}t|Dcgc]}|jc}}|jjD]U}t |j|js$|j |jj|jW|j|Scc}wcc}w)zEValidates that parent attributes values exitst before making request.Nrf) set parametersrrYrkrmaddattribute_to_params_mapissubset)r# aggregationsrGparameters_needing_resolutionaresolved_parametersrps rValidateAttributeSourcesz2ResourceArgumentCompleter.ValidateAttributeSourcess$'"9M(N9MA9M(N$O!|<|!qvv|<=''22 t119>> B    6 6y~~ F H3 ) 1 12E FF)OrEt/dj t1|t)j*|j-t/dj |t1|t)j*|j-d}~wwxYwcc} w) Nz9Cache query parameters={} aggregations={}resource info={}r7)parent_translatorz0Could not build query to list completions: {} {})rErrzcache items={}zVUpdate query failed, may not have enough information to list existing resources: {} {}zUpdate query [{}]: {} {})!r_rr rAformatr9rgr=rrHr|attribute_to_args_map_GetParentTranslatorBuildListQuery ExceptionrVALUEScoreprint_completion_tracebacksGetBoolr&rBrCrtypeCallr!_ParseResponse RelativeName isinstancerValidationError StringToRow) r#rErrGrqueryrJresponseresponse_collectionritemsroitems rUpdatez ResourceArgumentCompleter.Updates {{   ( ( 6 HH !6**44R8 :8((+ ,8 :(4 5 1affagg  5  ( ( > > @A11.,O/!!.,4E"Ge6!!%(h KK22 ! "!1 ""1&92@0<5F#H!  "  hh&&%* +U1>> U +-.?D DedD  T..0 1e DDK : 5 /      ; ; C C E  hhs}}Q&&() DKK q'3==#**,. // /" ,  6      ; ; C C E  hhs}}Q&&() Ax// 055;VQq!1!8!8!:6<= = ,33 a#--*1135 66 6 EsbF 7F$ F)5I2I(0 I2I- ' I26&N) I%2B.I  I%( I22 N&&004R8K T*2"(l; QsB!ctjtj|jd|d}|j S)z9Parse the parent for a resource using default collection.r7Nr8)rr>r?r9Parent)r#r;rs rrz-ResourceArgumentCompleter._ParseDefaultParents<!!,,! H ?? rc|j|}|j||} |s|j|S|j|j ||S#t j $r@}tjtj|jYd}~yd}~wt j$r@}tjtj|jYd}~yd}~wwxYw)aGets the parent reference of the parsed parameters. Args: parameter_info: the runtime ResourceParameterInfo object. aggregations: a list of _RuntimeParameter objects. parent_translator: a ParentTranslator for translating to a special parent collection, if needed. Returns: googlecloudsdk.core.resources.Resource | None, the parent resource or None if no parent was found. rN) rrrrKrhrrr rAr&rBrCr)r#rErrrFr;rJs rrz#ResourceArgumentCompleter.GetParents77 E33*;4=L  '' 55  $ $T%7%7%9>%688  8 8 hhs}}Q&&()  ?? hhs}}Q&&() s(A AC>-6B((C>>6C99C>ct|tsy|j|jk(xr4|j|jk(xr|j|jk(S)z Overrides.F)rrTrYr!r_)r#others r__eq__z ResourceArgumentCompleter.__eq__sW e6 7    %"5"5 5 ( OOu// / ( KK5<< ')r)NNN)NNr )rrrrr$propertyr_rhrtr}rrrrrrrrrrr __classcell__)rbs@rrTrTsG9=   30$$ G/Ed8<'+M"DH26 9='+B( 48"&>)rrTc|j}tj|}|y||jdk(r |jS|jj |}|j }tj||j}|jd|dz}|D]}|j|k(s|jcSy)z3Gets the collection for an attribute in a resource.Nrf) _collection_infor APICollectionrkrindexapi_nameGetAPICollectionsrVr) rYrpresource_collection_inforesource_collectionrqrresource_collectionsr)cs r_MatchCollectionrs*;; ..  -**2..  # ##**00;- % . .(!33**,  . ./A 0A B& aF" VVm rcd}t||}|r3|jjdz|z}|jj}n|jdk(rd}nyt j j||S)z4Gets collection info for an attribute in a resource.NrprojectrrV)rrrrVrrr?r@)rYrprVr!full_collection_names r_GetCompleterCollectionInfors+ y9* &&//#5 B00< !  &  1 8 8 H JJ((8J77M,,H1-KO  $<<-11/2K2KLl!!/";";VDf"00m&&h**j  =(D$ KK!!  # #V%113f, /6# %%'  & &  # #'; ;''--c226*D0334HI ##$8&A$22 (( >>fs;,D66BG$ G$#G$c|jS)z1Get the APIMethod for an attribute in a resource.)r_rds rrzCompleterInfo.GetMethod)s ;;r)NNNNN)rrrrr$ classmethodrrrrrrrs*HHL'+!;%;%zrrcTGfddtjsyS)z.Completerz5A specific completer for this attribute and resource.ctj||}t|||j|j f|j |j|jd|y)N)rOrPrW) rrr^r$r9r_rOrPrQ)r#rYrr`completer_info Completerrbs rr$z1CompleterForAttribute..Completer.__init__4se$11-Pn It%   ( (   '44!**))  rc^ttjjS)z:Checks whether the completer is valid (has a list method).)boolrrr)rrrYs rvalidatez1CompleterForAttribute..Completer.validateAs-  $ $]N C M M OQQr)rrrrr$rrr)rbrrrYs@rrr1s'?%2 . QQrrN)rTr)rYrrs``@rrmrm.s/QQ+Q,      r)(r __future__rrr collectionsapitools.base.protorpclitergooglecloudsdk.api_lib.utilr resource_libgooglecloudsdk.command_lib.utilr$googlecloudsdk.command_lib.util.apisr r (googlecloudsdk.command_lib.util.conceptsr googlecloudsdk.corer r rrr&r\r_PROJECT_ID_FIELDrobjectrr namedtuplerNrResourceCompleterrTrrrrmrrrrs '&'/@6:9L*#*) 6*J  *FvF\##7$02C#DF!"679 J-{--  J**H6M+N+<+<>? [) < <[)|(G&KFK\r