[ao3] match URLs with 'com' and 'net' TLDs and 'ao3.org' (#6013)

mikf · Sep 17, 2024 · 8f77766 · 8f77766
1 parent 73f833d
commit 8f77766
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/gallery_dl/extractor/ao3.py b/gallery_dl/extractor/ao3.py
@@ -11,7 +11,8 @@
 from .common import Extractor, Message
 from .. import text, util
 
-BASE_PATTERN = r"(?:https?://)?(?:www\.)?archiveofourown.org"
+BASE_PATTERN = (r"(?:https?://)?(?:www\.)?"
+                r"a(?:rchiveofourown|o3)\.(?:org|com|net)")
 
 
 class Ao3Extractor(Extractor):
@@ -188,7 +189,6 @@ def items(self):
             yield Message.Queue, base + series_id, data
 
     def series(self):
-        path, user, pseud, query = self.groups
         return self._pagination(self.groups[0], '<li id="series_')
 
 

diff --git a/test/results/ao3.py b/test/results/ao3.py
@@ -139,6 +139,21 @@
         "https://archiveofourown.org/users/Fyrelass/series",
     ),
 },
+{
+    "#url"     : "https://archiveofourown.com/users/Fyrelass",
+    "#category": ("", "ao3", "user"),
+    "#class"   : ao3.Ao3UserExtractor,
+},
+{
+    "#url"     : "https://archiveofourown.net/users/Fyrelass",
+    "#category": ("", "ao3", "user"),
+    "#class"   : ao3.Ao3UserExtractor,
+},
+{
+    "#url"     : "https://ao3.org/users/Fyrelass",
+    "#category": ("", "ao3", "user"),
+    "#class"   : ao3.Ao3UserExtractor,
+},
 
 {
     "#url"     : "https://archiveofourown.org/users/Fyrelass/profile",