179 lines
No EOL
8 KiB
Python
179 lines
No EOL
8 KiB
Python
from redbot.core import commands, Config
|
|
from googletrans import Translator, LANGUAGES
|
|
import discord
|
|
from discord import app_commands
|
|
import asyncio
|
|
|
|
class TranslatorCog(commands.Cog):
|
|
"""Translate messages using Google Translate"""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.translator = Translator()
|
|
self.languages = LANGUAGES
|
|
self.config = Config.get_conf(self, identifier=95932766180)
|
|
default_guild = {
|
|
"use_embeds": True
|
|
}
|
|
self.config.register_guild(**default_guild)
|
|
|
|
@commands.group(name="transset")
|
|
@commands.admin_or_permissions(manage_guild=True)
|
|
async def _transset(self, ctx):
|
|
"""Configure translator settings"""
|
|
pass
|
|
|
|
@_transset.command(name="embed")
|
|
async def _transset_embed(self, ctx, toggle: bool):
|
|
"""Toggle whether to use embeds for translations or plain text"""
|
|
await self.config.guild(ctx.guild).use_embeds.set(toggle)
|
|
await ctx.send(f"Embed display has been {'enabled' if toggle else 'disabled'}.")
|
|
|
|
async def translate_and_respond(self, ctx, text_to_translate, dest_lang="en", source_lang="auto", original_author=None, ephemeral=False, reference_message=None):
|
|
try:
|
|
# Validate destination language
|
|
if dest_lang.lower() not in self.languages and dest_lang.lower() not in {v.lower(): k for k, v in self.languages.items()}:
|
|
return await ctx.send(
|
|
f"Invalid language code. Use `{ctx.prefix}langlist` to see available languages.",
|
|
ephemeral=True,
|
|
reference=reference_message
|
|
)
|
|
|
|
# Convert language name to code if full name was provided
|
|
if dest_lang.lower() in {v.lower(): k for k, v in self.languages.items()}:
|
|
dest_lang = next(k for k, v in self.languages.items() if v.lower() == dest_lang.lower())
|
|
|
|
# Translate the text
|
|
translation = self.translator.translate(text_to_translate, dest=dest_lang, src=source_lang)
|
|
|
|
use_embeds = await self.config.guild(ctx.guild).use_embeds()
|
|
|
|
if use_embeds:
|
|
embed = discord.Embed(title="Translation", color=discord.Color.blue())
|
|
if original_author:
|
|
embed.set_author(name=f"Original message by {original_author.display_name}", icon_url=original_author.display_avatar.url)
|
|
embed.add_field(name=f"Original ({translation.src}):", value=text_to_translate[:1024], inline=False)
|
|
embed.add_field(name=f"Translation ({translation.dest}):", value=translation.text[:1024], inline=False)
|
|
await ctx.send(
|
|
content=ctx.author.mention,
|
|
embed=embed,
|
|
ephemeral=ephemeral,
|
|
reference=reference_message
|
|
)
|
|
else:
|
|
author_text = f"\nOriginal message by: {original_author.display_name}" if original_author else ""
|
|
response = (
|
|
f"{ctx.author.mention}\n"
|
|
f"**Original** ({translation.src}):{author_text}\n{text_to_translate}\n\n"
|
|
f"**Translation** ({translation.dest}):\n{translation.text}"
|
|
)
|
|
await ctx.send(
|
|
response,
|
|
ephemeral=ephemeral,
|
|
reference=reference_message
|
|
)
|
|
|
|
except Exception as e:
|
|
await ctx.send(
|
|
f"An error occurred while translating: {str(e)}",
|
|
ephemeral=True,
|
|
reference=reference_message
|
|
)
|
|
|
|
@commands.command(name="translate")
|
|
async def translate_text(self, ctx, lang_code: str = "en", *, message=None):
|
|
"""Translate a message
|
|
Reply to a message with just the language code to translate to that language
|
|
Or provide text after the language code to translate that text
|
|
|
|
Examples:
|
|
- Reply to a message: [p]translate jp
|
|
- Translate text: [p]translate fr Hello, how are you?
|
|
- Auto-detect and translate to English: [p]translate This will detect the language
|
|
"""
|
|
# Initialize variables
|
|
text_to_translate = None
|
|
original_author = None
|
|
reference_message = None
|
|
|
|
# If replying to a message
|
|
if ctx.message.reference:
|
|
referenced_msg = await ctx.fetch_message(ctx.message.reference.message_id)
|
|
text_to_translate = referenced_msg.content
|
|
original_author = referenced_msg.author
|
|
reference_message = referenced_msg
|
|
|
|
# If no language code is provided in a reply, default to English
|
|
if not lang_code or lang_code.lower() not in self.languages:
|
|
if message: # If there's additional text, it means the first argument was text, not a lang code
|
|
text_to_translate = f"{lang_code} {message}"
|
|
lang_code = "en"
|
|
else:
|
|
# Valid language code provided
|
|
if message: # If there's a message after the lang code, use that instead of the replied message
|
|
text_to_translate = message
|
|
original_author = ctx.author
|
|
reference_message = ctx.message
|
|
|
|
else:
|
|
# Not replying to a message
|
|
original_author = ctx.author
|
|
reference_message = ctx.message
|
|
|
|
if not message: # Only language code provided without text
|
|
await ctx.send("Please provide text to translate or reply to a message.", reference=ctx.message)
|
|
return
|
|
|
|
# Check if the "lang_code" is actually the start of the message
|
|
if lang_code.lower() not in self.languages:
|
|
text_to_translate = f"{lang_code} {message}" if message else lang_code
|
|
lang_code = "en" # Default to English for direct translation
|
|
else:
|
|
text_to_translate = message
|
|
|
|
# If we still don't have text to translate
|
|
if not text_to_translate:
|
|
await ctx.send("Please provide text to translate or reply to a message.", reference=ctx.message)
|
|
return
|
|
|
|
await self.translate_and_respond(
|
|
ctx,
|
|
text_to_translate,
|
|
dest_lang=lang_code,
|
|
source_lang="auto",
|
|
original_author=original_author,
|
|
reference_message=reference_message
|
|
)
|
|
|
|
@commands.command(name="langlist")
|
|
async def language_list(self, ctx):
|
|
"""List all available language codes"""
|
|
lang_list = []
|
|
for code, lang in self.languages.items():
|
|
lang_list.append(f"`{code}`: {lang}")
|
|
|
|
# Split into chunks of 20 for readability
|
|
chunks = [lang_list[i:i + 20] for i in range(0, len(lang_list), 20)]
|
|
|
|
for i, chunk in enumerate(chunks):
|
|
if i == 0:
|
|
embed = discord.Embed(title="Available Languages",
|
|
description="Use these codes with the translate command\n\n" + "\n".join(chunk),
|
|
color=discord.Color.blue())
|
|
else:
|
|
embed = discord.Embed(description="\n".join(chunk), color=discord.Color.blue())
|
|
await ctx.send(embed=embed, reference=ctx.message if i == 0 else None)
|
|
|
|
@app_commands.command(name="translate")
|
|
@app_commands.describe(
|
|
text="The text to translate",
|
|
to_language="The language to translate to (e.g., 'en' for English)",
|
|
from_language="The language to translate from (optional)"
|
|
)
|
|
async def slash_translate(self, interaction: discord.Interaction, text: str, to_language: str, from_language: str = "auto"):
|
|
"""Translate text from one language to another"""
|
|
await interaction.response.defer(ephemeral=True)
|
|
await self.translate_and_respond(interaction, text, to_language, from_language, interaction.user, ephemeral=True)
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(TranslatorCog(bot)) |