80 lines
No EOL
3.7 KiB
Python
80 lines
No EOL
3.7 KiB
Python
from discord import app_commands
|
|
import discord
|
|
from GammaRPBot_Define import GammaRPBot
|
|
import aiosqlite
|
|
import base64
|
|
|
|
|
|
_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,guild_id,image) VALUES (?,?,?,?,?,?,?,?,?)",(nom_court,nom_complet,histoire,pouvoirs,age,sexe,uuid,guild_id,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="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()
|
|
message=""
|
|
for row in rows:
|
|
print(row)
|
|
message+=f"Nom court: {row[0]}\nNom complet: {row[1]}\nHistoire: {row[3]}\nPouvoirs: {row[4]}\nAge: {row[6]}\nSexe: {row[7]}\n\n"
|
|
await interaction.response.send_message("Liste des personnages\n"+message,ephemeral=True)
|
|
|