telegram-botup 1.0.2

Telegram bot API

Telegram-botup

Install

$ pip install telegram-botup

Example

# app.py

from web_lib import App, Request
from botup.core.dispatcher import Dispatcher
from botup.core.api import Api
from botup.core.types import CoreContext

TOKEN = "token"

app = App()
dispatcher = Dispatcher()
api = Api(TOKEN)


@dispatcher.message_handler('hello')
async def hello_handler(ctx: CoreContext):
    await api.send_message(ctx.chat_id, f'Hello {ctx.update.message.from_.first_name}')


@app.post(f'/{TOKEN}')
async def index(request: Request):
    await dispatcher.handle(await request.json())
    return "!"

Widgets example

# app.py

from asyncio import gather

from web_lib import App, Request

from botup.widget import Widget, Context
from botup.core.dispatcher import Dispatcher
from botup.core.api import Api
from botup.bot import Bot
from botup.navigation import Navigation
from botup.widgets.date_picker import DatePicker
from botup.mixins.echo import EchoMixin
from botup.core.types import InlineKeyboardMarkup, InlineKeyboardButton

TOKEN = ""
WEBHOOK = f'https:///{TOKEN}'

app = App(docs_url=None, redoc_url=None)


class MyCustomMixin:

    def build(self, dispatcher: Dispatcher):
        dispatcher.register_command_handler('/test', self.cmd_test)

    @staticmethod
    async def cmd_test(ctx: Context):
        await ctx.api.send_message(
            chat_id=ctx.chat_id,
            text='Test'
        )


class RootWidget(Widget, MyCustomMixin, EchoMixin):
    """
    Type "/start"
    """

    KEY = 'root'
    DATE_PICKER_KEY = 'date_picker'
    DATE_PICKER_RESULT_KEY = 'dp_result'

    async def entry(self, ctx: Context, **kwargs):
        botup_date_picker_result = kwargs.get(RootWidget.DATE_PICKER_RESULT_KEY)
        if botup_date_picker_result:
            await ctx.api.send_message(
                chat_id=ctx.chat_id,
                text=f'date: {botup_date_picker_result}'
            )
            return

    def build(self, dispatcher: Dispatcher):
        MyCustomMixin.build(self, dispatcher)
        dispatcher.register_message_handler('go', self.go_handler)
        dispatcher.register_command_handler('/start', self.cmd_start)
        dispatcher.register_callback_handler('ready', self.clb_ready)
        EchoMixin.build(self, dispatcher)

    @staticmethod
    async def cmd_start(ctx: Context):
        message = await ctx.api.send_message(
            chat_id=ctx.chat_id,
            text='Are you ready?',
            reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(callback_data='ready', text='Yeah!')]])
        )
        await ctx.state_manager.set(
            chat_id=ctx.chat_id,
            key='message_id',
            value=str(message.message_id)
        )

    @staticmethod
    async def clb_ready(ctx: Context):
        _, message_id, nav = await gather(
            ctx.api.answer_callback_query(ctx.update.callback_query.id),
            ctx.state_manager.get(
                chat_id=ctx.chat_id,
                key='message_id'
            ),
            Navigation.of(ctx)
        )

        await nav.push(RootWidget.DATE_PICKER_KEY, message_id=int(message_id))

    @staticmethod
    async def go_handler(ctx: Context):
        nav = await Navigation.of(ctx)
        await nav.push(RootWidget.DATE_PICKER_KEY)


root_widget = RootWidget(
    key=RootWidget.KEY,
    children=[
        DatePicker(
            key=RootWidget.DATE_PICKER_KEY,
            result_key=RootWidget.DATE_PICKER_RESULT_KEY
        )
    ]
)

bot = Bot(
    token=TOKEN,
    root_widget=root_widget
)


@app.on_event("startup")
async def startup_event():
    async with Api(TOKEN) as api:
        await api.set_webhook(WEBHOOK)


@app.on_event("shutdown")
async def shutdown_event():
    async with Api(TOKEN) as api:
        await api.delete_webhook()

    await bot.close_session()


@app.post(f'/{TOKEN}')
async def index(request: Request):
    await bot.handle(await request.json())
    return {'ok': True}