[kemonoparty] add 'max-posts' option (#1674)

2021-07-09 18:19:02 +02:00
parent 5612ca31c2
commit 8b036778e3
2 changed files with 17 additions and 1 deletions
--- a/gallery_dl/extractor/kemonoparty.py
+++ b/gallery_dl/extractor/kemonoparty.py
@@ -10,6 +10,7 @@

 from .common import Extractor, Message
 from .. import text
+import itertools
 import re

 BASE_PATTERN = r"(?:https?://)?kemono\.party/([^/?#]+)/user/([^/?#]+)"
@@ -40,7 +41,12 @@ class KemonopartyExtractor(Extractor):
        else:
            username = None

-        for post in self.posts():
+        posts = self.posts()
+        max_posts = self.config("max-posts")
+        if max_posts:
+            posts = itertools.islice(posts, max_posts)
+
+        for post in posts:

            files = []
            append = files.append