cleanup when leaving review state

This commit is contained in:
Damien Elmes 2011-04-17 00:09:03 +09:00
parent d012bdc1a1
commit f55c7492fd
2 changed files with 12 additions and 1 deletions

View file

@ -92,7 +92,10 @@ class AnkiQt(QMainWindow):
def moveToState(self, state, *args): def moveToState(self, state, *args):
print "-> move from", self.state, "to", state print "-> move from", self.state, "to", state
oldState = self.state oldState = self.state or "dummy"
cleanup = getattr(self, "_"+oldState+"Cleanup", None)
if cleanup:
cleanup(state)
self.state = state self.state = state
getattr(self, "_"+state+"State")(oldState, *args) getattr(self, "_"+state+"State")(oldState, *args)
@ -117,6 +120,10 @@ class AnkiQt(QMainWindow):
def _reviewState(self, oldState): def _reviewState(self, oldState):
self.reviewer.show() self.reviewer.show()
def _reviewCleanup(self, newState):
if newState != "resetRequired":
self.reviewer.cleanup()
def _editCurrentState(self, oldState): def _editCurrentState(self, oldState):
pass pass

View file

@ -40,6 +40,10 @@ class Reviewer(object):
if not self.card or self._answeredIds[-1] != self.card.id: if not self.card or self._answeredIds[-1] != self.card.id:
return self.mw.deck.getCard(self._answeredIds[-1]) return self.mw.deck.getCard(self._answeredIds[-1])
def cleanup(self):
self._hideStatus()
self.mw.disableCardMenuItems()
# Fetching a card # Fetching a card
########################################################################## ##########################################################################