89 lines
No EOL
3.2 KiB
Python
89 lines
No EOL
3.2 KiB
Python
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 type(channel) == type(discord.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) |