Display dev in logging webhook (#43)

* lints -_-

* Add config key for logging in dev mode

* use get_or_fetch_user

* fix: nitpicks

Author: IAmTomahawkx

config.template.toml

@@ -18,6 +18,7 @@ dsn = 'postgres://pythonistabot:pythonistabot@database:5432/pythonistabot' # ass
 [LOGGING]
 webhook_url = ""  # optional
 webhook_avatar_url = "" # optional
+runner = 123456789 # optional: sets the webhook avatar url (will be overridden by the webhook_avatar_url attribute)
 level = 20
 
 [SNEKBOX] # optional

modules/logging.py

@@ -27,14 +27,17 @@
 
 import discord
 from discord.ext import commands, tasks
-from discord.utils import format_dt
+from discord.utils import MISSING, format_dt
 
 import core
 
 
 class Logging(commands.Cog):
     def __init__(self, bot: core.Bot) -> None:
         self.bot = bot
+
+        self.user: discord.User | None = MISSING  # none is for a failed fetch
+
         if url := core.CONFIG["LOGGING"].get("webhook_url"):
             self.webhook = discord.Webhook.from_url(url, session=bot.session, client=bot)
         else:
@@ -64,9 +67,23 @@ async def logging_loop(self) -> None:
 
         embed = to_log.__dict__.get("embed") or discord.utils.MISSING
 
+        avatar_url: str | None = core.CONFIG["LOGGING"].get("webhook_avatar_url")
+        actor_name = "PythonistaBot Logging"
+
+        if not avatar_url and self.user is not None and "runner" in core.CONFIG["LOGGING"]:
+            runner_id: int = core.CONFIG["LOGGING"]["runner"]
+            try:
+                user = self.user or self.bot.get_or_fetch_user(runner_id)
+            except:
+                self.user = user = None  # This will tell us not to attempt again.
+
+            if user:
+                avatar_url = user.display_avatar.url
+                actor_name = f"Logging: Dev: {user.display_name}"
+
         await self.webhook.send(
             message,
-            username="PythonistaBot Logging",
+            username=actor_name,
             avatar_url=core.CONFIG["LOGGING"].get("webhook_avatar_url"),
             embed=embed,
         )

types_/config.py

@@ -20,6 +20,7 @@ class Logging(TypedDict):
     webhook_url: NotRequired[str]
     webhook_avatar_url: NotRequired[str]
     level: int
+    runner: NotRequired[int]
 
 
 class Snekbox(TypedDict):