Optimize formatter.py

lib/formatter.py

@@ -1,25 +1,26 @@
 import textwrap
 
 import discord
+from discord.ext import commands
 
 from services.config_service import GuildConfig
 
 
-def template(text, username, level=None):
+def template(text: str, username: str, level: int | None = None) -> str:
     """
     Replaces placeholders in the given text with actual values.
 
     Args:
         text (str): The template text containing placeholders.
-        username (str): The username to replace "{user}" placeholder.
-        level (int, optional): The level to replace "{level}" placeholder. Defaults to None.
+        username (str): The username to replace the "{user}" placeholder.
+        level (int | None, optional): The level to replace the "{level}" placeholder. Defaults to None.
 
     Returns:
-        str: The formatted text.
+        str: The formatted text with placeholders replaced by actual values.
     """
-    replacements = {
+    replacements: dict[str, str] = {
         "{user}": username,
-        "{level}": str(level) if level is not None else ""
+        "{level}": str(level) if level is not None else "",
     }
 
     for placeholder, value in replacements.items():
@@ -28,27 +29,39 @@ def template(text, username, level=None):
     return text
 
 
-def shorten(text, width) -> str:
+def shorten(text: str, width: int = 200) -> str:
     return textwrap.shorten(text, width=width, placeholder="...")
 
 
-def get_prefix(ctx):
+def get_prefix(ctx: commands.Context) -> str:
     """
-    Attempt to get the prefix.
+    Attempts to retrieve the prefix for the given guild context.
+
+    Args:
+        ctx (discord.ext.commands.Context): The context of the command invocation.
+
+    Returns:
+        str: The prefix for the guild. Defaults to "." if the guild or prefix is not found.
     """
     try:
-        return GuildConfig.get_prefix(ctx.guild.id)
-    except AttributeError:
+        return GuildConfig.get_prefix(ctx.guild.id if ctx.guild else 0)
+    except (AttributeError, TypeError):
         return "."
 
 
-def get_invoked_name(ctx):
+def get_invoked_name(ctx: commands.Context) -> str | None:
     """
-    Attempts to get the alias of the command used, if the user did a SlashCommand, return the name.
+    Attempts to get the alias of the command used. If the user used a SlashCommand, return the command name.
+
+    Args:
+        ctx (discord.ext.commands.Context): The context of the command invocation.
+
+    Returns:
+        str: The alias or name of the invoked command.
     """
     try:
-        invoked_with = ctx.invoked_with
+        invoked_with: str | None = ctx.invoked_with
     except (discord.ApplicationCommandInvokeError, AttributeError):
-        invoked_with = ctx.command.name
+        invoked_with: str | None = ctx.command.name if ctx.command else None
 
     return invoked_with