Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(interactions): Support callback extra check.

Browse Source
This commit is contained in:
Conatum
2022-04-26 13:44:20 +03:00
parent 82e1141b2f
commit f5909e8414
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/meta/interactions/components.py
View File

@@ -48,12 +48,12 @@ class AwaitableComponent:
return await client.wait_for('interaction_create', timeout=timeout, check=_check)
def add_callback(self, timeout=None, repeat=True, pass_args=(), pass_kwargs={}):
def add_callback(self, timeout=None, repeat=True, check=None, pass_args=(), pass_kwargs={}):
def wrapper(func):
async def wrapped():
while True:
try:
button_press = await self.wait_for(timeout=timeout)
button_press = await self.wait_for(timeout=timeout, check=check)
except asyncio.TimeoutError:
break
asyncio.create_task(func(button_press, *pass_args, **pass_kwargs))