From c02c8437e70a7a827cd0b4b9ddcab69f3130b61d Mon Sep 17 00:00:00 2001 From: Zevaryx Date: Fri, 10 Jun 2022 08:12:36 -0600 Subject: [PATCH] Add malicious url block tracker --- jarvis/client.py | 9 +++++++++ jarvis/tracking.py | 8 ++++++++ 2 files changed, 17 insertions(+) create mode 100644 jarvis/tracking.py diff --git a/jarvis/client.py b/jarvis/client.py index 386349f..985c0df 100644 --- a/jarvis/client.py +++ b/jarvis/client.py @@ -44,6 +44,7 @@ from nafftrack.client import StatsClient from pastypy import AsyncPaste as Paste from jarvis import const +from jarvis.tracking import malicious_tracker from jarvis.utils import build_embed from jarvis.utils.embeds import warning_embed @@ -605,6 +606,10 @@ class Jarvis(StatsClient): embed = warning_embed(message.author, "Phishing URL") await message.channel.send(embeds=embed) await message.delete() + tracker = malicious_tracker.labels( + guild_id=message.guild.id, guild_name=message.gulid.name + ) + tracker.inc() return True return False @@ -637,6 +642,10 @@ class Jarvis(StatsClient): embed = warning_embed(message.author, reasons) await message.channel.send(embeds=embed) await message.delete() + tracker = malicious_tracker.labels( + guild_id=message.guild.id, guild_name=message.gulid.name + ) + tracker.inc() return True return False diff --git a/jarvis/tracking.py b/jarvis/tracking.py new file mode 100644 index 0000000..4b4e1b0 --- /dev/null +++ b/jarvis/tracking.py @@ -0,0 +1,8 @@ +"""JARVIS Prometheus extra tracking.""" +from prometheus_client import Counter + +malicious_tracker = Counter( + "jarvis_malicious_blocked", + "Amount of malicious messages blocked", + labelnames=["guild_id", "guild_name"], +)