refactor(utility): make command group

Introduces a new 'utilities' cog and a 'translate' command for translating text between languages. Refactors the translate functionality from a class-based cog to a function-based command, updates bot and help modules to register the new cog, and ensures proper language autocomplete and error handling.
This commit is contained in:
neoarz
2025-09-28 23:35:39 -04:00
parent 3bc5ebe192
commit a5eff449eb
4 changed files with 45 additions and 25 deletions

2
bot.py
View File

@@ -90,7 +90,7 @@ class DiscordBot(commands.Bot):
if os.path.exists(init_file): if os.path.exists(init_file):
try: try:
await self.load_extension(f"cogs.{folder}") await self.load_extension(f"cogs.{folder}")
if folder not in ["fun", "general", "idevice", "miscellaneous", "moderation", "owner", "sidestore"]: if folder not in ["fun", "general", "idevice", "miscellaneous", "moderation", "owner", "sidestore", "utilities"]:
self.logger.info(f"Loaded extension '{folder}'") self.logger.info(f"Loaded extension '{folder}'")
except Exception as e: except Exception as e:
exception = f"{type(e).__name__}: {e}" exception = f"{type(e).__name__}: {e}"

View File

@@ -166,7 +166,7 @@ class Help(commands.Cog, name="help"):
commands_in_category.append((app_command.name, description)) commands_in_category.append((app_command.name, description))
seen_names.add(app_command.name) seen_names.add(app_command.name)
if hasattr(app_command, 'commands') and category in ["fun", "general", "idevice", "miscellaneous", "moderation", "owner", "sidestore"]: if hasattr(app_command, 'commands') and category in ["fun", "general", "idevice", "miscellaneous", "moderation", "owner", "sidestore", "utilities"]:
for subcommand in app_command.commands: for subcommand in app_command.commands:
if subcommand.name in seen_names: if subcommand.name in seen_names:
continue continue

View File

@@ -0,0 +1,23 @@
import discord
from discord.ext import commands
from discord.ext.commands import Context
from .translate import translate_command
class Utilities(commands.GroupCog, name="utilities"):
def __init__(self, bot) -> None:
self.bot = bot
super().__init__()
@commands.hybrid_command(
name="translate",
description="Translate text to another language"
)
async def translate(self, context, text: str = None, to_lang: str = "en", from_lang: str = None):
return await translate_command()(self, context, text=text, to_lang=to_lang, from_lang=from_lang)
async def setup(bot) -> None:
cog = Utilities(bot)
await bot.add_cog(cog)
bot.logger.info("Loaded extension 'utilities.translate'")

View File

@@ -9,10 +9,8 @@ import json
import urllib.parse import urllib.parse
class Translate(commands.Cog, name="translate"): def translate_command():
def __init__(self, bot) -> None: languages = {
self.bot = bot
self.languages = {
"auto": "Auto-detect", "auto": "Auto-detect",
"en": "English", "en": "English",
"es": "Spanish", "es": "Spanish",
@@ -136,7 +134,7 @@ class Translate(commands.Cog, name="translate"):
"lv": "Latvian", "lv": "Latvian",
} }
async def send_embed(self, context: Context, embed: discord.Embed, *, ephemeral: bool = False) -> None: async def send_embed(context, embed: discord.Embed, *, ephemeral: bool = False) -> None:
interaction = getattr(context, "interaction", None) interaction = getattr(context, "interaction", None)
if interaction is not None: if interaction is not None:
if interaction.response.is_done(): if interaction.response.is_done():
@@ -146,7 +144,7 @@ class Translate(commands.Cog, name="translate"):
else: else:
await context.send(embed=embed) await context.send(embed=embed)
async def _translate_with_google_web(self, text: str, from_lang: str = "auto", to_lang: str = "en") -> dict: async def _translate_with_google_web(text: str, from_lang: str = "auto", to_lang: str = "en") -> dict:
try: try:
base_url = "https://translate.googleapis.com/translate_a/single" base_url = "https://translate.googleapis.com/translate_a/single"
@@ -196,11 +194,11 @@ class Translate(commands.Cog, name="translate"):
except Exception: except Exception:
return None return None
async def language_autocomplete(self, interaction: discord.Interaction, current: str) -> list[app_commands.Choice[str]]: async def language_autocomplete(interaction: discord.Interaction, current: str) -> list[app_commands.Choice[str]]:
current = current.lower() current = current.lower()
choices = [] choices = []
for code, name in self.languages.items(): for code, name in languages.items():
if current in code.lower() or current in name.lower(): if current in code.lower() or current in name.lower():
display_name = f"{code} - {name}" display_name = f"{code} - {name}"
if len(display_name) > 100: if len(display_name) > 100:
@@ -213,7 +211,7 @@ class Translate(commands.Cog, name="translate"):
if not choices: if not choices:
popular = ["en", "es", "fr", "de", "it", "pt", "ru", "ja", "ko", "zh-CN"] popular = ["en", "es", "fr", "de", "it", "pt", "ru", "ja", "ko", "zh-CN"]
for code in popular: for code in popular:
name = self.languages.get(code, code) name = languages.get(code, code)
choices.append(app_commands.Choice(name=f"{code} - {name}", value=code)) choices.append(app_commands.Choice(name=f"{code} - {name}", value=code))
return choices return choices
@@ -229,7 +227,7 @@ class Translate(commands.Cog, name="translate"):
) )
@app_commands.autocomplete(to_lang=language_autocomplete) @app_commands.autocomplete(to_lang=language_autocomplete)
@app_commands.autocomplete(from_lang=language_autocomplete) @app_commands.autocomplete(from_lang=language_autocomplete)
async def translate(self, context: Context, text: str = None, to_lang: str = "en", from_lang: str = None): async def translate(self, context, text: str = None, to_lang: str = "en", from_lang: str = None):
if not text or not text.strip(): if not text or not text.strip():
if context.message and context.message.reference and context.message.reference.resolved: if context.message and context.message.reference and context.message.reference.resolved:
replied_message = context.message.reference.resolved replied_message = context.message.reference.resolved
@@ -241,7 +239,7 @@ class Translate(commands.Cog, name="translate"):
description="The replied message has no text content to translate.", description="The replied message has no text content to translate.",
color=0xE02B2B, color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp") ).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
await self.send_embed(context, embed, ephemeral=True) await send_embed(context, embed, ephemeral=True)
return return
else: else:
embed = discord.Embed( embed = discord.Embed(
@@ -249,34 +247,34 @@ class Translate(commands.Cog, name="translate"):
description="Please provide text to translate or reply to a message with text.", description="Please provide text to translate or reply to a message with text.",
color=0xE02B2B, color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp") ).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
await self.send_embed(context, embed, ephemeral=True) await send_embed(context, embed, ephemeral=True)
return return
if to_lang not in self.languages: if to_lang not in languages:
embed = discord.Embed( embed = discord.Embed(
title="Error", title="Error",
description=f"Invalid target language code: `{to_lang}`. Use the autocomplete feature to see available languages.", description=f"Invalid target language code: `{to_lang}`. Use the autocomplete feature to see available languages.",
color=0xE02B2B, color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp") ).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
await self.send_embed(context, embed, ephemeral=True) await send_embed(context, embed, ephemeral=True)
return return
if from_lang and from_lang not in self.languages: if from_lang and from_lang not in languages:
embed = discord.Embed( embed = discord.Embed(
title="Error", title="Error",
description=f"Invalid source language code: `{from_lang}`. Use the autocomplete feature to see available languages.", description=f"Invalid source language code: `{from_lang}`. Use the autocomplete feature to see available languages.",
color=0xE02B2B, color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp") ).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
await self.send_embed(context, embed, ephemeral=True) await send_embed(context, embed, ephemeral=True)
return return
result = await self._translate_with_google_web(text, from_lang or "auto", to_lang) result = await _translate_with_google_web(text, from_lang or "auto", to_lang)
if result and result.get("translatedText"): if result and result.get("translatedText"):
detected_lang = result.get("detectedSourceLanguage", from_lang or "auto") detected_lang = result.get("detectedSourceLanguage", from_lang or "auto")
from_lang_name = self.languages.get(detected_lang, detected_lang) from_lang_name = languages.get(detected_lang, detected_lang)
to_lang_name = self.languages.get(to_lang, to_lang) to_lang_name = languages.get(to_lang, to_lang)
embed = discord.Embed( embed = discord.Embed(
title="Translation", title="Translation",
@@ -286,18 +284,17 @@ class Translate(commands.Cog, name="translate"):
embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp") embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
embed.set_footer(text=f"{from_lang_name} » {to_lang_name}") embed.set_footer(text=f"{from_lang_name} » {to_lang_name}")
await self.send_embed(context, embed) await send_embed(context, embed)
else: else:
embed = discord.Embed( embed = discord.Embed(
title="Error", title="Error",
description="Translation failed. Please try again later.", description="Translation failed. Please try again later.",
color=0xE02B2B, color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp") ).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
await self.send_embed(context, embed, ephemeral=True) await send_embed(context, embed, ephemeral=True)
async def setup(bot): return translate
await bot.add_cog(Translate(bot))