From d52b2d480160e324795b8c690a35a9fd916e2b25 Mon Sep 17 00:00:00 2001
From: Damien Elmes <git@ichi2.net>
Date: Mon, 5 Dec 2011 19:34:25 +0900
Subject: [PATCH] if counts() is passed a card, add the necessary count

---
 anki/sched.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/anki/sched.py b/anki/sched.py
index 8a8490f90..f9aef0d25 100644
--- a/anki/sched.py
+++ b/anki/sched.py
@@ -66,8 +66,15 @@ class Scheduler(object):
         card.usn = self.col.usn()
         card.flushSched()
 
-    def counts(self):
-        return (self.newCount, self.lrnCount, self.revCount)
+    def counts(self, card=None):
+        counts = [self.newCount, self.lrnCount, self.revCount]
+        if card:
+            idx = self.countIdx(card)
+            if idx == 1:
+                counts[1] += card.left
+            else:
+                counts[idx] += 1
+        return tuple(counts)
 
     def dueForecast(self, days=7):
         "Return counts over next DAYS. Includes today."