example_download.py

from swibots import (
    BotApp,
    BotCommand,
    BotContext,
    CallbackQueryEvent,
    CommandEvent,
    InlineMarkup,
    InlineKeyboardButton,
    regexp,
    CommunityUpdatedEvent,
    MessageEvent,
    Message,
    DownloadProgress,
)
import logging

logging.basicConfig(level=logging.INFO)

TOKEN = "YOUR_TOKEN_HERE"

app = BotApp(
    TOKEN, "A cool bot with annotations and everything you could possibly want :)"
).set_bot_commands(
    [
        BotCommand("echo", "Echoes the message", True),
        BotCommand("buttons", "Shows buttons", True),
        BotCommand("buttonfull", "Shows buttons", True),
        BotCommand("back", "Shows buttons", True),
        BotCommand("upload", "Reply with media", True),
    ]
)


@app.on_command("buttons")
async def buttons_handler(ctx: BotContext[CommandEvent]):
    message = f"Please select an option:"

    inline_keyboard = [
        [
            InlineKeyboardButton(text="Option 18", callback_data="option18"),
            InlineKeyboardButton(text="Option 19", callback_data="option19"),
            InlineKeyboardButton(text="Option 20", callback_data="option20"),
        ],
        [
            InlineKeyboardButton(text="Option 33", callback_data="option33"),
            InlineKeyboardButton(text="Option 43", callback_data="option43"),
            InlineKeyboardButton(text="Option 63", callback_data="option63"),
            InlineKeyboardButton(text="Option 73", callback_data="option73"),
        ],
        [
            InlineKeyboardButton(text="Optioniablek", callback_data="option53"),
            InlineKeyboardButton(text="Unitedstates", callback_data="option03"),
        ],
        [
            InlineKeyboardButton(text="Go Back on Press", callback_data="option543"),
        ],
        [
            InlineKeyboardButton(
                text="Go Back on Press for more Movies haha", callback_data="option543"
            ),
        ],
    ]

    inline_markup = InlineMarkup(
        inline_keyboard=inline_keyboard,
    )
    await ctx.event.message.reply_text(message, inline_markup=inline_markup)


@app.on_command("test")
async def test_handler(ctx: BotContext[CommandEvent]):
    await ctx.event.message.reply_text("Test command received")


@app.on_command("buttonfull")
async def buttons_handler(ctx: BotContext[CommandEvent]):
    message = f"Please select an option:"

    inline_keyboard1 = [
        [
            InlineKeyboardButton(text="Option 1111", callback_data="option1"),
            InlineKeyboardButton(text="Option 1112", callback_data="option2"),
            InlineKeyboardButton(text="Option 1115", callback_data="option5"),
        ],
        [
            InlineKeyboardButton(text="Option 18", callback_data="option18"),
            InlineKeyboardButton(text="Option 19", callback_data="option19"),
            InlineKeyboardButton(text="Option 20", callback_data="option20"),
        ],
        [
            InlineKeyboardButton(text="Optioniablek", callback_data="option53"),
            InlineKeyboardButton(text="Unitedstates", callback_data="option03"),
        ],
        [
            InlineKeyboardButton(text="Go Back on Press", callback_data="option543"),
        ],
        [
            InlineKeyboardButton(
                text="Go Back on Press for more Movies haha", callback_data="option543"
            ),
        ],
    ]

    inline_markup = InlineMarkup(
        inline_keyboard=inline_keyboard1,
    )
    await ctx.event.message.reply_text(message, inline_markup=inline_markup)


@app.on_command("echo")
async def buttons_handler(ctx: BotContext[CommandEvent]):
    message = "Please select an option:"

    inline_keyboard1 = [
        [
            InlineKeyboardButton(text="Option 1", callback_data="option1"),
            InlineKeyboardButton(text="Option 2", callback_data="option2"),
        ],
        [
            InlineKeyboardButton(text="Option 3", callback_data="option3"),
            InlineKeyboardButton(text="Option 4", callback_data="option4"),
        ],
    ]

    await ctx.event.message.reply_text(
        "Please select an option (echo):",
        inline_markup=InlineMarkup(
            inline_keyboard=inline_keyboard1,
        ),
    )


@app.on_callback_query()
async def query_callback_handler(ctx: BotContext[CallbackQueryEvent]):
    m = ctx.event.message
    await m.edit_text(f"Thank you! I received your callback: {ctx.event.callback_data}")


@app.on_message()
async def message_handler(ctx: BotContext[MessageEvent]):
    # m = await ctx.prepare_response_message(ctx.event.message)
    # m.message = f"Thank you! I received your message: {ctx.event.message.message}"
    # await ctx.send_message(m)
    print(f"Downloading received message: {ctx.event.message.id}")
    message: Message = ctx.event.message
    if message.is_media:
        print(message.media_link)
        await message.download(
            in_memory=False, block=False, progress=handle_download_progress
        )


async def handle_download_progress(progress: DownloadProgress):
    print(f"Downloaded {progress.downloaded} of {progress.total}")


@app.on_community_update()
async def community_update_handler(ctx: BotContext[CommunityUpdatedEvent]):
    print(ctx.event.community_id + " was updated")


# app.run()


async def show_upload_progress(obj):
    print(f"Uploaded {obj.current} of {obj.readed}")
    if obj == 0:
        return


@app.on_command("upload")
async def upload_handler(ctx: BotContext[CommandEvent]):
    params = ctx.event.params
    r = await ctx.event.message.reply_media(
        f"Here is your file {params}", document=params, progress=show_upload_progress
    )
    print(r)


app.run()