47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from discord.ui.button import Button
|
|
|
|
from .leo import LeoUI
|
|
|
|
__all__ = (
|
|
'HookedItem',
|
|
'AButton',
|
|
'AsComponents'
|
|
)
|
|
|
|
|
|
class HookedItem:
|
|
"""
|
|
Mixin for Item classes allowing an instance to be used as a callback decorator.
|
|
"""
|
|
def __init__(self, *args, pass_kwargs={}, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.pass_kwargs = pass_kwargs
|
|
|
|
def __call__(self, coro):
|
|
async def wrapped(view, interaction, **kwargs):
|
|
return await coro(view, interaction, self, **kwargs, **self.pass_kwargs)
|
|
self.callback = wrapped
|
|
return self
|
|
|
|
|
|
class AButton(HookedItem, Button):
|
|
...
|
|
|
|
|
|
class AsComponents(LeoUI):
|
|
"""
|
|
Simple container class to accept a number of Items and turn them into an attachable View.
|
|
"""
|
|
def __init__(self, *items, pass_kwargs={}, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.pass_kwargs = pass_kwargs
|
|
|
|
for item in items:
|
|
item.callback = self.wrap_callback(item.callback)
|
|
self.add_item(item)
|
|
|
|
def wrap_callback(self, coro):
|
|
async def wrapped(*args, **kwargs):
|
|
return await coro(self, *args, **kwargs, **self.pass_kwargs)
|
|
return wrapped
|