|
|
|
@@ -1,12 +1,10 @@
|
|
|
|
|
"""
|
|
|
|
|
discord channel logger
|
|
|
|
|
|
|
|
|
|
leveled logging to a discord channel via embeds. attach to your bot
|
|
|
|
|
(e.g. bot.log) and call bot.log.info(...), bot.log.critical(...), etc.
|
|
|
|
|
|
|
|
|
|
config-free: all static identity is injected at construction; dynamic
|
|
|
|
|
per-guild channel routing is read live from bot.settings at call time, so a
|
|
|
|
|
command that mutates bot.settings changes routing without a restart.
|
|
|
|
|
leveled logging to a discord channel via embeds. attach to your bot (e.g.
|
|
|
|
|
bot.log) and call bot.log.info(...), bot.log.critical(...), etc. config-free:
|
|
|
|
|
static identity is injected at construction; per-guild channel routing is
|
|
|
|
|
read live from bot.settings at call time (no restart needed).
|
|
|
|
|
|
|
|
|
|
from dpy_logger import DPYLogger
|
|
|
|
|
|
|
|
|
@@ -25,31 +23,32 @@ levels: debug, info, success, fail (alias failure), task, critical.
|
|
|
|
|
|
|
|
|
|
dynamic routing: a per-call guild= argument logs to that guild's configured
|
|
|
|
|
channel via bot.settings[guild.id]['channels']['logs']; omit it to use the
|
|
|
|
|
channel this logger was constructed with.
|
|
|
|
|
construction channel.
|
|
|
|
|
|
|
|
|
|
custom embeds: pass embed_builder=fn to restyle without subclassing, where
|
|
|
|
|
fn(logger, level, action, actor, details) -> discord.Embed. every level
|
|
|
|
|
(including critical) routes through it. for complex cases override the
|
|
|
|
|
build_embed method in a subclass instead.
|
|
|
|
|
custom embeds: pass embed_builder=fn(logger, level, action, actor, details)
|
|
|
|
|
-> discord.Embed to restyle without subclassing; every level routes through
|
|
|
|
|
it. for complex cases override build_embed in a subclass instead.
|
|
|
|
|
|
|
|
|
|
dual sink: every call also mirrors to the stdlib logger (getLogger(__name__)),
|
|
|
|
|
which the app routes to file/console. set log_to_file=False at construction to
|
|
|
|
|
disable, or pass log_to_file=True/False per call to override. the stdlib emit
|
|
|
|
|
happens before the discord send, so the record survives even if discord fails.
|
|
|
|
|
dual sink: every call also mirrors to the stdlib logger (getLogger(__name__));
|
|
|
|
|
the stdlib emit happens before the discord send, so the record survives even
|
|
|
|
|
if discord fails. set log_to_file=False at construction, or per call, to
|
|
|
|
|
disable.
|
|
|
|
|
|
|
|
|
|
extending: this base has no feed/announcement method by design. a project
|
|
|
|
|
that wants one subclasses DPYLogger and adds it, reusing _resolve.
|
|
|
|
|
extending: no feed/announcement method by design; subclass and reuse
|
|
|
|
|
_resolve for project-specific log types.
|
|
|
|
|
|
|
|
|
|
errors: resolution failures (unresolvable guild/channel, bad config) raise
|
|
|
|
|
ValueError from `initialize()` — a misconfigured logger should fail loudly at
|
|
|
|
|
setup. on a per-call send, resolution AND send failures do NOT propagate: they
|
|
|
|
|
fall back to the stdlib logger so a transient discord failure (or a per-call
|
|
|
|
|
`guild=` that doesn't resolve) never breaks the caller's command.
|
|
|
|
|
errors: setup raises, sends swallow. resolution failures (unresolvable
|
|
|
|
|
guild/channel, bad config, a non-text channel, an invalid timezone) raise
|
|
|
|
|
ValueError from `initialize()`. underlying discord exceptions (NotFound/
|
|
|
|
|
Forbidden/HTTPException/InvalidData and other ClientException subclasses)
|
|
|
|
|
are normalized to that same ValueError. on a per-call send, resolution and
|
|
|
|
|
send failures never propagate — they fall back to the stdlib logger so a
|
|
|
|
|
transient discord failure never breaks the caller's command.
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
import logging
|
|
|
|
|
import discord
|
|
|
|
|
from datetime import datetime
|
|
|
|
|
from datetime import datetime, tzinfo
|
|
|
|
|
from typing import Callable, Optional
|
|
|
|
|
|
|
|
|
|
_log = logging.getLogger(__name__)
|
|
|
|
@@ -63,6 +62,8 @@ DEFAULT_COLORS = {
|
|
|
|
|
"critical": 0x000000,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
DISCORD_CONTENT_LIMIT = 2000
|
|
|
|
|
|
|
|
|
|
LEVEL_MAP = {
|
|
|
|
|
"debug": logging.DEBUG,
|
|
|
|
|
"info": logging.INFO,
|
|
|
|
@@ -105,12 +106,24 @@ class DPYLogger:
|
|
|
|
|
|
|
|
|
|
async def initialize(self):
|
|
|
|
|
"""resolve guild/channel from ids to objects; call once after construction"""
|
|
|
|
|
if self.timezone is not None and not isinstance(self.timezone, tzinfo):
|
|
|
|
|
raise ValueError(f"[dpy_logger] timezone must be a tzinfo instance, not {self.timezone!r}")
|
|
|
|
|
if isinstance(self.guild, int):
|
|
|
|
|
self.guild = await self._get_guild(self.guild)
|
|
|
|
|
if isinstance(self.channel, int):
|
|
|
|
|
if not self.guild:
|
|
|
|
|
raise ValueError(f"[dpy_logger] cannot resolve channel {self.channel} without a guild")
|
|
|
|
|
self.channel = await self._get_channel(self.guild)
|
|
|
|
|
# a channel with no guild would otherwise blackout every send silently; derive
|
|
|
|
|
# the guild from it, or fail loud here rather than later
|
|
|
|
|
if self.guild is None and isinstance(self.channel, discord.TextChannel):
|
|
|
|
|
self.guild = self.channel.guild
|
|
|
|
|
if self.guild is None and self.channel is not None:
|
|
|
|
|
raise ValueError("[dpy_logger] channel provided without a resolvable guild")
|
|
|
|
|
# a wrong-type channel object (Thread/VoiceChannel/ForumChannel) would also pass
|
|
|
|
|
# silently and misroute/blackhole later; fail loud here instead
|
|
|
|
|
if self.channel is not None and not isinstance(self.channel, (discord.TextChannel, int)):
|
|
|
|
|
raise ValueError(f"[dpy_logger] channel {self.channel!r} is not a text channel")
|
|
|
|
|
|
|
|
|
|
async def _get_guild(self, guild):
|
|
|
|
|
"""resolve a guild from id-or-object, raising if unresolvable"""
|
|
|
|
@@ -119,9 +132,7 @@ class DPYLogger:
|
|
|
|
|
resolved = self.bot.get_guild(guild)
|
|
|
|
|
if resolved is not None:
|
|
|
|
|
return resolved
|
|
|
|
|
# fetch_guild never returns None — it raises NotFound/Forbidden/
|
|
|
|
|
# HTTPException; normalize those to the lib's ValueError so callers see
|
|
|
|
|
# one error type at setup
|
|
|
|
|
# fetch_guild raises rather than returning None; normalize to ValueError
|
|
|
|
|
try:
|
|
|
|
|
return await self.bot.fetch_guild(guild)
|
|
|
|
|
except discord.HTTPException as error:
|
|
|
|
@@ -139,10 +150,13 @@ class DPYLogger:
|
|
|
|
|
if isinstance(self.channel, discord.TextChannel):
|
|
|
|
|
return self.channel
|
|
|
|
|
if isinstance(self.channel, int):
|
|
|
|
|
# fetch_channel raises NotFound/Forbidden/HTTPException/InvalidData (a
|
|
|
|
|
# ClientException, not an HTTPException); normalize all to ValueError
|
|
|
|
|
try:
|
|
|
|
|
channel = await guild.fetch_channel(self.channel)
|
|
|
|
|
except (discord.HTTPException, discord.ClientException) as error:
|
|
|
|
|
raise ValueError(f"[dpy_logger] could not fetch channel {self.channel}: {error}") from error
|
|
|
|
|
if not isinstance(channel, discord.TextChannel):
|
|
|
|
|
# fetch_channel can return a Voice/Category/Forum channel; fail loud
|
|
|
|
|
# at setup like the settings path, not later via an AttributeError on .send
|
|
|
|
|
raise ValueError(f"[dpy_logger] channel {self.channel} is not a text channel")
|
|
|
|
|
return channel
|
|
|
|
|
|
|
|
|
@@ -152,9 +166,7 @@ class DPYLogger:
|
|
|
|
|
raise ValueError(f"[dpy_logger] no log channel configured for guild {guild.id}")
|
|
|
|
|
try:
|
|
|
|
|
channel = await guild.fetch_channel(channel_id)
|
|
|
|
|
except discord.HTTPException as error:
|
|
|
|
|
# fetch_channel raises NotFound/Forbidden/HTTPException; normalize to the
|
|
|
|
|
# lib's ValueError so a bad configured id fails loud with one error type
|
|
|
|
|
except (discord.HTTPException, discord.ClientException) as error:
|
|
|
|
|
raise ValueError(f"[dpy_logger] could not fetch channel {channel_id}: {error}") from error
|
|
|
|
|
if not isinstance(channel, discord.TextChannel):
|
|
|
|
|
raise ValueError(f"[dpy_logger] configured channel {channel_id} is not a text channel")
|
|
|
|
@@ -171,14 +183,13 @@ class DPYLogger:
|
|
|
|
|
if self._embed_builder is not None:
|
|
|
|
|
return self._embed_builder(self, level, action, actor, details)
|
|
|
|
|
em = discord.Embed(color=self.colors[level])
|
|
|
|
|
if action:
|
|
|
|
|
# is-not-None keeps falsy-but-valid values (0, False) rendering
|
|
|
|
|
if action is not None and str(action) != "":
|
|
|
|
|
em.add_field(name="Action", value=f"`{action}`", inline=True)
|
|
|
|
|
if actor:
|
|
|
|
|
if actor is not None and str(actor) != "":
|
|
|
|
|
em.add_field(name="Actor", value=f"`{actor}`", inline=True)
|
|
|
|
|
# discord rejects an empty field value (50035) and truncates nothing itself, so
|
|
|
|
|
# an empty or >1024-char message would 400 the send; substitute + cap to keep
|
|
|
|
|
# every logging call producing a valid embed
|
|
|
|
|
log_value = str(details) if details else "(no message)"
|
|
|
|
|
# discord 400s an empty or >1024-char field value; substitute + cap
|
|
|
|
|
log_value = str(details) if details is not None and str(details) != "" else "(no message)"
|
|
|
|
|
if len(log_value) > 1024:
|
|
|
|
|
log_value = log_value[:1021] + "..."
|
|
|
|
|
em.add_field(name="Log", value=log_value, inline=False)
|
|
|
|
@@ -192,20 +203,26 @@ class DPYLogger:
|
|
|
|
|
|
|
|
|
|
def _emit_stdlib(self, level, action, actor, log_msg):
|
|
|
|
|
"""mirror the log line to the stdlib logger (routed to file/console by the app)"""
|
|
|
|
|
parts = [p for p in (action, str(actor) if actor else None, log_msg) if p]
|
|
|
|
|
raw = (action, str(actor) if actor is not None else None, log_msg)
|
|
|
|
|
parts = [str(p) for p in raw if p is not None and str(p) != ""]
|
|
|
|
|
_log.log(LEVEL_MAP.get(level, logging.INFO), f"[{level}] " + " | ".join(parts))
|
|
|
|
|
|
|
|
|
|
async def _send(self, level, log_msg, action=None, actor=None, guild=None, log_to_file=None):
|
|
|
|
|
async def _send(self, level, log_msg, action=None, actor=None, guild=None,
|
|
|
|
|
log_to_file=None, content=None):
|
|
|
|
|
"""resolve channel and dispatch a leveled embed; mirror to stdlib unless opted out
|
|
|
|
|
|
|
|
|
|
stdlib emit happens first so the record survives even if the discord send fails.
|
|
|
|
|
a failed send falls back to stdlib rather than propagating into the caller.
|
|
|
|
|
stdlib emit happens first so the record survives a failed discord send, which
|
|
|
|
|
falls back to stdlib rather than propagating. `content` carries the critical-level
|
|
|
|
|
ping/@here text (None for other levels), so every level shares this one contract.
|
|
|
|
|
"""
|
|
|
|
|
if self.log_to_file if log_to_file is None else log_to_file:
|
|
|
|
|
self._emit_stdlib(level, action, actor, log_msg)
|
|
|
|
|
try:
|
|
|
|
|
channel = await self._resolve(guild)
|
|
|
|
|
return await channel.send(embed=self.build_embed(level, action, actor, log_msg))
|
|
|
|
|
return await channel.send(
|
|
|
|
|
content=content,
|
|
|
|
|
embed=self.build_embed(level, action, actor, log_msg),
|
|
|
|
|
)
|
|
|
|
|
except Exception:
|
|
|
|
|
_log.exception(f"[dpy_logger] failed to send {level} log: {log_msg}")
|
|
|
|
|
return None
|
|
|
|
@@ -237,21 +254,17 @@ class DPYLogger:
|
|
|
|
|
return await self._send("task", log, action, "SYSTEM/TASK", guild, log_to_file)
|
|
|
|
|
|
|
|
|
|
async def critical(self, log, action=None, actor=None, guild=None, log_to_file=None):
|
|
|
|
|
"""log a critical-level message and ping configured devs"""
|
|
|
|
|
if self.log_to_file if log_to_file is None else log_to_file:
|
|
|
|
|
self._emit_stdlib("critical", action, actor, log)
|
|
|
|
|
"""log a critical-level message and ping configured devs
|
|
|
|
|
|
|
|
|
|
delegates to _send with the ping/@here content so the send + error contract has a
|
|
|
|
|
single source of truth; critical only adds the alert `content`.
|
|
|
|
|
"""
|
|
|
|
|
if self.pings:
|
|
|
|
|
content = "alert: " + " ".join(f"<@{u}>" for u in self.pings)
|
|
|
|
|
elif self.alert_here:
|
|
|
|
|
content = "alert: @here"
|
|
|
|
|
else:
|
|
|
|
|
content = None
|
|
|
|
|
try:
|
|
|
|
|
channel = await self._resolve(guild)
|
|
|
|
|
return await channel.send(
|
|
|
|
|
content=content,
|
|
|
|
|
embed=self.build_embed("critical", action, actor, log),
|
|
|
|
|
)
|
|
|
|
|
except Exception:
|
|
|
|
|
_log.exception(f"[dpy_logger] failed to send critical log: {log}")
|
|
|
|
|
return None
|
|
|
|
|
if content is not None and len(content) > DISCORD_CONTENT_LIMIT:
|
|
|
|
|
content = content[:DISCORD_CONTENT_LIMIT - 3] + "..."
|
|
|
|
|
return await self._send("critical", log, action, actor, guild, log_to_file, content)
|
|
|
|
|