Migrated to cogs

2023-01-17 15:50:47 +01:00
parent be9ed44614
commit e56268543c
5 changed files with 213 additions and 174 deletions
--- a/cogs/cogLocale.py
+++ b/cogs/cogLocale.py
@@ -0,0 +1,48 @@
+from os import listdir
+from discord import ApplicationContext, Option, SlashCommandGroup
+from discord.ext import commands
+from functions import appendLog, getMsg, guildConfGet, guildConfReset, guildConfSet, loadJson, makeEmbed, strToColor
+
+#=========================================================================================================================
+class CogLocale(commands.Cog):
+
+    def __init__(self, client):
+        self.client = client
+
+    locale = SlashCommandGroup("locale", "Commands related to bot's locale")
+
+    valid_locales = []
+    files_locales = listdir(f"locale/")
+    for entry in files_locales:
+        valid_locales.append(".".join(entry.split(".")[:-1]))
+
+    @locale.command(name="set", description="Set bot's messages language")
+    async def locale_set(self, ctx: ApplicationContext, language: Option(str, "One of the languages in list", choices=valid_locales)): # type: ignore
+        config = loadJson("config.json")
+        if ctx.guild is not None:
+            if language+".json" in listdir(f"locale/"):
+                guildConfSet(ctx.guild, "locale", language)
+                appendLog(f"Server's locale is now set to {language}", ctx.guild)
+                await ctx.respond(embed=makeEmbed(title=getMsg("set_locale_title", ctx.guild), description=getMsg("set_locale_description", ctx.guild).format(getMsg("locale_name", ctx.guild)), color=strToColor(config["color_ok"])))
+            else:
+                valid_locales = []
+                files_locales = listdir(f"locale/")
+                for entry in files_locales:
+                    valid_locales.append(entry.split(".")[:-1])
+                await ctx.respond(embed=makeEmbed(title=getMsg("error_locale_title", ctx.guild), description=getMsg("error_locale_description", ctx.guild).format(", ".join(valid_locales)), color=strToColor(config["color_error"])))
+        else:
+            await ctx.respond(embed=makeEmbed(title=getMsg("dm_title", ctx.guild), description=getMsg("dm_description", ctx.guild), color=strToColor(config["color_error"])))
+
+    @locale.command(name="reset", description="Reset the bot's language in this guild")
+    async def locale_reset(self, ctx: ApplicationContext): # type: ignore
+        config = loadJson("config.json")
+        if ctx.guild is not None:
+            if guildConfGet(ctx.guild, "locale") is not None:
+                guildConfReset(ctx.guild, "locale")
+                appendLog(f"Server's locale has been reset", ctx.guild)
+                await ctx.respond(embed=makeEmbed(title=getMsg("reset_locale_title", ctx.guild), description=getMsg("reset_locale_description", ctx.guild).format(getMsg("locale_name", ctx.guild)), color=strToColor(config["color_ok"])))
+            else:
+                await ctx.respond(embed=makeEmbed(title=getMsg("hint_none_locale_title", ctx.guild), description=getMsg("hint_none_locale_description", ctx.guild).format(getMsg("locale_name", ctx.guild)), color=strToColor(config["color_warn"])))
+        else:
+            await ctx.respond(embed=makeEmbed(title=getMsg("dm_title", ctx.guild), description=getMsg("dm_description", ctx.guild), color=strToColor(config["color_error"])))
+#=========================================================================================================================