Pass timeout to SimpleMenu (#6173)

This commit is contained in:
Om 2023-06-21 02:48:58 +05:30 committed by GitHub
parent b1c3b37235
commit 41204ccf77
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -110,6 +110,7 @@ async def menu(
view._source = _SimplePageSource(pages) view._source = _SimplePageSource(pages)
new_page = await view.get_page(page) new_page = await view.get_page(page)
view.current_page = page view.current_page = page
view.timeout = timeout
await view.message.edit(**new_page) await view.message.edit(**new_page)
return return
if not isinstance(pages[0], (discord.Embed, str)): if not isinstance(pages[0], (discord.Embed, str)):
@ -137,12 +138,12 @@ async def menu(
# This is not normally the way we recommend sending this because # This is not normally the way we recommend sending this because
# internally we already include the emojis we expect. # internally we already include the emojis we expect.
if controls == DEFAULT_CONTROLS: if controls == DEFAULT_CONTROLS:
view = SimpleMenu(pages) view = SimpleMenu(pages, timeout=timeout)
await view.start(ctx) await view.start(ctx)
await view.wait() await view.wait()
return return
else: else:
view = SimpleMenu(pages) view = SimpleMenu(pages, timeout=timeout)
view.remove_item(view.last_button) view.remove_item(view.last_button)
view.remove_item(view.first_button) view.remove_item(view.first_button)
has_next = False has_next = False