This commit is contained in:
parent
2dada0dd80
commit
db4d51f5a3
3 changed files with 125 additions and 0 deletions
3
translator/__init__.py
Normal file
3
translator/__init__.py
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
from .translator import setup
|
||||||
|
|
||||||
|
__red_end_user_data_statement__ = "This cog does not persistently store data about users."
|
21
translator/info.json
Normal file
21
translator/info.json
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
{
|
||||||
|
"name": "Translator",
|
||||||
|
"short": "Translate messages in different languages",
|
||||||
|
"description": "A cog that allows you to translate messages by replying to them. Supports multiple languages using Google Translate.",
|
||||||
|
"end_user_data_statement": "This cog does not store any user data.",
|
||||||
|
"author": [
|
||||||
|
"Valerie"
|
||||||
|
],
|
||||||
|
"required_cogs": {},
|
||||||
|
"requirements": [
|
||||||
|
"googletrans==3.1.0a0"
|
||||||
|
],
|
||||||
|
"tags": [
|
||||||
|
"utility",
|
||||||
|
"translation"
|
||||||
|
],
|
||||||
|
"min_bot_version": "3.5.0",
|
||||||
|
"hidden": false,
|
||||||
|
"disabled": false,
|
||||||
|
"type": "COG"
|
||||||
|
}
|
101
translator/translator.py
Normal file
101
translator/translator.py
Normal file
|
@ -0,0 +1,101 @@
|
||||||
|
from redbot.core import commands
|
||||||
|
from googletrans import Translator, LANGUAGES
|
||||||
|
import discord
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
class TranslatorCog(commands.Cog):
|
||||||
|
"""Translate messages using Google Translate"""
|
||||||
|
|
||||||
|
def __init__(self, bot):
|
||||||
|
self.bot = bot
|
||||||
|
self.translator = Translator()
|
||||||
|
self.languages = LANGUAGES
|
||||||
|
|
||||||
|
@commands.command(name="translate")
|
||||||
|
async def translate_text(self, ctx, *, message=None):
|
||||||
|
"""Translate a message to English by replying to it
|
||||||
|
You can also specify a language code after your message to translate to that language
|
||||||
|
Example: [p]translate bonjour fr -> en
|
||||||
|
"""
|
||||||
|
if not message and not ctx.message.reference:
|
||||||
|
await ctx.send("Please either provide text to translate or reply to a message to translate it.")
|
||||||
|
return
|
||||||
|
|
||||||
|
# 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
|
||||||
|
|
||||||
|
# Check if language was specified in the command
|
||||||
|
if message:
|
||||||
|
try:
|
||||||
|
source_lang, dest_lang = message.split("->")
|
||||||
|
source_lang = source_lang.strip()
|
||||||
|
dest_lang = dest_lang.strip()
|
||||||
|
except ValueError:
|
||||||
|
dest_lang = "en"
|
||||||
|
source_lang = "auto"
|
||||||
|
else:
|
||||||
|
dest_lang = "en"
|
||||||
|
source_lang = "auto"
|
||||||
|
else:
|
||||||
|
# Direct message translation
|
||||||
|
if "->" in message:
|
||||||
|
try:
|
||||||
|
text, langs = message.rsplit("->", 1)
|
||||||
|
text_to_translate = text.strip()
|
||||||
|
dest_lang = langs.strip()
|
||||||
|
source_lang = "auto"
|
||||||
|
except ValueError:
|
||||||
|
text_to_translate = message
|
||||||
|
dest_lang = "en"
|
||||||
|
source_lang = "auto"
|
||||||
|
else:
|
||||||
|
text_to_translate = message
|
||||||
|
dest_lang = "en"
|
||||||
|
source_lang = "auto"
|
||||||
|
|
||||||
|
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()}:
|
||||||
|
await ctx.send(f"Invalid destination language code. Available languages: {', '.join(self.languages.keys())}")
|
||||||
|
return
|
||||||
|
|
||||||
|
# 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)
|
||||||
|
|
||||||
|
# Create embed
|
||||||
|
embed = discord.Embed(title="Translation", color=discord.Color.blue())
|
||||||
|
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(embed=embed)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
await ctx.send(f"An error occurred while translating: {str(e)}")
|
||||||
|
|
||||||
|
@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)
|
||||||
|
|
||||||
|
async def setup(bot):
|
||||||
|
await bot.add_cog(TranslatorCog(bot))
|
Loading…
Add table
Reference in a new issue