From ab83f968b9da06a833790e616de29bf3f3334d50 Mon Sep 17 00:00:00 2001 From: "Dusty.P" Date: Wed, 6 Jun 2018 23:32:58 -0800 Subject: [PATCH] Added option to select page number --- src/imports/utils.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/src/imports/utils.py b/src/imports/utils.py index 850ac2d..2f41fce 100644 --- a/src/imports/utils.py +++ b/src/imports/utils.py @@ -289,7 +289,22 @@ class Book: elif str(reaction.emoji) == self._bot.book_emojis['start']: self._current_page = 0 elif str(reaction.emoji) == self._bot.book_emojis['hash']: - raise NotImplementedError + m = await self._channel.send(f'Please enter a number between 1 and {self._len_pages}') + def num_check(message): + if self._locked: + return message.content.isdigit() \ + and 0 < int(message.content) < self._len_pages \ + and message.author == self._calling_message.author + else: + return message.content.isdigit() \ + and 0 < int(message.content) < self._len_pages + + try: + msg = self._bot.wait_for('message', timeout=30, check=num_check()) + except asyncio.TimeoutError: + await m.edit(content='Message Timed out.') + else: + self._current_page = int(msg.content) elif str(reaction.emoji) == self._bot.book_emojis['unlock']: self._locked = False await self._message.remove_reaction(reaction, self._channel.guild.me)