dZddlmZddlmZddlmZddlmZddlmZddlZddlm Z dd l m Z  ejZ Gd d e jZd Zddd ZddZddZddZddZGdde j.Zy#e$reZ YKwxYw)z-Wrapper for Cloud Run InstanceSplit messages.)absolute_import) annotations)division)print_function)unicode_literalsN) exceptions)run_v1_messagesceZdZdZy)&InvalidInstanceSplitSpecificationErrorz:Error to indicate an invalid instance split specification.N)__name__ __module__ __qualname____doc__0lib/googlecloudsdk/api_lib/run/instance_split.pyr r "sBrr LATESTch|tk(r|jd|}|S|j||}|S)a)Creates a new InstanceSplit. Args: messages: The message module that defines InstanceSplit. key: The key for the instance split assignment in the InstanceSplits mapping. percent: Optional percent of instance split to assign. Returns: The newly created InstanceSplit. T)latestRevisionpercent) revisionNamer)LATEST_REVISION_KEY InstanceSplit)messageskeyrresults rNewInstanceSplitr-sE   # #4 # IF - # #g # FF -rc>|jrtS|jS)zReturns the key for a InstanceSplit. Args: split: InstanceSplit, the InstanceSplit to check Returns: LATEST_REVISION_KEY if split is for the latest revison or split.revisionName if not. )rrr)splits rGetKeyr @s!& 4 4 L%:L:LLrc,|tk(rd|f}|Sd|f}|S)a}Sorted key function to order InstanceSplit keys. InstanceSplits keys are one of: o revisionName o LATEST_REVISION_KEY Note LATEST_REVISION_KEY is not a str so its ordering with respect to revisionName keys is hard to predict. Args: key: Key for a InstanceSplits dictionary. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )r)rrs rSortKeyFromKeyr$Ms-" XF -XF -rc.t|}t|S)zSorted key function to order InstanceSplit objects by key. Args: split: A InstanceSplit. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )r r$)rrs rSortKeyFromSplitr&es u #  rc>|jr|j}|Sd}|S)zBKey function for sorting InstanceSplit objects during __getitem__.r)r)rrs r_GetItemSortKeyr(ss$ #]]EMM' .12' .rc\eZdZdZdZdZdZdZdZe dZ dZ d Z d Z d Zd Zy )InstanceSplitsanWraps a repeated InstanceSplit message and provides dict-like access. The dictionary key is one of LATEST_REVISION_KEY for the latest revision InstanceSplit.revisionName for InstanceSplits with a revision name. The dictionary value is a list of all instance split assignments referencing the same revision, either by name or the latest revision. cV||_||_|jj|_y)aConstructs a new InstanceSplits instance. The InstanceSplits instance wraps the to_wrap argument, which is a repeated proto message. Operations that mutate to_wrap will usually occur through this class, but that is not a requirement. Callers can directly mutate to_wrap by accessing the proto directly. Args: messages_module: The message module that defines InstanceSplit. to_wrap: The instance split assignments to wrap. N) _messages_mr_instance_split_cls)selfmessages_moduleto_wraps r__init__zInstanceSplits.__init__s$%DNDG#~~;;Drcjtfd|jDt}|s t|S)aGets a sorted list of instance split assignments associated with the given key. Allows accessing instance split assignments based on the revision they reference (either directly by name or the latest ready revision by specifying "LATEST" as the key). Returns a sorted list of instance split assignments to support comparison operations on InstanceSplits objects which should be independent of the order of the instance split assignments for a given key. Args: key: A revision name or "LATEST" to get the instance split assignments for. Returns: A sorted list of instance split assignments associated with the given key. Raises: KeyError: If this object does not contain the given key. c3@K|]}t|k(s|ywNr ).0irs r z-InstanceSplits.__getitem__..s0GqvayC/Gs)r)sortedr-r(KeyError)r/rrs ` r __getitem__zInstanceSplits.__getitem__s1.0DGG0oF  SM MrctzNot implemented for now.NotImplementedError)r/rs r __delitem__zInstanceSplits.__delitem__  rctr>r?)r/r new_splitss r __setitem__zInstanceSplits.__setitem__rBrcF|jD]}|t|k(syy)z(Implements evaluation of `item in self`.TF)r-r )r/rrs r __contains__zInstanceSplits.__contains__s$ u  rc:td|jDS)z$A set containing the mapping's keys.c32K|]}t|ywr5r6)r7r8s rr9z*InstanceSplits._key_set..s*'Qvay's)setr-r/s r_key_setzInstanceSplits._key_sets *$''* **rc,t|jS)z%Implements evaluation of `len(self)`.)lenrLrKs r__len__zInstanceSplits.__len__s t}} rc,t|jS)z.s G,$!Q1-,s!z[%s])joinitems)r/contents r__repr__zInstanceSplits.__repr__s'iiG$**,GGG G rcd|jjt|j||y)zAdd a new instance split assignments for the given key. Mainly for testing. Args: key: Name of the revision (or "LATEST") to set the percent for. percent: Percent of instance split to set. N)r-appendrr,)r/rrs rAddSplitzInstanceSplits.AddSplits" GGNN#DNNCABrN)r r rrr2r<rArErGpropertyrLrOrRrTr]r`rrrr*r*}sN< <   + + Crr*r5)rstr)rzrun_v1_messages.InstanceSplit)r __future__rrrrr collectionsgooglecloudsdk.corer,googlecloudsdk.generated_clients.apis.run.v1r abccollections_abcAttributeErrorErrorr rrr r$r&r(MutableMappingr*rrrrls4&"%'*H OO/ Z-=-=& M0 eC_33eC / s BB  B