From 8bf081d2956a60747b84cce44e275beb849f4fe2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= <mike_faehrmann@web.de>
Date: Sat, 21 Sep 2024 13:54:42 +0200
Subject: [PATCH] [ao3] fix bookmark extraction, support 'series' bookmarks
 (#6013)

---
 gallery_dl/extractor/ao3.py | 18 ++++++++++++++++++
 test/results/ao3.py         |  2 +-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/gallery_dl/extractor/ao3.py b/gallery_dl/extractor/ao3.py
index c0eea37d..1f570e83 100644
--- a/gallery_dl/extractor/ao3.py
+++ b/gallery_dl/extractor/ao3.py
@@ -282,3 +282,21 @@ class Ao3UserBookmarkExtractor(Ao3Extractor):
     pattern = (BASE_PATTERN + r"(/users/([^/?#]+)/(?:pseuds/([^/?#]+)/)?"
                r"bookmarks(?:/?\?.+)?)")
     example = "https://archiveofourown.org/users/USER/bookmarks"
+
+    def items(self):
+        self.login()
+
+        base = self.root + "/"
+        data_work = {"_extractor": Ao3WorkExtractor}
+        data_series = {"_extractor": Ao3SeriesExtractor}
+
+        for item in self._pagination(
+                self.groups[0], '<span class="count"><a href="/'):
+            path = item.rpartition("/")[0]
+            url = base + path
+            if item.startswith("works/"):
+                yield Message.Queue, url, data_work
+            elif item.startswith("series/"):
+                yield Message.Queue, url, data_series
+            else:
+                self.log.warning("Unsupported bookmark type '%s'", path)
diff --git a/test/results/ao3.py b/test/results/ao3.py
index 4fa1aa8f..d539bda4 100644
--- a/test/results/ao3.py
+++ b/test/results/ao3.py
@@ -236,7 +236,7 @@ __tests__ = (
     "#url"     : "https://archiveofourown.org/users/Fyrelass/bookmarks",
     "#category": ("", "ao3", "user-bookmark"),
     "#class"   : ao3.Ao3UserBookmarkExtractor,
-    "#pattern" : ao3.Ao3WorkExtractor.pattern,
+    "#pattern" : r"https://archiveofourown\.org/(work|serie)s/\d+",
     "#range"   : "1-50",
     "#count"   : 50,
 },