matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Add a way to wait for a view to finish its interactions

Browse Source
This commit is contained in:
Rapptz 2021-05-27 23:31:48 -04:00
parent 77ed476129
commit 65db814d4a
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
discord/ui/view.py
View File

@ -118,6 +118,7 @@ class View:
self.id = os.urandom(16).hex() self.id = os.urandom(16).hex()
self._cancel_callback: Optional[Callable[[View], None]] = None self._cancel_callback: Optional[Callable[[View], None]] = None
self._stopped = asyncio.Event()
def to_components(self) -> List[Dict[str, Any]]: def to_components(self) -> List[Dict[str, Any]]:
def key(item: Item) -> int: def key(item: Item) -> int:
@ -212,9 +213,17 @@ class View:
This operation cannot be undone. This operation cannot be undone.
""" """
self._stopped.set()
if self._cancel_callback: if self._cancel_callback:
self._cancel_callback(self) self._cancel_callback(self)
async def wait(self) -> None:
"""Waits until the view has finished interacting.
A view is considered finished when :meth:`stop` is called.
"""
await self._stopped.wait()
class ViewStore: class ViewStore:
def __init__(self, state): def __init__(self, state):