Add JARVIS version to prometheus

This commit is contained in:
Zeva Rose 2022-06-10 09:04:28 -06:00
parent 16764fe463
commit e27a3e5f77
2 changed files with 9 additions and 2 deletions

View file

@ -44,7 +44,7 @@ from nafftrack.client import StatsClient
from pastypy import AsyncPaste as Paste
from jarvis import const
from jarvis.tracking import malicious_tracker, warnings_tracker
from jarvis.tracking import jarvis_info, malicious_tracker, warnings_tracker
from jarvis.utils import build_embed
from jarvis.utils.embeds import warning_embed
@ -125,6 +125,11 @@ class Jarvis(StatsClient):
self.phishing_domains = await response.json()
self.logger.info(f"Protected from {len(self.phishing_domains)} phishing domains")
@listen()
async def on_startup(self) -> None:
"""NAFF on_startup override. Prometheus info generated here."""
jarvis_info.info({"version": const.__version__})
@listen()
async def on_ready(self) -> None:
"""NAFF on_ready override."""

View file

@ -1,5 +1,5 @@
"""JARVIS Prometheus extra tracking."""
from prometheus_client import Counter
from prometheus_client import Counter, Info
malicious_tracker = Counter(
"jarvis_malicious_blocked",
@ -10,3 +10,5 @@ malicious_tracker = Counter(
warnings_tracker = Counter(
"jarvis_warnings", "Amount of warnings given out", labelnames=["guild_id", "guild_name"]
)
jarvis_info = Info("jarvis", "Jarvis version")