sir-rollington/bot.py

import discord
from discord.app_commands import CommandTree
from os import environ

from dice import Dice, DiceTextError


intents = discord.Intents.default()
intents.message_content = True

bot = discord.Client(intents=intents)
tree = CommandTree(bot)


@bot.event
async def on_ready():
    await tree.sync()  # Bad idea


@tree.command()
@discord.app_commands.describe(
    dice="Dice to be rolled in format [number]d[sides]",
    private="Roll privatly or publicly",
)
async def roll(interaction: discord.Interaction, dice: str, private: bool):
    dice_objs = [Dice(d) for d in dice.split(" ")]

    try:
        text = []
        for die in dice_objs:
            roll = die.roll()
            text.append(f"{die.text} => {roll} = {sum(roll)}")
            print(text)
    except DiceTextError as e:
        print(e)
        await interaction.response.send_message(str(e))
        return

    if not private:
        intro = interaction.user.display_name + " rolled:\n"
    else:
        intro = "You rolled:\n"

    await interaction.response.send_message(intro + "\n".join(text), ephemeral=private)


token = environ.get("DISCORD_TOKEN")
if token is None:
    print("Could not load token")
    quit()

bot.run(token)