"<dZddlmZddlmZddlmZddlmZddlZddlmZddl Z dZ d Z d Z d Z Gd d eZGddeZdZdZ ddZy)z.Wrapper for Cloud Run TrafficTargets messages.)absolute_import)division)print_function)unicode_literalsN)traffic-z, cB|tk(rtSdj|S)Nz{}%)_MISSING_PERCENT_OR_TAGSformatpercents .lib/googlecloudsdk/api_lib/run/traffic_pair.py_FormatPercentager#s  (( ## <<  c&td|DS)z'Sums the percents of the given targets.c3NK|]}|js|jywNr .0ts r z_SumPercent..,s 51199QYY%%)sum)targetss r _SumPercentr*s 5 5 55rceZdZdZddZy) TrafficTagaContains the spec and status state for a traffic tag. Attributes: tag: The name of the tag. url: The tag's URL, or an empty string if the tag does not have a URL assigned yet. Defaults to an empty string. inSpec: Boolean that is true if the tag is present in the spec. Defaults to False. inStatus: Boolean that is true if the tag is present in the status. Defaults to False. c<||_||_||_||_y)zReturns a new TrafficTag. Args: tag: The name of the tag. url: The tag's URL. in_spec: Boolean that is true if the tag is present in the spec. in_status: Boolean that is true if the tag is present in the status. N)tagurlinSpecinStatus)selfrr in_spec in_statuss r__init__zTrafficTag.__init__<s DHDHDKDMrN)FF)__name__ __module__ __qualname____doc__r&rrrr/s   rrceZdZdZ ddZedZedZedZedZ edZ edZ ed Z ed Z ed Zd Zed ZedZedZedZy)TrafficTargetPaira< Holder for TrafficTarget status information. The representation of the status of traffic for a service includes: o User requested assignments (spec.traffic) o Actual assignments (status.traffic) Each of spec.traffic and status.traffic may contain multiple traffic targets that reference the same revision, either directly by name or indirectly by referencing the latest ready revision. The spec and status traffic targets for a revision may differ after a failed traffic update or during a successful one. A TrafficTargetPair holds all spec and status TrafficTargets that reference the same revision by name or reference the latest ready revision. Both the spec and status traffic targets can be empty. The latest revision can be included in the spec traffic targets two ways o by revisionName o by setting latestRevision to True. Managed cloud run provides a single combined status traffic target for both spec entries with: o revisionName set to the latest revision's name o percent set to combined percentage for both spec entries o latestRevision not set In this case both spec targets are paired with the combined status target and a status_percent_override value is used to allocate the combined traffic. Attributes: key: Either the referenced revision name or 'LATEST' if the traffic targets reference the latest ready revision. latestRevision: Boolean indicating if the traffic targets reference the latest ready revision. revisionName: The name of the revision referenced by these traffic targets. specPercent: The percent of traffic allocated to the referenced revision in the service's spec. statusPercent: The percent of traffic allocated to the referenced revision in the service's status. specTags: Tags assigned to the referenced revision in the service's spec as a comma and space separated string. statusTags: Tags assigned to the referenced revision in the service's status as a comma and space separated string. urls: A list of urls that directly address the referenced revision. tags: A list of TrafficTag objects containing both the spec and status state for each traffic tag. displayPercent: Human-readable representation of the current percent assigned to the referenced revision. displayRevisionId: Human-readable representation of the name of the referenced revision. displayTags: Human-readable representation of the current tags assigned to the referenced revision. serviceUrl: The main URL for the service. cf||_||_||_||_||_||_d|_y)aCreates a new TrafficTargetPair. Args: spec_targets: A list of spec TrafficTargets that all reference the same revision, either by name or the latest ready. status_targets: A list of status TrafficTargets that all reference the same revision, either by name or the latest ready. revision_name: The name of the revision referenced by the traffic targets. latest: A boolean indicating if these traffic targets reference the latest ready revision. status_percent_override: Percent to use for the status percent of this TrafficTargetPair, overriding the values in status_targets. service_url: The main URL for the service. Optional. Returns: A new TrafficTargetPair instance. N) _spec_targets_status_targets_revision_name_latest_status_percent_override _service_url_tags)r# spec_targetsstatus_targets revision_namelateststatus_percent_override service_urls rr&zTrafficTargetPair.__init__s:(&D)D'DDL$;D!#DDJrcd|jrtjStj|Sr)latestRevisionrLATEST_REVISION_KEYGetKeyr#s rkeyzTrafficTargetPair.keys.""  ' '>(/t(<>rc|jS)zBReturns true if the traffic targets reference the latest revision.)r3rAs rr>z TrafficTargetPair.latestRevisions <<rc|jSr)r2rAs r revisionNamezTrafficTargetPair.revisionNames   rcv|jr(tjt|jStSr)r0six text_typerr rAs r specPercentzTrafficTargetPair.specPercents+  ]];t'9'9: ;; %%rc|jtj|jS|jr(tjt |jSt Sr)r4rGrHr1rr rAs r statusPercentzTrafficTargetPair.statusPercentsL $$0 ]]488 99   ]];t';';< == %%rcttjtd|jD}|r|StS)Nc3NK|]}|js|jywrrrs rrz-TrafficTargetPair.specTags..s:0AEEquu0r)_TAGS_JOIN_STRINGjoinsortedr0r )r# spec_tagss rspecTagszTrafficTargetPair.specTagss4!&&:d00::.s<2aeequu2r)rOrPrQr1r )r# status_tagss r statusTagszTrafficTargetPair.statusTagss4#((K%;C+CCrc:td|jDS)Nc3NK|]}|js|jywr)r rs rrz)TrafficTargetPair.urls..s?!5A!%%!5r)rQr1rAs rurlszTrafficTargetPair.urlss ?!5!5? ??rcR|j|j|jSr)r6 _ExtractTagsrAs rtagszTrafficTargetPair.tagss" zz  ::rci}|jD]3}|jst|jd||j<5|jD]}|js|j|vr||j}n/|j |jt|j}|j |j nd|_d|_t|jtjd|_ y)zEExtracts the traffic tag state from spec and status into TrafficTags.T)r$Nr'rrB) r0rrr1 setdefaultr r"rQvaluesoperator attrgetterr6)r#r] spec_target status_targetrs rr\zTrafficTargetPair._ExtractTagss D)) __($Gd;??*--      d "=$$%oom//M+>u+EFDJrc|j|jk(rt|jSdjt|jt|jS)z(Returns human readable revision percent.z{:4} (currently {}))rKrIrr rAs rdisplayPercentz TrafficTargetPair.displayPercentsX T--- t11 22 " ) ) D,, - D.. /11rc||jr%tj|d|jdS|jS)z+Returns human readable revision identifier.z (currently ))r>rr@rErAs rdisplayRevisionIdz#TrafficTargetPair.displayRevisionIds: $+NN4$8$($5$577  rcz|j}|j}||k(r |tk7r|SdSdj||S)Nr'z{} (currently {}))rSrWr r )r#rRrVs r displayTagszTrafficTargetPair.displayTagssB I//KK'+CC[KK ' ' ; ??rc|jS)zThe main URL for the service.)r5rAs r serviceUrlzTrafficTargetPair.serviceUrl s   rNr')r(r)r*r+r&propertyrBr>rErIrKrSrWrZr]r\rgrjrlrnr,rrr.r.Ks$8J,.8 > >     & &  & & @ @  D D  @ @   G$ 1 1   @ @  rr.cd}|rTtj|vrBtj|vr0||vr,||}||tj<||vr t|}|S||=|S)aSplits the fully-managed latest status target. For managed the status target for the latest revision is included by revisionName only and may hold the combined traffic percent for both latestRevisionName and latestRevision spec targets. Here we adjust keys in status_dict to match with spec_dict. Args: spec_dict: Dictionary mapping revision name or 'LATEST' to the spec traffic target referencing that revision. status_dict: Dictionary mapping revision name or 'LATEST' to the status traffic target referencing that revision. Modified by this function. is_platform_managed: Boolean indicating if the current platform is Cloud Run fully-managed. latest_ready_revision_name: The name of the latest ready revision. Returns: Optionally, the id of the list of status targets containing the combined traffic referencing the latest ready revision by name and by latest. N)rr?id) spec_dict status_dictis_platform_managedlatest_ready_revision_namecombined_status_targets_idlatest_status_targetss r_SplitManagedLatestStatusTargetrysy, $g99YF !!4 K/'(BC/DK++,!Y.#%&;#<  $# 0 1 ##rcd}t||k(rOt|tj}t|}t ||}|tjk(r|}|S||z }|S)zFComputes the optional override percent to apply to the status percent.N)rrrrr?min)rBrsr8rwpercent_overridespec_by_latest_percentstatus_percentstatus_by_latest_percents r_PercentOverrider6sv55(73N3N)OP 0N"#9>J g)))1 (*BB rc t|}t|}t||||}g}t|j|D]o} |j | g} |j | g} t | || |} | t jk(r|} d}n| } d}|jt| | | || |qt|t jS)aReturns a list of TrafficTargetPairs for a Service. Given the spec and status traffic targets wrapped in a TrafficTargets instance for a sevice, this function pairs up all spec and status traffic targets that reference the same revision (either by name or the latest ready revision) into TrafficTargetPairs. This allows the caller to easily see any differences between the spec and status traffic. For fully-managed Cloud Run, the status target for the latest revision is included by revisionName only and may hold the combined traffic percent for both latestRevisionName and latestRevision spec targets. This function splits the fully-managed status target for the latest revision into a target for the percent allocated to the latest revision by name and a target for the percent allocated to the latest revision because it is latest. Args: spec_traffic: A traffic.TrafficTargets instance wrapping the spec traffic. status_traffic: A traffic.TrafficTargets instance wrapping the status traffic. is_platform_managed: Boolean indicating whether the current platform is fully-managed or Anthos/GKE. latest_ready_revision_name: The name of the service's latest ready revision. service_url: The main URL for the service. Optional. Returns: A list of TrafficTargetPairs representing the current state of the service's traffic assignments. The TrafficTargetPairs are sorted by revision name, with targets referencing the latest ready revision at the end. TFr_) dictrysetuniongetrrr?appendr.rQSortKeyFromTarget) spec_trafficstatus_trafficrurvr<rsrtrwresultkr7r8r|r9r:s rGetTrafficTargetPairsrEsF< )^$+>13M O & y>   ,a==B'L __Q+N'9n(BDG ' ''0mfmf MM, v*K 9: - G55 66rro)r+ __future__rrrrrbgooglecloudsdk.api_lib.runrrGr rOrrobjectrr.ryrrr,rrrsk5&%'. !6 8EEP $F  CE87r