diff options
| author | 2020-01-12 11:32:33 +0100 | |
|---|---|---|
| committer | 2020-01-12 11:58:58 +0100 | |
| commit | 08ec201cafd1a276bf8f571c471dab911833cf44 (patch) | |
| tree | 2fb56932ebc7c8d1ff8fd72c47d7cd2225041b15 | |
| parent | Track messages by channel. (diff) | |
Add documentation.
Diffstat (limited to '')
| -rw-r--r-- | bot/cogs/metrics.py | 11 | 
1 files changed, 10 insertions, 1 deletions
diff --git a/bot/cogs/metrics.py b/bot/cogs/metrics.py index dc86f8e82..5d0ce4f98 100644 --- a/bot/cogs/metrics.py +++ b/bot/cogs/metrics.py @@ -8,7 +8,11 @@ from bot.bot import Bot  class Metrics(Cog): -    """Exports metrics for Prometheus.""" +    """ +    Exports metrics for Prometheus. + +    See https://github.com/prometheus/client_python for metric documentation. +    """      PREFIX = 'pydis_bot_' @@ -28,6 +32,7 @@ class Metrics(Cog):      @Cog.listener()      async def on_ready(self) -> None: +        """Initialize the guild member counter."""          members_by_status = defaultdict(lambda: defaultdict(int))          for guild in self.bot.guilds: @@ -42,20 +47,24 @@ class Metrics(Cog):      @Cog.listener()      async def on_member_join(self, member: Member) -> None: +        """Increment the member gauge."""          self.guild_members.labels(guild_id=member.guild.id, status=str(member.status)).inc()      @Cog.listener()      async def on_member_leave(self, member: Member) -> None: +        """Decrement the member gauge."""          self.guild_members.labels(guild_id=member.guild.id, status=str(member.status)).dec()      @Cog.listener()      async def on_member_update(self, before: Member, after: Member) -> None: +        """Update member gauges for the new and old status if applicable."""          if before.status is not after.status:              self.guild_members.labels(guild_id=after.guild.id, status=str(before.status)).dec()              self.guild_members.labels(guild_id=after.guild.id, status=str(after.status)).inc()      @Cog.listener()      async def on_message(self, message: Message) -> None: +        """Increment the guild message counter."""          self.guild_messages.labels(              channel_id=message.channel.id, channel_name=message.channel.name          ).inc()  |