56 lines
1.5 KiB
Python
Executable File
56 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from dataclasses import dataclass
|
|
import disnake
|
|
from disnake.ext import commands
|
|
import logging
|
|
import yaml
|
|
|
|
|
|
@dataclass
|
|
class Config:
|
|
discord_token: str
|
|
|
|
|
|
with open('config.yml') as config_file:
|
|
config = Config(**yaml.safe_load(config_file))
|
|
|
|
|
|
logger = logging.getLogger('discord')
|
|
logger.setLevel(logging.DEBUG)
|
|
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
|
|
handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
|
|
logger.addHandler(handler)
|
|
|
|
bot = commands.Bot(command_prefix='!')
|
|
|
|
|
|
@bot.event
|
|
async def on_message(message: disnake.Message):
|
|
await bot.process_commands(message)
|
|
|
|
|
|
@bot.command()
|
|
async def vote(ctx: commands.Context):
|
|
view = Confirm()
|
|
await ctx.message.reply("plop", view=view)
|
|
await view.wait()
|
|
|
|
|
|
# Define a simple View that gives us a confirmation menu
|
|
class Confirm(disnake.ui.View):
|
|
@disnake.ui.button(label="S'allier", style=disnake.ButtonStyle.green)
|
|
async def confirm(self, button: disnake.ui.Button, interaction: disnake.MessageInteraction):
|
|
self.clear_items()
|
|
await interaction.response.edit_message(content="Vous vous êtes allié.", view=self)
|
|
self.stop()
|
|
|
|
@disnake.ui.button(label="Trahir", style=disnake.ButtonStyle.red)
|
|
async def cancel(self, button: disnake.ui.Button, interaction: disnake.MessageInteraction):
|
|
self.clear_items()
|
|
await interaction.response.edit_message("Vous avez trahi.", view=self)
|
|
self.stop()
|
|
|
|
|
|
bot.run(config.discord_token)
|