dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd lmZdd lmZddlm Z dd lmZdd lmZdd lmZddlmZddlmZddlZddiZdgZdZGdde j:ZGdde j:ZGdde j:Z Gdde j:Z!Gdde j:Z"Gdde j:Z#d Z$d!Z%d"Z& d`d#Z'd$Z(d%Z)d&Z*d'Z+ dad(Z,dbd)Z- dcd*Z.d+Z/d,Z0d-Z1d.Z2d/Z3 ddd0Z4d1Z5d2Z6d3Z7d4Z8d5Z9dbd6Z:dbd7Z;ded8Z< dfd9Z=d:Z>d;d<d?e?d;ze?d@<d?e?d@ze?dA<dBe?dAze?dC<dDe?dCze?dE<dFe?dCze?dG<ejdHdIje?zZBdJZCdKZDejdLZEejdMZFejdNZGdOZHejdPZIdQZJdRZKdSZLdTZMdUZNdbdVZOdWZPdbdXZQ dedYZRdZZSd[ZTd\ZUd]ZVd^ZW ded_ZXy)gz0Util methods for Stackdriver Monitoring Surface.)absolute_import)division)unicode_literalsN)encoding) exceptions)util) arg_utils) labels_util)log) properties) resources)yaml)times channelLabelslabelscriteriaz@Notification channel migrated from Prometheus alert manager fileceZdZdZy)YamlOrJsonLoadErrorzGException for when a JSON or YAML string could not loaded as a message.N__name__ __module__ __qualname____doc__1lib/googlecloudsdk/command_lib/monitoring/util.pyrr.sOrrceZdZdZy)NoUpdateSpecifiedErrorzEException for when user passes no arguments that specifies an update.Nrrrrrr2MrrceZdZdZy)ConditionNotFoundErrorz;Indiciates the Condition the user specified does not exist.Nrrrrr!r!6sCrr!ceZdZdZy)ConflictingFieldsErrorz@Inidicates that the JSON or YAML string have conflicting fields.Nrrrrr#r#:sHrr#ceZdZdZy)MonitoredProjectNameErrorzEInidicates that an invalid Monitored Project name has been specified.Nrrrrr%r%>rrr%ceZdZdZy)MissingRequiredFieldErrorzHInidicates that supplied policy/alert rule is missing required field(s).Nrrrrr'r'BsPrr'ct|Dcgc]}|j|c}stdj|ycc}w)Nz.Did not specify any flags for updating the {}.)any IsSpecifiedrformat)argsupdate_arg_destsresourcedests rValidateUpdateArgsSpecifiedr0FsM 1A B1Adt$1A B C 8??I KK D BsActj|D]=\}}||vs ||vrtdj|||j |||<?|S)NzCannot specify both {} and {}.)six iteritemsr#r+pop)yaml_objfield_remappings field_name remapped_names r _RemapFieldsr9Lsf#&==1A#Bj-X ( "$%E%L%L  &'( ( ( Z 8h} $C /rc@|D]}||vs|j||SN)r4)r5field_deletionsr7s r _DeleteFieldsr=Vs'#jXll:$ /rc tj|}|r t||}|r t||}t j ||}|S#t $r }tdj||d}~wwxYw)Nz2Could not parse YAML or JSON string for [{0}]: {1}) rloadr9r=rPyValueToMessage Exceptionrr+) msg_string message_type display_typer6r< msg_as_yamlmsgexcs rMessageFromStringrH]s{  ))J'K .>?k!+?k  # #L+ >C J  <CC #    sAA A4A//A4c>|tddjddS)z'Converts a --flag-arg to its dest name.z--N-_)lenreplace) flag_names r _FlagToDestrOms 3t9:  & &sC 00rc$dj|S)Nz{}s)r+)durations r_FormatDurationrRrs h rc|jds|jdr)|jxs |j}t||d}|S|}|S)z"Returns the base policy from args.policypolicy_from_file AlertPolicy)r*rTrUrH)r, policy_class policy_stringrTs rGetBasePolicyMessageFromArgsrYvsU h4#3#34F#GKK84#8#8M }lM JF -^F -rc|jdr(|jdstjddygd}|D]<}|dk(rd}n t|}|j|s(tj|dy) aChecks if condition arguments exist and are specified correctly. Args: args: argparse.Namespace, the parsed arguments. Returns: bool: True, if '--condition-filter' is specified. Raises: RequiredArgumentException: if '--if' is not set but '--condition-filter' is specified. InvalidArgumentException: if flag in should_not_be_set is specified without '--condition-filter'. condition_filterif_value--ifz;If --condition-filter is set then --if must be set as well.T)z --aggregationz --durationz--trigger-countz--trigger-percentz--condition-display-namer]z --combinerzAShould only be specified if --condition-filter is also specified.F)r* calliope_excRequiredArgumentExceptionrOInvalidArgumentException)r,should_not_be_setflagr/s rCheckConditionArgsrcs ()   J '  2 2  G II " 4   $ 33  OQ Q " rc @|s|j}|||_d} |s|r|j||} | ||d} |r|| d<|Z|\} } | s|jdi| |_|S|j j } |j dt| | | d| |_|S)aGPopulates the fields of a Condition message from args. Args: messages: module, module containing message classes for the stackdriver api condition: Condition or None, a base condition to populate the fields of. display_name: str, the display name for the condition. aggregations: list[Aggregation], list of Aggregation messages for the condition. trigger_count: int, corresponds to the count field of the condition trigger. trigger_percent: float, corresponds to the percent field of the condition trigger. duration: int, The amount of time that a time series must fail to report new data to be considered failing. condition_filter: str, A filter that identifies which time series should be compared with the threshold. if_value: tuple[str, float] or None, a tuple containing a string value corresponding to the comparison value enum and a float with the condition threshold value. None indicates that this should be an Absence condition. Returns: Condition, a condition with its fields populated from the args N)countpercent)triggerrQfilter aggregations) comparisonthresholdValuer) Condition displayNameTrigger MetricAbsenceconditionAbsentMetricThresholdComparisonValueValuesEnumgetattrconditionThreshold)messages condition display_nameri trigger_counttrigger_percentrQr[r\rgkwargs comparatorthreshold_valuecomparison_enums rBuildConditionr~s4 ""$I(I 'o_6G  &)F> "*J "8("8"8"B6"Bi  !00JJo%=X%=%=&_j9(& &i" rc|xs4tjjjj d}t j j|d|idS)NTrequired projectsIdz(monitoring.projects.notificationChannelsparams collectionr VALUEScoreprojectGetr REGISTRYParse) channel_namers rParseNotificationChannelrsW  Hz((--559949H'    ! !L'2; " ==rc |g}||jd||_||!|js|j|_|"|jd||j_|"|jd||j_||jd||_||jd||_|;|jdtj||jd}||_ yy) z?Override and/or add fields from other flags to an Alert Policy.Nrwzdocumentation.contentzdocumentation.mime_typeenablednotification_channelscombiner item_type) appendrm documentation DocumentationcontentmimeTypernotificationChannelsr ChoiceToEnumCombinerValueValuesEnumr) base_policyrurwrdocumentation_contentdocumentation_formatrchannels field_maskss rModifyAlertPolicyrsK~&*K(,@,L'' ( 6 6 8K&./(=K%%01)=K& y!!K./'/K$ z"%%+55MH#K rc t|Dcgc]}|jt|c}stj|ycc}wr;)r)r*rOr^MinimumArgumentException)r,flagsrbs rValidateAtleastOneSpecifiedrsP !${401! "  / / 66 "!s!Ac Vgd}t||t||j}|jdr |jnd}|jdr |j nd}|j jjxsg}|Dcgc]}|jc}xsd}|jxs |j} | r |jnd} t|||j|| | ||t|rd} |j r"t#|j |j$dg} t'||j(| |j*|j,t/|j0|j2|j4} |j6j9| |Scc}w)z(Builds an AleryPolicy message from args.)--display-namez--policyz--policy-from-filerNr)rwrrrrr Aggregation)rwrirxryrQr[r\)rrYrVr*rrCONCEPTSrr RelativeNamerdocumentation_from_filerrrwrc aggregationrHrr~condition_display_namerxryrRrQr[r\ conditionsr) r,rupolicy_base_flagsrTrr channel_refschannelrrrrirvs rCreateAlertPolicyFromArgsrsJd$56 (h.B.B C&"..z:T]]( ,,Y7DLLT'44::<B,4@ ALg""$L A IT(,,L0L0L#8dd $$1/L '   H00-ABl00!((,, /.. I Y' -?BsF&c gd}t||d}|jxs |j}|rt||jd}d}|j r"t|j |j dg}t|||j||j|jt|j|j|j S)z%Builds a Condition message from args.)z--condition-filterz --conditionz--condition-from-fileNrlr)rvrwrirxryrQr[r\)rrvcondition_from_filerHrlrrr~rrxryrRrQr[r\)r,rucondition_base_flagsrvcondition_stringris rGetConditionFromArgsrGs3d$89)^^?t'?'?!(,,k;I, % (.. ?@L ..&&**t}}-,,}}  rc~|jD]}|j|k(s|cStdj|Nz,No condition with name [{}] found in policy.)rnamer!r+)condition_namerTrvs rGetConditionFromPolicyrdsC$$i~~' % 4;;NK MMrct|jD]3\}}|j|k(s|jj||cSt dj |r) enumeraterrr4r!r+)rrTirvs rRemoveConditionFromPolicyrms_ 1 12la~~' A m3 4;;NK MMrc|g}||jd||_||jd||_||jd||_||jd||_|S)z>Modifies base_channel's properties using the passed arguments.typerw descriptionr)rrrmrr) base_channel channel_typerrwrrs rModifyNotificationChannelrwsKv$L~&+L}%*L y!"L rcgd}t|||jxs |j}|r[t||jdt }|j rBt|j jd|j _n|j }|jdr |jnd}t||j|j|j|S) z/Builds a NotificationChannel message from args.)rz--channel-contentz--channel-content-from-fileNotificationChannel)r6c|jSr;key)props rz0GetNotificationChannelFromArgs..srrrN)rrwrr)rchannel_contentchannel_content_from_filerHrCHANNELS_FIELD_REMAPPINGSrsortedadditionalPropertiesr*rrrrwr)r,ruchannels_base_flagschannel_stringrrs rGetNotificationChannelFromArgsrs8d$78''I4+I+I.0L0L 51JLG~~,2 .. - -3H-Jgnn)**,G ,,Y7DLLT' "704 040A0A/3/?/?+2  44rc2tj||dS)NT) sort_items)rDictToAdditionalPropertyMessage)r labels_clss rParseCreateLabelsrs  1 1 jT ++rc tjt|d|zt|d|zt|d|z}|jsy|j ||j S)aReturns the result of applying the diff constructed from args. This API doesn't conform to the standard patch semantics, and instead does a replace operation on update. Therefore, if there are no updates to do, then the original labels must be returned as writing None into the labels field would replace it. Args: args: argparse.Namespace, the parsed arguments with update_labels, remove_labels, and clear_labels labels_name: str, the name for the labels flag. labels_cls: type, the LabelsValue class for the new labels. orig_labels: message, the original LabelsValue value to be updated. Returns: LabelsValue: The updated labels of type labels_cls. Raises: ValueError: if the update does not change the labels. update_remove_clear_) additions subtractionsclearN)r DiffrsMayHaveUpdatesApply GetOrNone)r, labels_namer orig_labels labels_diffs rProcessUpdateLabelsrsn*  i+564[!89 D([0 13+  # # %    :{ 3 = = ??rctjd|}|rk|jd|k7r tdt j |jd}t j |jd}||fSt j t jjjjd}tjd|}|r(t j |jd}||fS|r>tjd j| t j |}||fStd) aMReturns the metrics scope and monitored project. Parse the specified monitored project name and return the metrics scope and monitored project. Args: monitored_project_name: The name of the monitored project to create/delete. project_fallback: When set, allows monitored_project_name to be just a project id or number. Raises: MonitoredProjectNameError: If an invalid monitored project name is specified. Returns: (metrics_scope_def, monitored_project_def): Project parsed metrics scope project id, Project parsed metrics scope project id zElocations/global/metricsScopes/([a-z0-9:\-]+)/projects/([a-z0-9:\-]+)r2Invalid monitored project name has been specified.Trzprojects/([a-z0-9:\-]+)Received an incorrectly formatted project name. Expected "projects/{identifier}" received "{identifier}". Assuming given resource is a project. identifier)rematchgroupr% projects_util ParseProjectr rrrrr warningr+)monitored_project_nameproject_fallbackmatchedmetrics_scope_defmonitored_project_def$monitored_resource_container_matcheds rParseMonitoredProjectrs^& HHO ' }}Q11 % > @@&227==3CD)66w}}Q7GH0 1 11-&22&&**D*9;+-88"$:,(,+88 . 4 4Q 7 1 11  kk ))//*0* ,889OP 1 11 & > rc$tjd|}|r"|jd|jdfS|rJtjdj |dt j|jfStd)aLReturns the monitored resource container identifier. Parse the specified monitored_resource_container_name and return the identifier. Args: monitored_resource_container_name: The monitored resource container. Ex - projects/12345. project_fallback: When set, allows monitored_resource_container_name to be just a project id or number. Raises: MonitoredProjectNameError: If an invalid monitored project name is specified. Returns: resource_type, monitored_resource_container_identifier: Monitored resource container type and identifier z(projects)/([a-z0-9:\-]+)rrrrprojectsr) rrrr rr+rrNamer%)!monitored_resource_container_namerrs rParseMonitoredResourceContainerrs, HH"$E ' == W]]1- --KK ''-v8(.(  ""#DEJJL  $< rc|xs4tjjjj d}t j j|d|idS)NTrrzmonitoring.projects.snoozesrr) snooze_namers r ParseSnoozer/sX  Hz((--559949H'    ! !G $. " rc|xs4tjjjj d}t j j|d|idS)NTrrzmonitoring.projects.alertsrr) alert_namers r ParseAlertr 8sX  Hz((--559949H'    ! !G $- " rc|jdr2|j}|rt||dt}|St||d}|S|}|S)z"Returns the base snooze from args.snooze_from_fileSnooze)r<)r*r rHSNOOZE_FIELD_DELETIONS)r, snooze_classupdate snooze_stringsnoozes rGetBaseSnoozeMessageFromArgsrAsf ()))M    0 f -!   f -^F -rc|g}d}d} ||jd|}n, tj|jj}d} d} d} ||jd|} n, tj|jj } d} |&| s$tj||j_| &| s$tj| |j_||jd||_ |Z|jd|j} || _ | |_ |(t|dk7r td || _| |_ yy| t!d y#t $rY'wxYw#t $rYwxYw#t $rN|j} tj|| _tj| | _| |_YwxYw) z9Override and/or add fields from other flags to an Snooze.NFzinterval.start_timeTzinterval.end_timerwcriteria_policiesrzCExactly 1 alert policy is required if criteria-filter is specified.z>criteria-policies is required if criteria-filter is specified.)rr ParseDateTimeinterval startTimeAttributeErrorendTimeFormatDateTime TimeIntervalrmCriteriapoliciesrrL ValueErrorrhr') base_snoozerurwrcriteria_filter start_timeend_timerstart_time_targetstart_time_from_baseend_time_targetend_time_from_baserrs r ModifySnoozer'WsK,-" --k.B.B.L.LM!/ *+O ++K,@,@,H,HIo $$-A',';';d$56 (hoo >& $$..**}} -rsrgMbP?ms<mhdwimyz ^([0-9]+)(%s)|cd}|}tjddz}|rtj|}|st dj | t |jd}|jd}tj|}||k\rt dj ||}||z}||z }||jdd}|r|S#t$rt dj |wxYw) asForked from datelib.py. Convert a formatted Prometheus string representing an interval into seconds. The accepted interval string syntax is: interval: (interval_part)* interval_part: decimal_integer unit unit: "ms" # Milliseconds | "s" # Seconds | "m" # Minutes | "h" # Hours | "d" # Days | "w" # Weeks (7 days) | "y" # Years (365 days) No whitespace is allowed. The empty string is valid (and equivalent to 0 seconds). |decimal_integer| cannot include a sign. No endianness ordering is required when using multiple interval_part-s. For example, "1s1Y" and "1Y1s" are both valid. Examples: "45m" = 45 minutes "14d12h" = 14 days + 12 hours "5d12h30m" = 5 days + 12 hours + 30 minutes Args: interval: String to interpret as an interval. See above for a description of the syntax. Raises: ValueError: If the provided time_string contains unexpected characters. Returns: A non-negative integer representing the number of seconds represented by the interval string. rr4rzE{} is invalid due to missing unit of time or unexpected character(s).zLFound invalid character in {}, which is neither an integer nor unit of time.rz6Time units not ordered from largest to smallest in {}.N) _INTERVAL_CONV_DICTget_INTERVAL_PART_REGEXPrrr+intrend)rtotaloriginal_intervalprevious_multiplierrnumsuffix multipliers rConvertIntervalToSecondsrBs#N %,//4q8 ! ' ' 1E   !6"34    A c[[^F$((0J((  B I I   %:C SLE ! 'HE F ,1   f./ s C$C9ct|}|dkrtdj|||dzdk7rtdj||t|S)aConverts Prometheus time to duration JSON string. Args: time_string: String provided by the alert rule YAML file defining time (ex:1h30m) Raises: ValueError: If the provided time_string is not a multiple of 30 seconds or is less than 30 seconds. Returns: Duration proto string representing the adjusted seconds (multiple of 30 seconds) value of the provided time_string z>{time_string} converted to {seconds}s is less than 30 seconds. time_stringsecondsrz4{} converted to {}s is not a multiple of 30 seconds.)rBrr+rRrEs r8ConvertPrometheusTimeStringToEvaluationDurationInSecondsrHsw %[ 1' r\ HOO# P   |q >EE     !!rz!\{\{ *(humanize )? *\$value *\}\}z"\{\{ *(humanize )? *\$labels *\}\}z<\{\{ *(humanize )? *\$labels\.([a-zA-Z_][a-zA-Z0-9_]*) *\}\}c ~tjdtjdtjd|S)aTranslate Prometheus templating language constructs to document variables. TranslatePromQLTemplateToDocumentVariables translates common Prometheus templating language constructs to their equivalent Cloud Alerting document variables. See: https://prometheus.io/docs/prometheus/latest/configuration/template_reference/ and https://cloud.google.com/monitoring/alerts/doc-variables. Only the following constructs will be translated: "{{ $value }}" will be translated to "${metric.label.value}". "{{ humanize $value }}" will be translated to "${metric.label.value}". "{{ $labels. }}" will be translated to "${metric_or_resource.label.}". "{{ humanize $labels. }}" will be translated to "${metric_or_resource.label.}". "{{ $labels }}" will be translated to "${metric_or_resource.labels}". "{{ humanize $labels }}" will be translated to "${metric_or_resource.labels}". The number of spaces inside the curly braces is immaterial. All other Prometheus templating language constructs are not translated. Notes: 1. A document variable reference that does not match a variable will be rendered as "(none)". 2. We do not know whether a {{ $labels. }} construct refers to a Cloud Alerting metric or a resource label. Thus we translate it to "${metric_or_resource.label.}". Note that a reference to a non-existent label will be rendered as "(none)". Examples: 1. "[{{$labels.a}}] VALUE = {{ $value }}" will be translated to "[${metric_or_resource.label.a}] VALUE = ${metric.label.value}". 2. "[{{humanize $labels.a}}] VALUE = {{ humanize $value }}" will be translated to "[${metric_or_resource.label.a}] VALUE = ${metric.label.value}". Args: template: String contents of the "subject" or "content" fields of an AlertPolicy protoco buffer. The contents of these fields is a template which may contain Prometheus templating language constructs. Returns: The translated template. z${metric_or_resource.label.\2}z${metric_or_resource.labels}z${metric.label.value})_LABELS_KEY_REGEXPsub_LABELS_VARIABLE_REGEXP_VALUE_VARIABLE_REGEXP)templates r*TranslatePromQLTemplateToDocumentVariablesrO;s?d   '!! ) $ $& 23 44rz[A-Za-z_][A-Za-z0-9_]*c|j}|j|_|jd t d|jd t dt j |jd td|jd t d|jd|j_|jd|_ |jd|j_ |jd|j_ |jd2tt|jd|j_|jd )t|jd |j_|jd |jj#|j_|jd j'D]Y\}}|jj$j(j+|jj"j-|| [|S) afPopulates Alert Policy conditions translated from a Prometheus alert rule. Args: messages: Object containing information about all message types allowed. group: Information about the parent group of the current rule. rule: The current alert rule being translated into an Alert Policy. Raises: MissingRequiredFieldError: If the provided group/rule is missing an required field needed for translation. ValueError: If the provided rule name is not a valid Prometheus label name. Returns: The Alert Policy condition corresponding to the Prometheus group and rule provided. rz+Missing rule group name in field group.namealertz2Missing alert rule name in field group.rules.alertzUAn invalid alert rule name in field group.rules.alert (not a valid PromQL label name)exprz6Missing a PromQL expression in field groups.rules.exprforrrrvalue)rl PrometheusQueryLanguageCondition conditionPrometheusQueryLanguager8r'_VALID_LABEL_REGEXP fullmatchr ruleGrouprm alertRulequeryrRrBrQrHevaluationInterval LabelsValueritemsrrAdditionalProperty)rurrulervkvs rBuildPrometheusConditionrdzs"  ")//1 , YYv #5  XXg #< ""488G#45=  *  XXf #@ :?69J),,6((7+)9='9J),,659XXf5E),,2 XXe_ :I %1;I..7 YYz&@ IIj ! ..A  XXh#11==?..5"((*10077LLSS  3 3 ? ? R R1 S + rcx tj|}| tdg}|jd td|jdD]8}|jdtd|jdz|jdD]}t |||}|j }|j j||jd|j|_ |jdjd 8t|jdjd |j_ |jdjd 8t|jdjd |j_ d |j_ tj|jd6d j!|jd|jd |_n5dj!|jd|jd |_t%j&d|j(d|_|||_|j|;|S#t.$r} t1dj!| d} ~ wwxYw)aPopulates Alert Policies translated from Prometheus alert rules. Args: rule_yaml: Opened object of the Prometheus YAML file provided. messages: Object containing information about all message types allowed. channels: List of Notification Channel names to be added to the translated policies. Raises: YamlOrJsonLoadError: If the YAML file cannot be loaded. Returns: A list of the Alert Policies corresponding to the Prometheus rules YAML file provided. Nz&Failed to load YAML file. Is it empty?groupsz No groupsruleszNo rules in group "%s"r annotationssubjectrz text/markdownz{0}/{1}rQz "{0}"/{1}ORrrCould not parse YAML: {0})rr?rr8rdrVrrrrrOrirrrXrYr+rmr rrrrrAr) rule_yamlrurcontentsrrrarvrTrGs rPrometheusMessageFromStringrnsf 3Gyy#H ? @@H||H% { ##h' 7  #1EIIf4EEFF))G$$,XudC %%'  + 88M " .!)!7!7!9&  XXm $ ( ( 3 ?:HH]+// :<  (XXm $ ( ( 7 C:HH]+// >@  (+:&   '  ( (6): ; G(//ii'!2 &  +11ii'!2 & $00 &00J   (0& %E% (P O G 9@@E FFGsJJ J9J44J9c|j}||_t|_|jj |_|S)anHelper function for creating a basic Notification Channel translated from a Prometheus alert_manager YAML. Args: channel_name: The display name of the desired channel. messages: Object containing information about all message types allowed. Returns: A base Notification Channel containing the requested display_name and other basic fields. )rrmMIGRATED_FROM_PROMETHEUS_TEXTrr^r)rrurs r#CreateBasePromQLNotificationChannelrqs=  ( ( *'$'5'//;;='. .rc Vg}|jd}| td|jd|jdD]}|jdt||}d|_|jj j |jjjd|jd|j ||jd|jdD]}|jd t||}d |_|jj j |jjjd |jd |j ||jd |jd D]}|jd t||}d |_|jj j |jjjd |jd |j |tgd}|jD]F}||vstjjdj||jdH|S)aUPopulates a Notification Channel translated from Prometheus alert manager. Args: receiver_config: Object containing information the Prometheus receiver. For example receiver_configs, see https://github.com/prometheus/alertmanager/blob/main/doc/examples/simple.yml messages: Object containing information about all message types allowed. Raises: MissingRequiredFieldError: If the provided alert manager file contains receivers with missing required field(s). Returns: The Notification Channel corresponding to the Prometheus alert manager provided. rzMSupplied alert manager file contains receiver without a required field "name" email_configstoemail email_addressrTpagerduty_configs service_key pagerdutywebhook_configsurlwebhook_tokenauth)rrsrwrzzOFound unsupported receiver type {field}. {name}.{field} will not be translated.)fieldr)r8r'rqrrrrrr^r`setkeysr outPrintr+)receiver_configrurrfieldsrsupported_receiver_fieldsr}s r$BuildChannelsFromPrometheusReceiversrsr"( $$V,, #   )5!%%o6 D  %5lHM ++22  ( ( 4 4 G G#6::d+; H   7,-9!%%&9: M " .5lHM" ++22  ( ( 4 4 G G!M)B H   ;*+7!%%&78 E  &5lHM* ++22  ( ( 4 4 G GE!2 H   9"G##%e -- ggmm e/2E2Ef2MN& /rc tj|}g}|j dD]}|t ||z }|S#t$r}tdj |d}~wwxYw)aPopulates Alert Policies translated from Prometheus alert rules. Args: alert_manager_yaml: Opened object of the Prometheus YAML file provided. messages: Object containing information about all message types allowed. Raises: YamlOrJsonLoadError: If the YAML file cannot be loaded. Returns: The Alert Policies corresponding to the Prometheus rules YAML file provided. rkN receivers)rr?rArr+r8r)alert_manager_yamlrurmrGrrs r$NotificationChannelMessageFromStringrdssGyy+,H(!k2o 4_hOOH3 / G 9@@E FFGsA A( A##A(cz|jdr'|j}g}|D]}|t|||z }|Sg}|S)aBuilds a PromQL policies message from args. Args: args: Flags provided by the user. messages: Object containing information about all message types allowed. channels: List of full Notification Channel names ("projects/<>/...") to be added to the translated policies. Returns: The Alert Policies corresponding to the Prometheus rules YAML file provided. In the case that no file is specified, the default behavior is to return an empty list. )policies_from_prometheus_alert_rules_yaml)r*rrn)r,rurall_rule_yamlsrrls rCreatePromQLPoliciesFromArgsr}sT ABCCNH# -i8LLh$ /H /rc`|jdr|j}t||}|Sg}|S)a~Builds a notification channel message from args. Args: args: Flags provided by the user. messages: Object containing information about all message types allowed. Returns: The notification channels corresponding to the Prometheus alert manager YAML file provided. In the case that no file is specified, the default behavior is to return an empty list. *channels_from_prometheus_alertmanager_yaml)r*rr)r,rurrs r"CreateNotificationChannelsFromArgsrsA BCHH3HH /H /rc|xs4tjjjj d}t j j|d|idS)NTrrz&monitoring.projects.uptimeCheckConfigsrr)uptime_check_namers rParseUptimeCheckrsX  Hz((--559949H'    ! !G $9 " rc v|j|j|_|jt|jdz|_|j'ddddd} | j |j|_||j tjdd|jjj|jjj|jjj|jjj|jjj|jjj d } g|_|D],} |j"j%| j | .||_t)|||t+||||||||S) aModifies an UptimeCheckConfig based on the args and other inputs. Args: uptime_check: UptimeCheckConfig that is being modified. messages: Object containing information about all message types allowed. args: Flags provided by the user. regions: Potentially updated selected regions. user_labels: Potentially updated user labels. headers: Potentially updated HTTP headers. status_classes: Potentially updated allowed status classes. status_codes: Potentially updated allowed status codes. update: If this is an update operation (true) or a create operation (false). Returns: The updated UptimeCheckConfig object. r+60s300s600s900s)151015regions3Should not be set or updated for Synthetic Monitor.)z usa-oregonzusa-iowaz usa-virginiaeuropez south-americaz asia-pacific)rwrmtimeoutstrperiodr8syntheticMonitorr^r`UptimeCheckConfig'SelectedRegionsValueListEntryValuesEnum USA_OREGONUSA_IOWA USA_VIRGINIAEUROPE SOUTH_AMERICA ASIA_PACIFICselectedRegionsr userLabelsSetUptimeCheckMatcherFieldsSetUptimeCheckProtocolFields) uptime_checkrur,r user_labelsheadersstatus_classes status_codesrperiod_mappingregion_mappingregions rModifyUptimeCheckrs6 "#00L \\t||,s2L [[   N ),,T[[9L $$0  1 1 J   & & N N Y Y  & & N N W W  & & N N [ [  & & N N U U  & & N N \ \  & & N N [ [#N($&L "")).*<*K  &&(1&&--  $ $ 4 4 G G1 H ) ' i   --/G,,$$&$!Q$$++    + + > >1A > N ',, ll(($$  rc ddddddddd}|j|_|j|j|j_|jj |j_|jjD]Y\}}|jj jj|jj j|| [y ) z,Set Monitored Resource fields based on args. uptime_url gce_instancegae_appaws_ec2_instanceaws_elb_load_balancerservicedirectory_servicecloud_run_revision)z uptime-url gce-instancezgae-appzaws-ec2-instanceaws-elb-load-balancerzservicedirectory-servicezcloud-run-revisionNrTN) MonitoredResourcemonitoredResourcer8 resource_typerr^rrr_rrr`)r,rurresource_mappingrbrcs rrr9s!$,6"<0  $,#=#=#?, (8(<(>EE""..AA B +rcdddd}|j|_|j|j_t j |j |j|jjd|j_ y)zSet Group fields based on args.INSTANCEAWS_ELB_LOAD_BALANCER)rrNz group typerN) ResourceGroup resourceGrouprgroupIdr rr8 group_typeResourceTypeValueValuesEnum resourceType)r,rur group_mappings rrrRsv!6 - (557,'+}},$,5,B,B(88-,)rc|j|_|j|j_|j|jj_y)z+Set Synthetic Monitor fields based on args.N)SyntheticMonitorTargetrCloudFunctionV2TargetcloudFunctionV2synthetic_targetr)r,rurs rrrbsI"*"A"A"C,$$&/8<7L7L,//4rcH|jdr|jtjdd|j }|j |_|jjj|jjj|jjj|jjj|jjj|jjj|jjjd}|j|j|_|jdr|j |jjj|jjjfvrtjdd|j#|_|j&|j$_|j"j*j,|j"j*j.|j"j*j,d }|j|j0|j$_g|_|j4j7|yy) z!Set Matcher fields based on args.matcher_contentNz--matcher_content(Should not be set for Synthetic Monitor.)zcontains-stringznot-contains-stringz matches-regexznot-matches-regexzmatches-json-pathznot-matches-json-pathN json_pathz --json-pathz1Should only be used with JSON_PATH matcher types.)z exact-matchz regex-matchN)r*rr^r`ContentMatcherrrMatcherValueValuesEnumCONTAINS_STRINGNOT_CONTAINS_STRING MATCHES_REGEXNOT_MATCHES_REGEXMATCHES_JSON_PATHNOT_MATCHES_JSON_PATHr8 matcher_typematcherJsonPathMatcherjsonPathMatcherrjsonPathJsonMatcherValueValuesEnum EXACT_MATCH REGEX_MATCHjson_path_matcher_type jsonMatchercontentMatchersr)r,rurcontent_matchermatcher_mappingjsonpath_matcher_mappings rrrks3 '($$0  1 1 I --/O"22O  # # : : J J  # # : : N N  # # : : H H  # # : : L L  # # : : L L  # # : : P P%%<<LL'O*.11$2C2CDO  $   ! ! 8 8 J J  ! ! 8 8 N N) 33 N  )1(@(@(Bo%15o%%.&&AAMM&&AAMM((CCOO" # & &t'B'B C%%1$&L   ''8o)rc|s|jds |jgd}|D]4}t|} |j| s tj|d|rtjdd|rtjdd|rtjddy|s|j d k(s |j /|j|j td |j |j|_|j } |j|j| _|j0|j| _ |j| j_ gd }|D]4}t|} |j| s tj|d |rtjdd |rtjdd |rtjdd y|j|j|_|j} |j |j | _|j"|j"| _|j&|j&| _|j*|j*| _| j.|j1| _|j2|j2| j._|j4|j4| j._|j0|j| _ |j| j_ |j6|j6j9| _|s|j dk(s | j:r8d| _|j|j| _| j?d| _n7d| _|j|j| _| jd| _d|j<j>j@i} |jB?|j=| _"| jG|jB| jD_$|jjJjL|jjJjN|jjJjLd} | jP |jR | jG|jR| _(|jjTjV|jjTjX|jjTjZ|jjTjVd}| j\ |j^ |jG|j^| _.|| _0|jbjdjf|jbjdjh|jbjdjj|jbjdjl|jbjdjn|jbjdjp|jbjdjrd}|Ig| _:|D]<}| jtjw|jc|jG|>y|:g| _:|D]-}| jtjw|jc|/y| jtBg| _:| jtjw|jc|jGdyy)z"Set Protocol fields based on args.rN) --path--validate-ssl--mask-headers--custom-content-type --username --password--body--request-method--content-typez--portz --pings-count--service-agent-authrrrzstatus-classesz status-codestcpzMissing required field "port") rrrrrrrrrrz'Should not be set for TCP Uptime Check.z2Should not be set or updated for TCP Uptime Check.httpsTiFPz oidc-token)r8postN) unspecifiedz url-encodedz user-providedN)1xx2xx3xx4xx5xxr)N) statusClass) statusValuer )>BB""??DD  ==AAN '4+>+>+J!/!3!3D4G4G!Hj    9 9 J J    9 9 E E    9 9 G G  ;;LL O%):):)F.2243D3DEj J  ' ' B B S S  ' ' B B S S  ' ' B B S S  ' ' B B S S  ' ' B B S S  ' ' B B S S  ' ' B B [ [)N.!/1j,"&..55  ' 'N4F4Fv4N ' O #  !/1j, &..55  ' 'F ' ; !  / / 7/1j,,,33  % %.2D2DU2K % L 8r)NN)NNNNNNNNr;)NNNNNNN)NNNNN)F)NNNNNN)Yr __future__rrrrapitools.base.pyrgooglecloudsdk.callioperr^#googlecloudsdk.command_lib.projectsrr$googlecloudsdk.command_lib.util.apisr $googlecloudsdk.command_lib.util.argsr googlecloudsdk.corer r r rgooglecloudsdk.core.utilrr2rr rpErrorrrr!r#r%r'r0r9r=rHrOrRrYrcr~rrrrrrrrrrrrrrr rr'r*r7compilejoinr9rBrHrMrLrJrOrXrdrnrqrrrrrrrrrrrrrrrrUs7&' %>E:<*#*)$* -h7$G P***PNZ--NDZ--DIZ--IN 0 0NQ 0 0QK >B  1  'T;?48IM :z=JNGK?C#$L7 )X:MMHL=A*.,44+ @>42n*Z2 EP,Ah!$7$<<D 3C 88C 3C 88C 3C 88C2377C!4S!99C" chh233 Od"D$$HI$"**%JKRZZCE74x!bjj!:;AHCGL$K\24.$ Od-`2  M99F Jr