GammaRPBot/perso_commands.py

from discord import app_commands
import discord
from GammaRPBot_Define import GammaRPBot
import aiosqlite
import base64
import io


_sql_table="""CREATE TABLE personnages (
  nom_court varchar(255),
  nom_complet text,
  image blob,
  histoire text,
  pouvoirs text,
  uuid TEXT,
  age INTEGER,
  sexe TEXT,
  guild_id INTEGER NULL
)"""

async def setup(bot:GammaRPBot):
    bot.tree.add_command(PersoCommands(bot))
class PersoCommands(app_commands.Group):
    bot:GammaRPBot
    def __init__(self,bot):
        super().__init__(name="perso",description="Commandes pour les personnages")
        self.bot=bot

    @app_commands.command(name="créer",description="Crée un personnage")
    @app_commands.describe(short_name="Le nom court du personnage (son ID)", full_name="Le nom complet du personnage",
                            story="L'histoire du personnage",powers="Les pouvoirs du personnage",age="L'âge du personnage",
                            sexe="Le sexe du personnage",
                            # guild_only="Personnage global ou uniquement sur ce serveur",
                            image="L'image du personnage")
    # @app_commands.choices(guild_only=[app_commands.Choice(name="Global",value="global"),app_commands.Choice(name="Ce serveur uniquement", value="guild")])
    async def create_character(self,interaction:discord.Interaction,short_name:str,full_name:str,image:discord.Attachment,story:str,powers:str,age:int,sexe:str,
                                # guild_only:app_commands.Choice[str]
                                ):
        nom_court = short_name
        nom_complet= full_name
        if image.content_type!="image/png" and image.content_type!="image/jpeg":
            await interaction.response.send_message("L'image doit être un fichier PNG ou JPEG",ephemeral=True)
            return
        histoire = story
        pouvoirs = powers
        uuid=interaction.user.id
        guild_id = interaction.guild.id
        image_blob_raw=await image.read()
        image_blob_b64 = base64.b64encode(image_blob_raw).decode("utf-8")
        ext = image.filename.split(".")[-1]
        image_blob=f"data:image/{ext};base64,{image_blob_b64}"
        print(nom_court,nom_complet,histoire,pouvoirs,age,sexe,uuid,guild_id)
        print(image.filename)
        print(image.url)
        print(image_blob)
        try:
            res = await self.bot.db.execute("INSERT INTO personnages (nom_court,nom_complet,histoire,pouvoirs,age,sexe,uuid,image) VALUES (?,?,?,?,?,?,?,?,?)",(nom_court,nom_complet,histoire,pouvoirs,age,sexe,uuid,image_blob))
            print(res)
            await res.close()
            await self.bot.db.commit()
        except Exception as e:
            print(e)
            await interaction.response.send_message("Erreur lors de la création du personnage",ephemeral=True)
            return
        except aiosqlite.Warning as w:
            print(w)
            await interaction.response.send_message("Erreur lors de la création du personnage",ephemeral=True)
            return
        except aiosqlite.Error as e:
            print(e)
            await interaction.response.send_message("Erreur lors de la création du personnage",ephemeral=True)
            return
        await interaction.response.send_message("Personnage créé",ephemeral=True)
    @app_commands.command(name="afficher",description="Affiche un personnage")
    @app_commands.describe(short_name="Le nom court du personnage")
    async def show_character(self,interaction:discord.Interaction,short_name:str):
        res = await self.bot.db.execute("SELECT * FROM personnages WHERE nom_court=? AND uuid=? AND (guild_id=? OR guild_id IS NULL)",(short_name,interaction.user.id,interaction.guild.id))
        row = await res.fetchone()
        await res.close()
        if row is None:
            await interaction.response.send_message("Personnage non trouvé",ephemeral=True)
            return
        embed = await perso_embed(self.bot,row)
        await interaction.response.send_message("Personnage trouvé",embed=embed,ephemeral=True)
    @app_commands.command(name="lister",description="Liste tous les personnages")
    async def list_characters(self,interaction:discord.Interaction):
        print("perso list")
        res = await self.bot.db.execute("SELECT * FROM personnages")
        rows = await res.fetchall()
        await res.close()
        embeds = []
        for row in rows:
            embeds.append(await perso_embed(self.bot,row))
        await interaction.response.send_message("Liste des personnages\n",embeds=embeds,ephemeral=True)


async def perso_embed(bot:GammaRPBot,perso_datas)->discord.embeds.Embed:
    perso_image = perso_datas["image"] #data:image/png;base64,xxxxxx
    perso_image_b64=perso_image.split(",")[1]
    perso_image_raw = base64.b64decode(perso_image_b64)
    perso_image_file = discord.File(io.BytesIO(perso_image_raw))
    a = await bot.get_guild(1037663859621765160).get_channel(1317876780635394068).send(file=perso_image_file)
    print(a.attachments)
    perso_image_url = a.attachments[0].url
    embed = discord.Embed(title=perso_datas["nom_complet"],description=perso_datas["histoire"])
    embed.add_field(name="Pouvoirs",value=perso_datas["pouvoirs"])
    embed.add_field(name="Age",value=perso_datas["age"])
    embed.add_field(name="Sexe",value=perso_datas["sexe"])
    embed.set_footer(text=f"Nom court (ID): {perso_datas['nom_court']}")
    embed.set_thumbnail(url=perso_image_url)
    return embed