some help with mixin

[jamesalphanome]

I find mixin very helpful, but I want a specific case to work.

I have the following code:

class A(rx.State, mixin=True):

    @rx.event
    def handle_processing(self):
        ...

class B(A, rx.state):

    @rx.event
    def process(self):
        yield B.handle_processing

    @rx.event
    def on_success(self):
        ...

I want to yield the on_success event handler after handle processing is successfully completed.

How can I achieve this ?

[masenf]

have you tried yield type(self).on_success?

that won't work in a background task due to #4947, but in that case you can use self.__class__.on_success

[jamesalphanome]

This works

class A(rx.State, mixin=True):
    """mixin"""

    @rx.event(background=True)
    async def on_success(self):
        raise NotImplementedError("Implement this eventhandler - mixin")

    @rx.event(background=True)
    async def process(self):
        await asyncio.sleep(1)

        yield self.__class__.on_success


class B(A, rx.State):
    """The app state."""

    @rx.event
    async def completed_toast(self):
        return rx.toast.success("Done.")

    @rx.event(background=True)
    async def on_success(self):
        yield B.completed_toast


def index() -> rx.Component:
    return rx.container(
        rx.color_mode.button(position="top-right"),
        rx.button(
            "Click me",
            on_click=B.process,
        ),
    )

[jamesalphanome]

Thanks, @masenf