def set_messages_metrics( metric_messages_sent: Metric, message_type: str, service_version: str, number_messages_sent: int, trigger_message: str = "", ): """Set message metrics to be sent to pushgateway.""" if DEPLOYMENT_NAME: metric_messages_sent.labels( message_type=message_type, trigger=trigger_message, env=DEPLOYMENT_NAME, version=service_version, ).inc(number_messages_sent) else: _LOGGER.warning("THOTH_DEPLOYMENT_NAME env variable is not set.")
def get_metrics_for_label(metric: Metric, enum: MetricsEnum) -> Metric: return metric.labels(**enum.to_label_dict()) # type: ignore