nicegui/nicegui/elements/button.py

import asyncio
from typing import Awaitable, Callable, Union
import justpy as jp

from ..binding import bind_from, bind_to, BindableProperty
from ..utils import handle_exceptions, provide_arguments, async_provide_arguments
from .element import Element

class Button(Element):
    text = BindableProperty

    def __init__(self,
                 text: str = '',
                 *,
                 on_click: Union[Callable, Awaitable] = None,
                 ):
        """Button Element

        :param text: the label of the button
        :param on_click: callback which is invoked when button is pressed
        """

        view = jp.QButton(label=text, color='primary')
        super().__init__(view)

        self.text = text
        self.bind_text_to(self.view, 'label')

        if on_click is not None:
            if asyncio.iscoroutinefunction(on_click):
                view.on('click', handle_exceptions(async_provide_arguments(func=on_click, update_function=view.update)))
            else:
                view.on('click', handle_exceptions(provide_arguments(on_click)))

    def set_text(self, text: str):
        self.text = text

    def bind_text_to(self, target_object, target_name, forward=lambda x: x):
        bind_to(self, 'text', target_object, target_name, forward=forward)
        return self

    def bind_text_from(self, target_object, target_name, backward=lambda x: x):
        bind_from(self, 'text', target_object, target_name, backward=backward)
        return self

    def bind_text(self, target_object, target_name, forward=lambda x: x, backward=lambda x: x):
        bind_from(self, 'text', target_object, target_name, backward=backward)
        bind_to(self, 'text', target_object, target_name, forward=forward)
        return self