PdZddlmZddlmZddlmZddlmZddlZddlZddlm Z ddl m Z dd l m Z ddlZ ej Zd Zd Zd Zd ZdZdZdZeeeeeefZedzZedzZedzZedzZedzZedzZedzZedzZ edzZ!edzZ"edzZ#edzZ$edzZ%edzZ&edzZ'ed zZ(ed!zZ)d"Z*d#Z+d$Z,d%Z-ej\ej^Gd&d'e0Z1d(Z2d)Z3Gd*d+ejhZ5Gd,d-ejlZ7Gd.d/ejpZ9Gd0d1ejtZ;Gd2d3e7Z<Gd4d5e9Z=y#e$reZYwxYw)6z8Utilities for wrapping/dealing with a k8s-style objects.)absolute_import)division)print_function)unicode_literalsN)messages) condition) console_attrzserving.knative.devzautoscaling.knative.devzeventing.knative.devzclient.knative.devzcloud.googleapis.comzrun.googleapis.comzrunapps.googleapis.comz/creatorz /locationz /client-namez/client-versionz /descriptionz /launch-stagez/binary-authorizationz /binary-authorization-breakglassz/execution-environmentz/custom-audiencesz/network-interfacesz/container-dependenciesz /acceleratorz/regionsz/multi-region-idz/gclb-domain-namez/threat-detectioncxt|dr |jSt|dr |jStd)z#Metadta class from messages module. ObjectMeta(K8sIoApimachineryPkgApisMetaV1ObjectMeta4Provided module does not have a known metadata class)hasattrr r ValueErrorms ,lib/googlecloudsdk/api_lib/run/k8s_object.pyMetarTs9 Q  <<q<= 5 55IJJcxt|dr |jSt|dr |jStd)z(List Metadta class from messages module.ListMeta&K8sIoApimachineryPkgApisMetaV1ListMetar )rrrrrs rrr]s9 Q  ::q:; 3 33IJJrc$t||i|S)z+Make metadata message from messages module.)r)rargskwargss rMakeMetarfs a$ !& !!rcd}d}|jDcic]}||r|j||}}|di|Scc}w)zProduce an instance of msg_cls, with all sub-messages initialized. Args: msg_cls: A message-class to be instantiated. Returns: An instance of the given class, with all fields initialized blank objects. cH|jrgSt|jSN)repeatedInitializedInstance message_typefields rInstancez%InitializedInstance..Instanceus ~~ i u11 22rc6t|tjSr) isinstancer MessageFieldr"s r IncludeFieldz)InitializedInstance..IncludeFieldzs eX22 33r) all_fieldsname)msg_clsr$r(r#rs rr r ks_3 4 %%' '% e  jj(5/!'  4  s"AceZdZdZdZed2dZed2dZed2dZed2dZ ed2dZ ed3d Z d2d Z d Z d Zd ZedZedZedZedZedZej,dZedZej,dZedZedZedZej,dZedZedZedZedZedZedZ ed Z!e!j,d!Z!ed"Z"d2d#Z#ed$Z$ed%Z%ed&Z&ed'Z'ed(Z(d)Z)ed*Z*d+Z+d,Z,d-Z-d.Z.d/Z/d0Z0d1Z1y)4KubernetesObjectaBase class for wrappers around Kubernetes-style Object messages. Requires subclasses to provide class-level constants KIND for the k8s Kind field, and API_CATEGORY for the k8s API Category. It infers the API version from the version of the client object. Additionally, you can set READY_CONDITION and TERMINAL_CONDITIONS to be the name of a condition that indicates readiness, and a set of conditions indicating a steady state, respectively. ReadyNc"||S|jS)z5Returns the passed str if given, else the class KIND.)KIND)clskinds rKindzKubernetesObject.Kinds#411rc"||S|jS)z=Returns the passed str if given, else the class API_CATEGORY.) API_CATEGORY)r2 api_categorys r ApiCategoryzKubernetesObject.ApiCategorys(3<I9I9IIrc.||Sdj||S)z4Returns the api version with group prefix if exists.z{}/{}format)r2 api_versionr7s r ApiVersionzKubernetesObject.ApiVersions!  >>, 44rcZt||j|}|||||S)a|Produces a wrapped message with only the given spec. It is meant to be used as part of another message; it will error if you try to access the metadata or status. Arguments: spec: messages.Message, The spec to include messages_mod: the messages module kind: str, the resource kind Returns: A new k8s_object with only the given spec. )spec)getattrr4)r2r? messages_modr3r,s rSpecOnlyzKubernetesObject.SpecOnlys,lCHHTN3G wD!< 66rct||j|}|||j|j||S)z;Wraps a template object: spec and metadata only, no status.)r?metadata)r@r4r?rD)r2templaterAr3r,s rTemplatezKubernetesObject.Templates?lCHHTN3G X]]X->->?  rc2|j|}|jt|d|}|j}|j |}t t||} ||_||_||j_ ||||S#t$rY&wxYw)aProduces a new wrapped message of the appropriate type. All the sub-objects in it are recursively initialized to the appropriate message types, and the kind, apiVersion, and namespace set. Arguments: client: the API client to use namespace: str, The namespace to create the object in kind: str, the resource kind api_category: str, the api group of the resource Returns: The newly created wrapped message. _VERSION) r8r=r@MESSAGES_MODULEr4r r3 apiVersionAttributeErrorrD namespace)r2clientrLr3r7r<rArets rNewzKubernetesObject.News ??<0L..!D glD9 :C ch"cn 'CLL sL$ ''    s!B BBct||j|}t||s td||_||_y)Nz*Oops, trying to wrap wrong kind of message)r@r4r&r_m _messages)selfto_wraprAr3r,s r__init__zKubernetesObject.__init__s:lDIIdO4G gw ' C DDDG!DNrc|jS)zReturn the messages module.)rRrSs rMessagesModulezKubernetesObject.MessagesModules >>rcF|jjs tdy)NzThis instance is spec-only.)rQrDrrWs rAssertFullObjectz!KubernetesObject.AssertFullObjects 77   4 55 rc.|jjSrrQrDrWs r IsFullObjectzKubernetesObject.IsFullObjects 77  rcN|j|jjSr)rZrQr3rWs rr3zKubernetesObject.kinds 77<<rcN|j|jjSr)rZrQrJrWs rrJzKubernetesObject.apiVersions 77  rc.|jjSr)rQr?rWs rr?zKubernetesObject.specs 77<<rcN|j|jjSr)rZrQstatusrWs rrbzKubernetesObject.statuss 77>>rcN|j|jjSr)rZrQrDrWs rrDzKubernetesObject.metadatas 77  rc&||j_yrr\rSvalues rrDzKubernetesObject.metadata sDGGrcb|j|jjjSrrZrQrDr+rWs rr+zKubernetesObject.names$ 77    rcZ|j||jj_yrrhres rr+zKubernetesObject.names !DGGrc@|jjtSr) annotationsgetAUTHOR_ANNOTATIONrWs rauthorzKubernetesObject.authors      1 22rc.|jjSr)rDcreationTimestamprWs rcreation_timestampz#KubernetesObject.creation_timestamp s == * **rcb|j|jjjSrrZrQrDrLrWs rrLzKubernetesObject.namespace$s$ 77   % %%rcZ|j||jj_yrrsres rrLzKubernetesObject.namespace)s !&DGGrcb|j|jjjSr)rZrQrDresourceVersionrWs rresource_versionz!KubernetesObject.resource_version.$ 77   + ++rc|j|jjjj dS)N/)rZrQrDselfLinklstriprWs r self_linkzKubernetesObject.self_link3s/ 77   $ $ + +C 00rcb|j|jjjSr)rZrQrDuidrWs rrzKubernetesObject.uid8s$ 77    rcb|j|jjjSr)rZrQrDownerReferencesrWs rownerszKubernetesObject.owners=rxrc&t|jvSr) REGION_LABELlabelsrWs r is_managedzKubernetesObject.is_managedBs 4;; &&rcH|j|jtSr)rZrrrWs rregionzKubernetesObject.regionFs ;;| $$rcb|j|jjjSr)rZrQrD generationrWs rrzKubernetesObject.generationKs$ 77   & &&rc:||jj_yr)rQrDrres rrzKubernetesObject.generationPs"'DGGrc"|jSr) GetConditionsrWs r conditionszKubernetesObject.conditionsTs    rc 2|j|jjr!|jjj}ng}t j ||r|n |j t|jjdd|jS)NobservedGeneration) rZrQrbrr ConditionsREADY_CONDITIONr@r)rSterminal_conditioncs rrzKubernetesObject.GetConditionsXsq ww~~ ''.. # #a a    0d6J6J 4d;   rcv|jt|j|jjSr)rZAnnotationsFromMetadatarRrQrDrWs rrkzKubernetesObject.annotationses) "4>>4773C3C DDrcv|jt|j|jjSr)rZLabelsFromMetadatarRrQrDrWs rrzKubernetesObject.labelsjs) dnndgg.>.> ??rct|dsJ|jr2|j|jvr|j|jSyy)Nr)rrrrWs rready_conditionz KubernetesObject.ready_conditionosH 4* ++ + 4//4??B __T11 22CrcVt|dsJ|jr|jdSy)NrrbrrrWs rreadyzKubernetesObject.readyus2 4* ++ +   ! !( ++rcVt|dsJ|jr|jdSy)NrlastTransitionTimerrWs rlast_transition_timez%KubernetesObject.last_transition_time{s3 4* ++ +   ! !"6 77rcL |j||S#t$r|cYSwxYw)zBChoose the best symbol (if it's in this encoding) or an alternate.)encode UnicodeError)rSbestaltencodings r _PickSymbolzKubernetesObject._PickSymbols, kk( k  js  ##c(|jdS)z6Return a symbol summarizing the status of this object.r)ReadySymbolAndColorrWs r ready_symbolzKubernetesObject.ready_symbols  # # %a ((rctjj}|j|j dd|dfS|jr|j dd|dfSy)zAReturn a tuple of ready_symbol and display color for this object.u….yellowu✔+green)Xred)r GetConsoleAttr GetEncodingrr)rSrs rrz$KubernetesObject.ReadySymbolAndColorsl **,88:H zz   4c8 D     4c8 Dg MM rc|jj|j|j|j|j |j S)N)r3rLr+rrJ)rRObjectReferencer3rLr+rrJrWs rAsObjectReferencez"KubernetesObject.AsObjectReferencesA >> ) ) YY.. YY HH?? * rc|jS)z(Return the actual message we've wrapped.rQrWs rMessagezKubernetesObject.Messages 77Nrc"|jSr)rrWs rMakeSerializablez!KubernetesObject.MakeSerializables <<>rct|jdr|jj|i|St|j|jdz|i|S)NGoogleCloudRunV1Condition Condition)rrRrr@r3)rSrrs r MakeConditionzKubernetesObject.MakeConditionsSt~~:; 5T^^ 5 5t Fv FF =WT^^TYY%< =t Nv NNrcpt|t|r!|j|jk(Sy)NF)r&typer)rSothers r__eq__zKubernetesObject.__eq__s*%d$ \\^u}} .. rcrdjt|jt|jS)Nz{}({}))r;r__name__reprrQrWs r__repr__zKubernetesObject.__repr__s% ??4:..TWW >>rr)NN)2r __module__ __qualname____doc__r classmethodr4r8r=rBrFrOrUrXrZr]propertyr3rJr?rbrDsetterr+rnrqrLrwr}rrrrrrrrkrrrrrrrrrrrrrr)rrr.r.s /22JJ55 77"((<" 6            ?? ! !;;"" 3 3 + + & & '' , , 1 1     , , ' ' % % ' '((      E E @ @ 3 3  , ,  8 8  ) ) O  ?rr.c|jst|j|_t|jjt|jj ddSNkeyrf) key_field value_field)rkrAnnotationsValueKeyValueListAsDictionaryWrapperadditionalPropertiesAdditionalPropertyrArDs rrrsW    ->>@H (// <))<<  rc|jst|j|_t|jjt|jj ddSr)rr LabelsValuerrrrs rrrsR <(446HO (oo** <$$77  rc4eZdZdZdZdZdZdZdZdZ y) LazyListWrapperaWraps a list that does not exist at object creation time. We sometimes have a need to allow access to a list property of a nested message, when we're not sure if all the layers above the list exist yet. We want to arrange it so that when you write to the list, all the above messages are lazily created. When you create a LazyListWrapper, you pass in a create function, which must do whatever setup you need to do, and then return the list that it creates in an underlying message. As soon as you start adding items to the LazyListWrapper, it will do the setup for you. Until then, it won't create any underlying messages. c ||_d|_yr)_create_l)rScreates rrUzLazyListWrapper.__init__sDLDGrcL|jr|j|Strr IndexErrorrSis r __getitem__zLazyListWrapper.__getitem__s ww WWQZ ,rcd|j|j|_||j|<yr)rrrSrvs r __setitem__zLazyListWrapper.__setitem__s% ww dgDGGAJrcJ|jr|j|=ytrrrs r __delitem__zLazyListWrapper.__delitem__s ww ''!* LrcF|jrt|jSy)Nr)rlenrWs r__len__zLazyListWrapper.__len__s ww \ rc~|j|j|_|jj||yr)rrinsertrs rrzLazyListWrapper.inserts* ww dgGGNN1arN) rrrrrUrrrrrr)rrrrs%     rrcTeZdZdZddZdZdZdZdZdZ d Z d Z d Z d Z d Zy)ListAsDictionaryWrapperzWraps repeated messages field with name in a dict-like object. Operations in these classes are O(n) for simplicity. This needs to match the live state of the underlying list of messages, including edits made by others. Nc8||_||_|xsd|_y)aWraps list of messages to be accessible as a read-only dictionary. Arguments: to_wrap: List[Message], List of messages to treat as a dictionary. key_field: attribute to use as the keys of the dictionary filter_func: filter function to allow only considering certain messages from the wrapped list. This function should take a message as its only argument and return True if this message should be included. cy)NTr))_s rz2ListAsDictionaryWrapper.__init__..sTrN)rQ _key_field_filter)rSrTr filter_funcs rrUz ListAsDictionaryWrapper.__init__ sDGDO2>DLrcZ|jD]\}}||k(s |cSt|)z%Implements evaluation of `self[key]`.)itemsKeyError)rSrkitems rrz#ListAsDictionaryWrapper.__getitem__s.::<4 c   3-rc.t||j|t|jD]K\}}t ||j|k(s |j |s t |||j|<y|jj|yr)setattrr enumeraterQr@rrappend)rSrrfindexrs rrz#ListAsDictionaryWrapper.__setitem__st E4??C( ) t t '3 .||D!  *  GGNN5rc|jD];}t||j|k(s|j|s t ||cSt ||j||jj ||Sr)rQr@rrrrr)rSrdefaultrs r setdefaultz"ListAsDictionaryWrapper.setdefault(se t '3 .||D!    GT__c*GGNN7 Nrcd}t|jD]3\}}t||j|k(s |j |r|}n| t ||j|=y)z)Implements evaluation of `del self[key]`.N)rrQr@rrr)rSrindex_to_deleterrs rrz#ListAsDictionaryWrapper.__delitem__2scO ) t t '3 . << !/ *  SM  rcBtd|jDS)z%Implements evaluation of `len(self)`.c3 K|]}dyw)Nr)).0rs r z2ListAsDictionaryWrapper.__len__..@s',Qq,s )sumrrWs rrzListAsDictionaryWrapper.__len__>s '$**,' ''rc0d|jDS)z.Returns a generator yielding the message keys.c3&K|] }|d yw)rNr))r rs rr z3ListAsDictionaryWrapper.__iter__..Ds - DG s)rrWs r__iter__z ListAsDictionaryWrapper.__iter__Bs - --rc|jSrrrWs rrz(ListAsDictionaryWrapper.MakeSerializableFs 77Nrcdjt|jdjd|j DS)Nz{}{{{}}}z, c3FK|]\}}dj||yw)z{}: {}Nr:)r rrs rr z3ListAsDictionaryWrapper.__repr__..Ls ALDAq(//!Q'Ls!)r;rrjoinrrWs rrz ListAsDictionaryWrapper.__repr__Is:    T  ADJJLAA rct|dS)Nnone_key) ListItemsViewrWs rrzListAsDictionaryWrapper.itemsOs  ++rct|Sr)ListValuesViewrWs rvalueszListAsDictionaryWrapper.valuesRs $ r)r+N)rrrrrUrrrrrrrrrrr)rrrrs> 3 !(. , rrc.eZdZdZddfd ZdZxZS)rz*Item iterator for ListAsDictionaryWrapper.Nrc2t||i|||_yr)superrU _none_key)rSrrr __class__s rrUzListItemsView.__init__Ys Gd%f%DNrc#K|jjD]R}|jj|st||jj}| |j }||fTywr)_mappingrQrr@rr!)rSrrs rrzListItemsView.__iter__]sZ      t $dDMM445 ;#Dk !s 5A/7A/)rrrrrUr __classcell__r"s@rrrVs2%)rrceZdZdZdZy)rc2t|D] }||k(s yy)NTF)iter)rSrfrs r __contains__zListValuesView.__contains__hs $Z e rc#XK|jjD] \}}| ywr)r$r)rSrrfs rrzListValuesView.__iter__ns&MM'')5 k*s(*N)rrrr*rr)rrrrfs  rrcDeZdZdZ dfd ZfdZfdZdZxZS)raWraps repeated messages field with name and value in a dict-like object. Properties which resemble dictionaries (e.g. environment variables, build template arguments) are represented in the underlying messages fields as a list of objects, each of which has a name and value field. This class wraps that list in a dict-like object that can be used to mutate the underlying fields in a more Python-idiomatic way. cNtt| |||||_||_y)a6Wrap a list of messages to be accessible as a dictionary. Arguments: to_wrap: List[Message], List of messages to treat as a dictionary. item_class: type of the underlying Message objects key_field: attribute to use as the keys of the dictionary value_field: attribute to use as the values of the dictionary filter_func: filter function to allow only considering certain messages from the wrapped list. This function should take a message as its only argument and return True if this message should be included. )rrN)r rrU _item_class _value_field)rSrT item_classrrrr"s rrUz(KeyValueListAsDictionaryWrapper.__init__}s3& )499+:"D#Drcxtt| ||j}t ||j |y)a9Implements evaluation of `self[key] = value`. Args: key: value of the key field value: value of the value field Raises: KeyError: if a message with the same key value already exists, but is hidden by the filter func, this is raised to prevent accidental overwrites N)r rrr.rr/)rSrrfrr"s rrz+KeyValueListAsDictionaryWrapper.__setitem__s9 0$ B T    D D$##U+rc|jdi|j|i}tt|||}t ||jS)Nr))r.r/r rrr@)rSrr default_itemrr"s rrz*KeyValueListAsDictionaryWrapper.setdefaultsO#4##Ct'8'8'&BCL 0$ B \ D 4** ++rct|Sr)KeyValueListItemsViewrWs rrz%KeyValueListAsDictionaryWrapper.itemss  &&r)r+rfN) rrrrrUrrrr%r&s@rrrss' $2,",'rrceZdZfdZxZS)r5c#Ktt| D])\}}|t||jj f+ywr)r r5rr@r$r/)rSrrr"s rrzKeyValueListItemsView.__iter__s=0$@B T '$ : :; <<CsAA)rrrrr%r&s@rr5r5s ==rr5)>r __future__rrrrabc collectionsapitools.base.protorpclitergooglecloudsdk.api_lib.runrgooglecloudsdk.core.consoler sixcollections_abcrK SERVING_GROUPAUTOSCALING_GROUPEVENTING_GROUP CLIENT_GROUP GOOGLE_GROUP RUN_GROUP RUNAPPS_GROUPINTERNAL_GROUPSrmrCLIENT_NAME_ANNOTATIONCLIENT_VERSION_ANNOTATIONDESCRIPTION_ANNOTATIONLAUNCH_STAGE_ANNOTATIONBINAUTHZ_POLICY_ANNOTATIONBINAUTHZ_BREAKGLASS_ANNOTATION EXECUTION_ENVIRONMENT_ANNOTATIONCUSTOM_AUDIENCES_ANNOTATIONNETWORK_INTERFACES_ANNOTATION!CONTAINER_DEPENDENCIES_ANNOTATIONGPU_TYPE_NODE_SELECTORMULTI_REGION_REGIONS_ANNOTATIONMULTI_REGION_ID_LABELGCLB_DOMAIN_NAME_ANNOTATIONTHREAT_DETECTION_ANNOTATIONrrrr add_metaclassABCMetaobjectr.rrMutableSequencerMutableMappingr ItemsViewr ValuesViewrrr5r)rrr^s ?&%' /04  OO/ & -'# % (  "J.k) "^3%(99"^3#o5&)@@!*-O!O#,/G#G '*== ),A A$-0I$I!"^3"+j"8!$66+.AA'*==KK" 43;;t?vt? t?n ,o55,^Q o<<Q h O--  _// <'&=<'~=M=c / s EE#"E#