From d480e42f55ec8c505d5e2711e74cdcd6dd72e135 Mon Sep 17 00:00:00 2001
From: Damien Elmes <git@ichi2.net>
Date: Sat, 6 Dec 2008 16:13:34 +0900
Subject: [PATCH] rebuild counts after undo, rebuild counts more efficiently

---
 anki/deck.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/anki/deck.py b/anki/deck.py
index 27406738c..8c335f1ab 100644
--- a/anki/deck.py
+++ b/anki/deck.py
@@ -476,8 +476,12 @@ end)""" + where)
         self.failedNowCount = self.s.scalar("""
 select count(*) from cards where type = 0 and isDue = 1
 and combinedDue <= :t""", t=time.time())
-        self.revCount = self.s.scalar("select count(*) from revCardsOld")
-        self.newCount = self.s.scalar("select count(*) from acqCardsOrdered")
+        self.revCount = self.s.scalar(
+            "select count(*) from cards where "
+            "type = 1 and priority in (1,2,3,4)")
+        self.newCount = self.s.scalar(
+            "select count(*) from cards where "
+            "type = 2 and priority in (1,2,3,4)")
 
     def checkDue(self):
         "Mark expired cards due, and update counts."
@@ -1682,10 +1686,12 @@ seq > :s and seq <= :e order by seq desc""", s=start, e=end)
     def undo(self):
         self._undoredo(self.undoStack, self.redoStack)
         self.refresh()
+        self.rebuildCounts()
 
     def redo(self):
         self._undoredo(self.redoStack, self.undoStack)
         self.refresh()
+        self.rebuildCounts()
 
 # Shared decks
 ##########################################################################