FSrSSKJr SSKJr SSKJr SSKJr SSKJr SSKrSSKJ r J r SS K J r \Rr"S S \ R 5rS rSS jrSrSrSrSrSr"SS\R25rg!\a \rNKf=f)z.Wrapper for Cloud Run TrafficTargets messages.)absolute_import) annotations)division)print_function)unicode_literalsN) ContainerMapping) exceptionsc\rSrSrSrSrg) InvalidTrafficSpecificationError"z3Error to indicate an invalid traffic specification.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r)lib/googlecloudsdk/api_lib/run/traffic.pyr r "s;rr LATESTc^U[:XaURSX#S9nU$URXUS9nU$)aNCreates a new TrafficTarget. Args: messages: The message module that defines TrafficTarget. key: The key for the traffic target in the TrafficTargets mapping. percent: Optional percent of traffic to assign to the traffic target. tag: Optional tag to assign to the traffic target. Returns: The newly created TrafficTarget. T)latestRevisionpercenttag) revisionNamerr)LATEST_REVISION_KEY TrafficTarget)messageskeyrrresults rNewTrafficTargetr"-sJ   # #W$F - # #3 # OF -rcHUR(a[$UR$)zReturns the key for a TrafficTarget. Args: target: TrafficTarget, the TrafficTarget to check Returns: LATEST_REVISION_KEY if target is for the latest revison or target.revisionName if not. )rrr)targets rGetKeyr%Bs!' 5 5 N6;N;NNrc.U[:XaSU4nU$SU4nU$)ahSorted key function to order TrafficTarget keys. TrafficTargets 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 TrafficTargets dictionary. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )r)r r!s rSortKeyFromKeyr)Os-" XF -XF -rc.[U5n[U5$)zSorted key function to order TrafficTarget objects by key. Args: target: A TrafficTarget. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )r%r))r$r s rSortKeyFromTargetr+gs v#  rcUR(a UROSnUR(a UROSnX4$)zBKey function for sorting TrafficTarget objects during __getitem__.r)rr)r$rrs r_GetItemSortKeyr.us. %nnFNN!'jj b# rcDUupSU[U5- - U[U5/$)aReturns object that sorts in the order we correct traffic rounding errors. The caller specifies explicit traffic percentages for some revisions and this module scales traffic for remaining revisions that are already serving traffic up or down to assure that 100% of traffic is assigned. This scaling can result in non integrer percentages that Cloud Run does not supprt. We correct by: - Trimming the decimal part of float_percent, int(float_percent) - Adding an extra 1 percent traffic to enough revisions that have had their traffic reduced to get us to 100% The returned value sorts in the order we correct revisions: 1) Revisions with a bigger loss due are corrected before revisions with a smaller loss. Since 0 <= loss < 1 we sort by the value: 1 - loss. 2) In the case of ties revisions with less traffic are corrected before revisions with more traffic. 3) In case of a tie revisions with a smaller key are corrected before revisions with a larger key. Args: key_and_percent: tuple with (key, float_percent) Returns: An value that sorts with respect to values returned for other revisions in the order we correct for rounding errors. r()intr))key_and_percentr float_percents rNewRoundingCorrectionPrecedencer3s38'#=3}- -.S rc\rSrSrSrSrSrSrSrSr Sr S r \ S 5r S rS rS rSrSrSrSrSrSrSrSrSSjrSrSrSSjrSrg)TrafficTargetsaWWraps a repeated TrafficTarget message and provides dict-like access. The dictionary key is one of LATEST_REVISION_KEY for the latest revision TrafficTarget.revisionName for TrafficTargets with a revision name. The dictionary value is a list of all traffic targets referencing the same revision, either by name or the latest revision. cRXlX lURRUlg)aConstructs a new TrafficTargets instance. The TrafficTargets 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 TrafficTarget. to_wrap: The traffic targets to wrap. N) _messages_mr_traffic_target_cls)selfmessages_moduleto_wraps r__init__TrafficTargets.__init__s %NG#~~;;Drcp^[U4SjUR5[S9nU(d [T5eU$)avGets a sorted list of traffic targets associated with the given key. Allows accessing traffic targets 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 traffic targets to support comparison operations on TrafficTargets objects which should be independent of the order of the traffic targets for a given key. Args: key: A revision name or "LATEST" to get the traffic targets for. Returns: A sorted list of traffic targets associated with the given key. Raises: KeyError: If this object does not contain the given key. c3J># UHn[U5T:XdMUv M g7fNr%).0tr s r -TrafficTargets.__getitem__..s0GqvayC/Gs# #r )sortedr9r.KeyError)r;r r!s ` r __getitem__TrafficTargets.__getitem__s1(0DGG0oF  SM MrchURVs/sHn[U5U:wdMUPM sn$s snf)z1Gets all targets that do not match the given key.r9r%)r;r rEs r _OtherTargetsTrafficTargets._OtherTargetss)ww 3w!&)s"2Aw 33 3s//cX;aURRU5 gURU5U-URSS&g)z/Implements evaluation of `self[key] = targets`.N)r9extendrO)r;r new_targetss r __setitem__TrafficTargets.__setitem__s4  ggnn[!%%c*[8dggajrc rURU5nU(akUVs/sH4nUR(dM[URXRS9PM6 nnUR [URX55 XPU'gUR R [URX55 gs snf)zSet the given percent in the traffic targets. Moves any tags on existing targets with the specified key to zero percent targets. Args: key: Name of the revision (or "LATEST") to set the percent for. percent: Percent of traffic to set. rN)getrr"r8appendr9)r;r rexistingrErSs r SetPercentTrafficTargets.SetPercentsxx}Ha UU ; 4>>3EE : )$..#GH3i ggnn%dnncCDs B4"B4c`X;a [U5eURU5URSS&g)z)Implements evaluation of `del self[key]`.N)rJrOr9)r;r s r __delitem__TrafficTargets.__delitem__s)  SM##C(DGGAJrcLURHnU[U5:XdM g g)z(Implements evaluation of `item in self`.TFrN)r;r r$s r __contains__TrafficTargets.__contains__s$'' v  rc:[SUR55$)z$A set containing the mapping's keys.c38# UHn[U5v M g7frBrC)rDrEs rrF*TrafficTargets._key_set.. s*'Qvayy')setr9r;s r_key_setTrafficTargets._key_sets *$''* **rc,[UR5$)z%Implements evaluation of `len(self)`.)lenrirhs r__len__TrafficTargets.__len__ s t}} rc,[UR5$)z1Returns an iterator over the traffic target keys.)iterrirhs r__iter__TrafficTargets.__iter__s  rcUR$rB)r9rhs rMakeSerializableTrafficTargets.MakeSerializables 77NrcXSRSUR555nSU-$)Nz, c3H# UHupSRX5v M g7f)z{}: {}N)format)rDkvs rrF*TrafficTargets.__repr__..sG,$!--,s "z[%s])joinitems)r;contents r__repr__TrafficTargets.__repr__s'iiG$**,GGG G rc v/n0nURHn[U5nUR(a-UR[ UR XCRS95 UR (dM_URU[ UR US55=R UR - slM X!4$)aReturns normalized targets, split into percent and tags targets. Moves all tags to 0% targets. Combines all targets with a non-zero percent that reference the same revision into a single target. Drops 0% targets without tags. Does not modify the underlying repeated message field. Returns: A tuple of (percent targets, tag targets), where percent targets is a dictionary mapping key to traffic target for all targets with percent greater than zero, and tag targets is a list of traffic targets with tags and percent equal to zero. rWr)r9r%rrYr"r8r setdefault)r; tag_targetspercent_targetsr$r s r_GetNormalizedTraffic$TrafficTargets._GetNormalizedTrafficsKO'' 6Nc  T^^Sjj A  "" !$..#q9   'V^^ $   ''rcSnUHnX#R- nM US:wa[SU-5eUH:nURS:dM[S[U5<SUR<S35e g)z!Validate current traffic targets.rdz3Current traffic allocation of %s is not 100 percentzCurrent traffic for target z is negative ()N)r ValueErrorr%)r;existing_percent_targetsrr$s r_ValidateCurrentTraffic&TrafficTargets._ValidateCurrentTraffic7srG* g+#~  ?' I + ! f~v~~ /  +rc0nURH,n[U5nUR(dM!XA;dM(X2U'M. U$)z7Get TrafficTargets with traffic not in new_percentages.)r9r%r)r;new_percentagesr!r$r s r_GetUnassignedTargets$TrafficTargets._GetUnassignedTargetsIs: F'' 6Nc C6s  Mrc[UR55nUS:a [S5eUH)nXS:d XS:dM[SU<SX<S35e U(dUS:a [S5egg) z8Validate the new traffic percentages the user specified.rz"Over 100% of traffic is specified.rzNew traffic for target z is z, not between 0 and 100zPEvery target with traffic is updated but 100% of traffic has not been specified.N)sumvaluesr )r;runspecified_targetsspecified_percentr s r_ValidateNewPercentages&TrafficTargets._ValidateNewPercentagesRsO22453 , .    !_%9C%?.O( *   #4s#: , , $; rc>[UR55nSU- $)zAReturns percentage of traffic not explicitly specified by caller.r)rr)r;rrs r_GetPercentUnspecifiedTraffic,TrafficTargets._GetPercentUnspecifiedTrafficgs"O2245 " ""rc@UVs0sHo"[X5_M nn[[[UR5555[UR55- n[ UR 5[ S9nUSUHupgX6==S- ss'M U$s snf)z$Returns rounded integer percentages.rHNr()r0roundrrrIr}r3)r;float_percentagesryrounded_percentageslosscorrection_precedencer _s r_IntPercentagesTrafficTargets._IntPercentagesls/@.?3 # $$.? uS*11345 6""$: D#!'F(.!#/ sBcURU5nUS:Xa0$[SUR555n0nUH"nX&R[ U5-U- XV'M$ U$)Nrc38# UHoRv M g7frB)r)rDr$s rrF9TrafficTargets._GetAssignedPercentages..s!%@6%@rf)rrrrfloat)r;runassigned_targetspercent_to_assignpercent_to_assign_fromassigned_percentagesrys r_GetAssignedPercentages&TrafficTargets._GetAssignedPercentages{s::?KA i !%7%>%>%@!    ' ' # $ % " # rcUR5up#URUR55 UR5nUR U5nUR XE5 UR URXE55 URU5n/nUHOnX;a XS:XaMX;aX(n XhU l O[URXU5n URU 5 MQ [U[S9nURSS2 URR!U5 URR!U5 g)aUpdate traffic percent assignments. The updated traffic percent assignments will include assignments explicitly specified by the caller. If the caller does not assign 100% of traffic explicitly this function will scale traffic for targets the user does not specify with an existing percent greater than zero up or down based on the provided assignments as needed. This method normalizes the traffic targets while updating the traffic percent assignments. Normalization merges all targets referencing the same revision without tags into a single target with the combined percent. Normalization also moves any tags referencing a revision to zero percent targets. The update removes targets with 0% traffic unless: o The user explicitly specifies under 100% of total traffic o The user does not explicitly specify 0% traffic for the target. o The 0% target has a tag. Args: new_percentages: Map from revision to percent traffic for the revision. 'LATEST' means the latest rev. Raises: ValueError: If the current traffic for the service is invalid. InvalidTrafficSpecificationError: If the caller attempts to set the traffic for the service to an incorrect state. rrHN)rrrcopyrrupdaterrrr"r8rYrIr+r9rR) r;rrrupdated_percentagesrint_percentagesnew_percent_targetsr r$s r UpdateTrafficTrafficTargets.UpdateTraffics):-1,F,F,H)  !9!@!@!BC)..0334GH  !4I $$%8M**+>?O O$8A$=  *).(-!$..#s7KL  (!!4:KL  GGNN&'GGNN;rcTUR5up#[U;aUR[5nX;a"X!=RUR- slO"[ UR XR5X!'[ UR5[S9nXS-URSS&gg)z;Reasign traffic from LATEST to the current latest revision.rHN) rrpoprr"r8rIrr+r9)r;latest_ready_revision_namerrlatestsorted_percent_targetss rZeroLatestTraffic TrafficTargets.ZeroLatestTraffics#'#=#=#? Oo-""#67f # 63;;v~~M;6F NN67 3 &  "(9 *7dggaj.rcURVs0sH+oR(dMUR[U5_M- sn$s snfrB)r9rr%)r;r$s rTagToKeyTrafficTargets.TagToKeys259WW KW6 &FJJv &W KK Ks AAc /nUR(d)[UR[S5/URSS&URHfnU(d URU;dURU;aSUlUR (dUR(dMUUR U5 Mh [UR55H(upgUR [URXvS95 M* X@RSS&g)aUpdate traffic tags. Removes and/or clears existing traffic tags as requested. Always adds new tags to zero percent targets for the specified revision. Treats a tag update as a remove and add. Args: to_update: A dictionary mapping tag to revision name or 'LATEST' for the latest ready revision. to_remove: A list of tags to remove. clear_others: A boolean indicating whether to clear tags not specified in to_update. rNrW) r9r"r8rrrrYrIr})r; to_update to_remove clear_othersrSr$r revision_keys r UpdateTagsTrafficTargets.UpdateTagss&K 77$T^^5H#NOdggaj'' y0FJJ)4K 6:::6"  $IOO$56 4>>< A7GGAJr)r9r8r:N)rzMapping[str, int])rzMapping[str, str]rzContainer[str]rbool)rrrrrr>rKrOrTr[r^rapropertyrirmrqrtrrrrrrrrrrrrrrrrr5r5s< 649E,)  + +(6 $*#  84 l8 L""" " "rr5)NN)r __future__rrrrr collectionscollections.abcrr googlecloudsdk.corer abccollections_abcAttributeErrorErrorr rr"r%r)r+r.r3MutableMappingr5rrrrs5&"%'.* OO/ z'7'7* O0 !H^_33^M / s BB  B