From c0c922d27cf5477d80d5439f0d540ec0417f16cc Mon Sep 17 00:00:00 2001
From: Austin Siew <17107540+Aquafina-water-bottle@users.noreply.github.com>
Date: Tue, 13 Sep 2022 17:12:20 -0600
Subject: [PATCH] changed getReviewsOfCard -> getReviewsOfCards, changed result
 to be formatted in a dictionary rather than a list

---
 README.md           | 27 ++++++++++++++++-----------
 plugin/__init__.py  | 13 +++++++++----
 tests/test_stats.py |  4 ++--
 3 files changed, 27 insertions(+), 17 deletions(-)

diff --git a/README.md b/README.md
index 5caa86d..87e631d 100644
--- a/README.md
+++ b/README.md
@@ -2759,18 +2759,21 @@ corresponding to when the API was available for use.
     }
     ```
 
-*   **getReviewsOfCard**
+*   **getReviewsOfCards**
 
-    Requests all card reviews for a specific card ID.
-    Returns a list of 9-tuples in the same format as `cardReviews`: `(reviewTime, cardID, usn, buttonPressed, newInterval, previousInterval, newFactor, reviewDuration, reviewType)`
+    Requests all card reviews for each card ID.
+    Returns a dictionary mapping the card ID to 9-tuples in the same format as `cardReviews`:
+    `(reviewTime, cardID, usn, buttonPressed, newInterval, previousInterval, newFactor, reviewDuration, reviewType)`
 
     *Sample request*:
     ```json
     {
-        "action": "getReviewsOfCard",
+        "action": "getReviewsOfCards",
         "version": 6,
         "params": {
-            "card": "1653613948202"
+            "cards": [
+                "1653613948202"
+            ]
         }
     }
     ```
@@ -2778,12 +2781,14 @@ corresponding to when the API was available for use.
     *Sample result*:
     ```json
     {
-        "result": [
-            [1654102387663, 1653613948202, 1780, 3, 8, 3, 2500, 25796, 1],
-            [1654798974478, 1653613948202, 1861, 3, 20, 8, 2500, 18134, 1],
-            [1656556319328, 1653613948202, 2075, 3, 53, 20, 2500, 20530, 1],
-            [1661107990069, 1653613948202, 2478, 3, 131, 53, 2500, 24247, 1]
-        ],
+        "result": {
+            "1653613948202": [
+                [1654102387663, 1653613948202, 1780, 3,   8,  3, 2500, 25796, 1],
+                [1654798974478, 1653613948202, 1861, 3,  20,  8, 2500, 18134, 1],
+                [1656556319328, 1653613948202, 2075, 3,  53, 20, 2500, 20530, 1],
+                [1661107990069, 1653613948202, 2478, 3, 131, 53, 2500, 24247, 1]
+            ]
+        },
         "error": null
     }
     ```
diff --git a/plugin/__init__.py b/plugin/__init__.py
index 661935e..edadf4b 100644
--- a/plugin/__init__.py
+++ b/plugin/__init__.py
@@ -1374,10 +1374,15 @@ class AnkiConnect:
 
 
     @util.api()
-    def getReviewsOfCard(self, card):
-        return self.database().all(
-            'select id, cid, usn, ease, ivl, lastIvl, factor, time, type from revlog where cid = ?', card
-        )
+    def getReviewsOfCards(self, cards):
+        return {
+            card: self.database().all(
+                "select id, cid, usn, ease, ivl, lastIvl, factor, time, type from revlog where cid = ?",
+                card,
+            )
+            for card in cards
+        }
+
 
 
     @util.api()
diff --git a/tests/test_stats.py b/tests/test_stats.py
index 3237161..d3e94fd 100755
--- a/tests/test_stats.py
+++ b/tests/test_stats.py
@@ -29,5 +29,5 @@ class TestReviews:
 
         assert len(ac.cardReviews(deck="test_deck", startID=0)) == 2
         assert ac.getLatestReviewID(deck="test_deck") == 789
-        assert ac.getReviewsOfCard(card=setup.card_ids[0]) == \
-                [[456, setup.card_ids[0], -1, 3, 4, -60, 2500, 6157, 0]]
+        assert ac.getReviewsOfCards(cards=[setup.card_ids[0]]) == \
+                {setup.card_ids[0]: [[456, setup.card_ids[0], -1, 3, 4, -60, 2500, 6157, 0]]}