From 73e5a227a54a05cb1d83ce19148cd195e145717f Mon Sep 17 00:00:00 2001 From: Dustin Pianalto Date: Fri, 8 Jun 2018 11:42:15 -0800 Subject: [PATCH] Make message optional in book creation --- src/imports/utils.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/imports/utils.py b/src/imports/utils.py index b2c6117..e3b281f 100644 --- a/src/imports/utils.py +++ b/src/imports/utils.py @@ -356,9 +356,15 @@ class Book: async def display_page(self) -> None: if isinstance(self._pages[self._current_page], discord.Embed): - await self._message.edit(content=None, embed=self._pages[self._current_page]) + if self._message: + await self._message.edit(content=None, embed=self._pages[self._current_page]) + else: + await self._channel.send(embed=self._pages[self._current_page]) else: - await self._message.edit(content=self._pages[self._current_page], embed=None) + if self._message: + await self._message.edit(content=self._pages[self._current_page], embed=None) + else: + await self._channel.send(self._pages[self._current_page]) async def create_book(self) -> None: # noinspection PyUnresolvedReferences