Add get_context method (No docs)

PythonistaGuild · Nov 5, 2024 · f3d84e8 · f3d84e8
1 parent 3b1281c
commit f3d84e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/twitchio/ext/commands/bot.py b/twitchio/ext/commands/bot.py
@@ -292,8 +292,12 @@ async def remove_component(self, name: str, /) -> Component | None:
 
         return component
 
+    def get_context(self, message: ChatMessage, *, cls: Any = None) -> Any:
+        cls = cls or Context
+        return cls(message, bot=self)
+
     async def _process_commands(self, message: ChatMessage) -> None:
-        ctx: Context = Context(message, bot=self)
+        ctx: Context = self.get_context(message)
         await self.invoke(ctx)
 
     async def process_commands(self, message: ChatMessage) -> None: