From 79dece31240de2be61ee1c495330ec1bdb36ff0b Mon Sep 17 00:00:00 2001 From: Arthur Milchior Date: Mon, 30 Mar 2020 09:56:46 +0200 Subject: [PATCH] split _updatePreviewButtons This will allow to create a hierarchy in this class --- qt/aqt/previewer.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/qt/aqt/previewer.py b/qt/aqt/previewer.py index 9284c9fa1..242838a7b 100644 --- a/qt/aqt/previewer.py +++ b/qt/aqt/previewer.py @@ -260,18 +260,24 @@ class BrowserPreviewer(Previewer): def _updatePreviewButtons(self): if not self._previewWindow: return + self._previewPrev.setEnabled(self._should_enable_prev()) + self._previewNext.setEnabled(self._should_enable_next()) + + def _should_enable_prev(self): current = self.parent.currentRow() canBack = current > 0 or ( current == 0 and self._previewState == "answer" and not self._previewBothSides ) - self._previewPrev.setEnabled(bool(self.parent.singleCard and canBack)) + return bool(self.parent.singleCard and canBack) + + def _should_enable_next(self): canForward = ( self.parent.currentRow() < self.parent.model.rowCount(None) - 1 or self._previewState == "question" ) - self._previewNext.setEnabled(bool(self.parent.singleCard and canForward)) + return bool(self.parent.singleCard and canForward) def _onClosePreview(self): super()._onClosePreview()