Added tags extension from the other repo

leek/tags/__init__.py

@@ -0,0 +1,2 @@
+# flake8: noqa: F401
+from .tags import Tags

leek/tags/tags.en-US.json

@@ -0,0 +1,13 @@
+{
+    "TAGS_NOT_FOUND": "The tag `{0}` does not exists.",
+    "TAGS_ADD_OKAY": "The tag `{0}` was creado!",
+    "TAGS_ADD_DUPE": "The tag `{0}` already exists!",
+    "TAGS_DELETE_OKAY": "The tag `{0}` was successfully deleted!`",
+
+    "TAGS_COMMAND_TAG_NAME": "tag",
+    "TAGS_COMMAND_TAG_DESC": "Views a specific tag.",
+    "TAGS_COMMAND_CREATETAG_NAME": "createtag",
+    "TAGS_COMMAND_CREATETAG_DESC": "Creates a new tag.",
+    "TAGS_COMMAND_DELETETAG_NAME": "deletetag",
+    "TAGS_COMMAND_DELETETAG_DESC": "Deletes an existing tag."
+}

leek/tags/tags.es-ES.json

@@ -0,0 +1,13 @@
+{
+    "TAGS_NOT_FOUND": "El tag `{0}` no existe.",
+    "TAGS_ADD_OKAY": "El tag `{0}` fue creado!",
+    "TAGS_ADD_DUPE": "El tag `{0}` ya existe!",
+    "TAGS_DELETE_OKAY": "El tag `{0}` fue borrado satisfactoriamente.`",
+
+    "TAGS_COMMAND_TAG_NAME": "tag",
+    "TAGS_COMMAND_TAG_DESC": "Muestra un tag en especifico.",
+    "TAGS_COMMAND_CREATETAG_NAME": "creartag",
+    "TAGS_COMMAND_CREATETAG_DESC": "Crea un nuevo tag.",
+    "TAGS_COMMAND_DELETETAG_NAME": "borrartag",
+    "TAGS_COMMAND_DELETETAG_DESC": "Borra un tag existente."
+}

leek/tags/tags.py

@@ -0,0 +1,92 @@
+from aiomysql import Cursor
+from discord import Cog, ApplicationContext, slash_command, Option, AutocompleteContext, Permissions
+from leek import DatabaseRequiredError, LeekBot, localize, get_localizations, get_default
+from pymysql import IntegrityError
+
+PERMISSIONS = Permissions(manage_messages=True)
+CREATE = "CREATE TABLE IF NOT EXISTS tags_%s (id INT NOT NULL auto_increment, name TEXT NOT NULL UNIQUE, " \
+         "content TEXT NOT NULL, primary key (id))"
+FETCH_ALL = "SELECT (name) FROM tags_%s"
+FETCH_SINGLE = "SELECT (content) FROM tags_%s WHERE name=%s"
+ADD = "INSERT INTO tags_%s (name, content) VALUES (%s, %s)"
+DELETE = "DELETE FROM tags_%s WHERE name=%s"
+
+
+async def get_tag_names(ctx: AutocompleteContext):
+    if not isinstance(ctx.bot, LeekBot):
+        return []
+
+    bot: LeekBot = ctx.bot
+
+    if not bot.is_pool_available:
+        return []
+
+    async with bot.connection as connection:
+        cursor: Cursor = await connection.cursor()
+        await cursor.execute(CREATE, [ctx.interaction.guild.id])
+        await cursor.execute(FETCH_ALL, [ctx.interaction.guild.id])
+        await connection.commit()
+        tags = await cursor.fetchall()
+        await cursor.close()
+        return [x[0] for x in tags]
+
+
+class Tags(Cog):
+    def __init__(self, bot: LeekBot):
+        self.bot: LeekBot = bot
+
+    async def cog_before_invoke(self, ctx: ApplicationContext):
+        if not self.bot.is_pool_available:
+            raise DatabaseRequiredError(self)
+
+    @slash_command(name_localizations=get_localizations("TAGS_COMMAND_TAG_NAME"),
+                   description=get_default("TAGS_COMMAND_TAG_DESC"),
+                   description_localization=get_localizations("TAGS_COMMAND_TAG_DESC"))
+    async def tag(self, ctx: ApplicationContext, name: Option(str, "The tag name", autocomplete=get_tag_names)):
+        async with self.bot.connection as connection:
+            cursor: Cursor = await connection.cursor()
+            await cursor.execute(CREATE, [ctx.interaction.guild.id])
+            await cursor.execute(FETCH_SINGLE, [ctx.interaction.guild.id, name])
+            await connection.commit()
+            tag = await cursor.fetchone()
+            await cursor.close()
+
+        if tag is None:
+            await ctx.respond(localize("TAGS_NOT_FOUND", ctx.locale, name), ephemeral=True)
+        else:
+            await ctx.respond(tag[0])
+
+    @slash_command(name_localizations=get_localizations("TAGS_COMMAND_CREATETAG_NAME"),
+                   description=get_default("TAGS_COMMAND_CREATETAG_DESC"),
+                   description_localization=get_localizations("TAGS_COMMAND_CREATETAG_DESC"),
+                   default_member_permissions=PERMISSIONS)
+    async def createtag(self, ctx: ApplicationContext, name: Option(str, "The tag name"),
+                        content: Option(str, "The text content of the tag")):
+        try:
+            async with self.bot.connection as connection:
+                cursor: Cursor = await connection.cursor()
+                await cursor.execute(CREATE, [ctx.interaction.guild.id])
+                await cursor.execute(ADD, [ctx.interaction.guild.id, name, content])
+                await connection.commit()
+                await cursor.close()
+                await ctx.respond(localize("TAGS_ADD_OKAY", ctx.locale, name), ephemeral=True)
+        except IntegrityError:
+            await ctx.respond(localize("TAGS_ADD_DUPE", ctx.locale, name), ephemeral=True)
+
+    @slash_command(name_localizations=get_localizations("TAGS_COMMAND_DELETETAG_NAME"),
+                   description=get_default("TAGS_COMMAND_DELETETAG_DESC"),
+                   description_localization=get_localizations("TAGS_COMMAND_DELETETAG_DESC"),
+                   default_member_permissions=PERMISSIONS)
+    async def deletetag(self, ctx: ApplicationContext, name: Option(str, "The tag name", autocomplete=get_tag_names)):
+        async with self.bot.connection as connection:
+            cursor: Cursor = await connection.cursor()
+            await cursor.execute(CREATE, [ctx.interaction.guild.id])
+            await cursor.execute(DELETE, [ctx.interaction.guild.id, name])
+            await connection.commit()
+            rows = cursor.rowcount
+            await cursor.close()
+
+        if rows == 0:
+            await ctx.respond(localize("TAGS_NOT_FOUND", ctx.locale, name), ephemeral=True)
+        else:
+            await ctx.respond(localize("TAGS_DELETE_OKAY", ctx.locale, name), ephemeral=True)