from prometheus_client import Gauge


# Gauge: Certificate validity period remaining (seconds)
cert_expiry_gauge = Gauge(
    "athenz_cert_expiry_seconds",
    "Athenz cert remaining seconds",
    ["account", "flava_env", "flava_project_name", "flava_product"],
)


def set_cert_expiry_metric(
    account, flava_env, flava_project_name, flava_product, remaining_seconds
):
    cert_expiry_gauge.labels(
        account=account,
        flava_env=flava_env,
        flava_project_name=flava_project_name,
        flava_product=flava_product,
    ).set(remaining_seconds)