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
