GammaRPBot/main-bot.py

import discord
import aiosqlite
from discord.ext import commands
from discord import app_commands
import logging
import sys
from public_env import *
from GammaRPBot_Define import GammaRPBot
import base64
formatter = discord.utils._ColourFormatter()
handler = logging.StreamHandler(sys.stdout)
handler.setFormatter(formatter)
logging.basicConfig(level=logging.INFO, handlers=[handler])
logger = logging.getLogger()



intents = discord.Intents.all()
bot = GammaRPBot(intents)

@bot.event
async def on_ready():
    bot.db = await aiosqlite.connect("main.db")
    bot.db.row_factory = aiosqlite.Row
    await bot.load_extension("perso_commands")
    await bot.tree.sync()
    print(f"Logged in as {bot.user}")


@bot.event
async def on_message(message:discord.Message):
    if message.author.bot:
        return
    if message.content.startswith("g!"):
        if len(message.content.split(" "))<2:
            return
        short_name = message.content.split(" ")[0][2:]
        uuid = message.author.id
        guild_id = message.guild.id
        message_str = message.content[message.content.index(" ")+1:]
        if await say_message(uuid,guild_id,short_name,message_str,message.channel):
            await message.delete()

@bot.tree.command(name="s",description="Envoyer un message en tant que personnage")
@app_commands.user_install()
@app_commands.describe(id="Le nom court du personnage",message="Le message à envoyer")
@app_commands.allowed_contexts(guilds=True,dms=True,private_channels=True)
@app_commands.allowed_installs(guilds=True,users=True)
async def say(interaction:discord.Interaction,id:str,message:str):
    if not await say_message(interaction.user.id,(interaction.guild.id if interaction.guild is not None else -1),id,message,interaction.channel):
        print("ok9")
        await interaction.response.send_message("Vous n'avez pas ce personnage",ephemeral=True)
        return
    print("ok10")
    await interaction.response.send_message("Message envoyé",ephemeral=True)


async def say_message(userid:int, guildid:int, shortname:str, message:str,channel:discord.TextChannel):
    # print("HELLOWORODL")
    # print(userid,guildid,shortname,message,channel)
    res = await bot.db.execute("SELECT * FROM personnages WHERE uuid=? AND (guild_id=? OR guild_id IS NULL) AND nom_court=?",(userid,guildid,shortname))
    row = await res.fetchone()
    await res.close()
    if row is None:
        return False
    # print("ok")
    short_name = row["nom_court"]
    full_name = row["nom_complet"]
    perso_image = row["image"]
    # print(short_name,full_name,perso_image)
    webhook = None
    # print("ok2")
    if channel is discord.channel.DMChannel:
        await channel.send(f"**{full_name}({short_name})**: {message}")
        return
    if discord.utils.get(await channel.webhooks(),name=short_name) is None:
        # print("ok3");
        webhook = await channel.create_webhook(name=short_name)
        # print("ok4")
    else:
        # print("ok5")
        webhook = discord.utils.get(await channel.webhooks(),name=short_name)
        # print("ok6")
    # print("ok7")
    await webhook.send(message,username=full_name,avatar_url=perso_image)
    # print("ok8")
    return True

bot.run(bot_token)