jarvis/jarvis-bot
jarvis/jarvis-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add JARVIS version to prometheus

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

7
jarvis/client.py
View file

@ -44,7 +44,7 @@ from nafftrack.client import StatsClient
from pastypy import AsyncPaste as Paste from pastypy import AsyncPaste as Paste
from jarvis import const 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 import build_embed
from jarvis.utils.embeds import warning_embed from jarvis.utils.embeds import warning_embed
@ -125,6 +125,11 @@ class Jarvis(StatsClient):
self.phishing_domains = await response.json() self.phishing_domains = await response.json()
self.logger.info(f"Protected from {len(self.phishing_domains)} phishing domains") 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() @listen()
async def on_ready(self) -> None: async def on_ready(self) -> None:
"""NAFF on_ready override.""" """NAFF on_ready override."""

4
jarvis/tracking.py
View file

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