Check future state before setting result in View

This commit is contained in:
Rapptz 2021-05-29 05:52:05 -04:00
parent d78e5d979d
commit c21d12be5e

View File

@ -250,6 +250,7 @@ class View:
self._timeout_handler = loop.call_later(self.timeout, self.dispatch_timeout)
def dispatch_timeout(self):
if not self._stopped.done():
self._stopped.set_result(True)
asyncio.create_task(self.on_timeout(), name=f'discord-ui-view-timeout-{self.id}')
@ -282,7 +283,9 @@ class View:
This operation cannot be undone.
"""
if not self._stopped.done():
self._stopped.set_result(False)
if self._timeout_handler:
self._timeout_handler.cancel()