Skip to content
This repository was archived by the owner on Mar 1, 2022. It is now read-only.

Commit 2a50b03

Browse files
committed
Small bug fixes.
Major change to cancel (No longer requires ctx).
1 parent ae28dc7 commit 2a50b03

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

discord/ext/buttons/buttons.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def _gather_buttons(self):
6868

6969
def sort_buttons(self, *, buttons: dict = None):
7070
if buttons is None:
71-
buttons = self.buttons
71+
buttons = self._buttons
7272

7373
return {k[1]: v for k, v in sorted(buttons.items(), key=lambda t: t[0])}
7474

@@ -147,12 +147,12 @@ def is_cancelled(self):
147147
"""Return True if the session has been cancelled."""
148148
return self._cancelled
149149

150-
async def cancel(self, ctx):
150+
async def cancel(self):
151151
"""Cancel the session."""
152152
self._cancelled = True
153-
await self.teardown(ctx)
153+
await self.teardown()
154154

155-
async def teardown(self, ctx):
155+
async def teardown(self):
156156
"""Clean the session up."""
157157
self._session_task.cancel()
158158

@@ -305,11 +305,9 @@ async def _paginate(self, ctx: commands.Context):
305305
async def _session(self, ctx):
306306
if self.use_defaults:
307307
if len(self._pages) == 1:
308-
self.buttons = {**self._default_stop, **self._buttons}
308+
self._buttons = {**self._default_stop, **self._buttons}
309309
else:
310-
self.buttons = {**self._defaults, **self._buttons}
311-
else:
312-
self.buttons = self._buttons
310+
self._buttons = {**self._defaults, **self._buttons}
313311

314312
self.buttons = self.sort_buttons()
315313

@@ -321,7 +319,7 @@ async def _default_indexer(self, control, ctx, member):
321319
previous = self._index
322320

323321
if control == 'stop':
324-
return await self.cancel(ctx)
322+
return await self.cancel()
325323

326324
if control == 'end':
327325
self._index = len(self._pages) - 1

0 commit comments

Comments
 (0)