Add repl.py

v_repl_bot/__init__.py

@@ -0,0 +1,90 @@
+from __future__ import annotations
+
+__all__ = ("ReplBot",)
+
+import asyncio
+from inspect import cleandoc as strip
+from itertools import chain
+from pathlib import Path
+from time import time as get_time
+from traceback import format_exc as format_exception
+
+from aiohttp import ClientSession as AIOHTTPSession
+from discord import AllowedMentions, Game, Intents, Webhook
+from discord.ext.commands import (
+    Bot as CommandsBot,
+    ExtensionFailed,
+    NoEntryPointError,
+    when_mentioned_or,
+)
+
+
+class ReplBot(CommandsBot):
+    ready_timestamp: float
+    log_webhook: Webhook
+    session: AIOHTTPSession
+
+    def __init__(self, *, token: str, webhook_url: str) -> None:
+        self.token = token
+        self.webhook_url = webhook_url
+
+        super().__init__(
+            command_prefix = when_mentioned_or("&"),
+            strip_after_prefix = True,
+            case_insensitive = True,
+            status = Game("on play.minearchy.com"),
+            owner_ids = { 512640455834337290 },
+            allowed_mentions = AllowedMentions.none(),
+            max_messages = 100,
+            intents = Intents(
+                messages = True,
+                message_content = True,
+            ),
+            help_attrs = dict(
+                brief = "Sends help.",
+                help = "Sends all the commands of the bot, or help of a specific command or module.",
+            ),
+        )
+
+    async def on_ready(self) -> None:
+        print(
+            strip(
+                f"""
+                Connected to Discord!
+                User: {self.user}
+                ID: {self.user.id}
+                """
+            )
+        )
+
+        self.ready_timestamp = get_time()
+        await self.log_webhook.send("Bot is now online!")
+
+    async def load_extensions(self) -> None:
+        cogs = Path(__file__).parent / "cogs"
+        for file_name in chain(
+            map(
+                lambda file_path: ".".join(file_path.relative_to(cogs.parent.parent).parts)[:-3],
+                cogs.rglob("*.py"),
+            ),
+            ("jishaku",),
+        ):
+            try:
+                await self.load_extension(file_name)
+                print(f"Loaded {file_name}")
+            except (ExtensionFailed, NoEntryPointError):
+                print(f"Couldn't load {file_name}:\n{format_exception()}")
+
+    def run(self) -> None:
+        async def runner() -> None:
+            async with self, AIOHTTPSession() as self.session:
+                self.log_webhook = Webhook.from_url(
+                    self.webhook_url, session = self.session, bot_token = self.token
+                )
+                await self.load_extensions()
+                await self.start(self.token)
+
+        try:
+            asyncio.run(runner())
+        except KeyboardInterrupt:
+            pass