bot.py

"""Entry points for Discord bot commands."""

import os
import discord
from discord.ext import commands

from api import download_image
from otr import read_image_text
from schedule import Schedule
from ui import ScheduleView

token = os.getenv('DISCORD_TOKEN')
intents = discord.Intents.default()
intents.message_content = True
intents.guild_scheduled_events = True

bot = commands.Bot(command_prefix='!', intents=intents)
bot.last_texts = []

@bot.command()
async def echo(context: commands.Context, *, message:str):
    """Echoes back argument to Discord message"""
    await context.send(message)

@bot.command()
async def debug(context: commands.Context):
    """Prints all the last text read from the last images."""
    await context.send('Last images processed:')
    for text in bot.last_texts:
        await context.send(text)

@bot.command()
async def read_schedule(context: commands.Context):
    """Reads image attachments into Schedules."""
    bot.last_texts = []
    image_attachments = context.message.attachments
    count = len(image_attachments)
    if count == 0:
        await context.send('No images found 😔')
        return

    await context.send(f'Processing {count} image{"" if count == 1 else "s"}...')
    for image_attachment in image_attachments:
        image = await download_image(image_attachment.url)
        text = read_image_text(image)
        bot.last_texts.append(text)
        schedule = Schedule(text)
        if schedule.error is not None:
            await context.send(schedule.error)
            return # Nothing to do if the schedule didn't parse correctly.

        if context.guild is None:
            await context.send(str(schedule))
            return # Don't offer to create events outside of a guild.

        await context.send(str(schedule), view=ScheduleView(schedule))

bot.run(token)