From 1418c0ce38871db5c29de99926812c3132646c81 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= <mike_faehrmann@web.de>
Date: Fri, 15 Mar 2024 22:27:13 +0100
Subject: [PATCH] [kemonoparty] add 'revision_count' metadata field (#5334)

---
 gallery_dl/extractor/kemonoparty.py | 7 +++++--
 test/results/kemonoparty.py         | 3 +++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/gallery_dl/extractor/kemonoparty.py b/gallery_dl/extractor/kemonoparty.py
index 06dc861e..13897b49 100644
--- a/gallery_dl/extractor/kemonoparty.py
+++ b/gallery_dl/extractor/kemonoparty.py
@@ -232,6 +232,7 @@ class KemonopartyExtractor(Extractor):
         except exception.HttpError:
             post["revision_hash"] = self._revision_hash(post)
             post["revision_index"] = 1
+            post["revision_count"] = 1
             return (post,)
         revs.insert(0, post)
 
@@ -247,9 +248,10 @@ class KemonopartyExtractor(Extractor):
                     uniq.append(rev)
             revs = uniq
 
-        idx = len(revs)
+        cnt = idx = len(revs)
         for rev in revs:
             rev["revision_index"] = idx
+            rev["revision_count"] = cnt
             idx -= 1
 
         return revs
@@ -257,10 +259,11 @@ class KemonopartyExtractor(Extractor):
     def _revisions_all(self, url):
         revs = self.request(url + "/revisions").json()
 
-        idx = len(revs)
+        cnt = idx = len(revs)
         for rev in revs:
             rev["revision_hash"] = self._revision_hash(rev)
             rev["revision_index"] = idx
+            rev["revision_count"] = cnt
             idx -= 1
 
         return revs
diff --git a/test/results/kemonoparty.py b/test/results/kemonoparty.py
index 16e1b78f..1528f55f 100644
--- a/test/results/kemonoparty.py
+++ b/test/results/kemonoparty.py
@@ -195,6 +195,7 @@ __tests__ = (
     "hash"          : "88521f71822dfa2f42df3beba319ea4fceda2a2d6dc59da0276a75238f743f86",
     "revision_id"   : 142470,
     "revision_index": 2,
+    "revision_count": 9,
     "revision_hash" : "e0e93281495e151b11636c156e52bfe9234c2a40",
 },
 
@@ -210,6 +211,7 @@ __tests__ = (
     "hash"          : "88521f71822dfa2f42df3beba319ea4fceda2a2d6dc59da0276a75238f743f86",
     "revision_id"   : 0,
     "revision_index": 1,
+    "revision_count": 1,
     "revision_hash" : "e0e93281495e151b11636c156e52bfe9234c2a40",
 },
 
@@ -224,6 +226,7 @@ __tests__ = (
 
     "revision_id": range(134996, 3052965),
     "revision_index": range(1, 9),
+    "revision_count": 9,
     "revision_hash": "e0e93281495e151b11636c156e52bfe9234c2a40",
 },