[lolisafe] implement 'domain' option (#2575)

mikf · May 10, 2022 · 6b6eb0b · 6b6eb0b
1 parent f3408a9
commit 6b6eb0b
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/docs/configuration.rst b/docs/configuration.rst
@@ -877,6 +877,21 @@ Description
     Download embedded videos hosted on https://www.blogger.com/
 
 
+extractor.cyberdrop.domain
+--------------------------
+Type
+    ``string``
+Default
+    ``"auto"``
+Example
+    ``"cyberdrop.to"``
+Description
+    Specifies the domain used by ``cyberdrop`` regardless of input URL.
+
+    Setting this option to ``"auto"``
+    uses the same domain as a given input URL.
+
+
 extractor.danbooru.external
 ---------------------------
 Type
@@ -1531,6 +1546,20 @@ Description
     the first in the list gets chosen (usually `mp3`).
 
 
+extractor.lolisafe.domain
+-------------------------
+Type
+    ``string``
+Default
+    ``"auto"``
+Description
+    Specifies the domain used by a ``lolisafe`` extractor
+    regardless of input URL.
+
+    Setting this option to ``"auto"``
+    uses the same domain as a given input URL.
+
+
 extractor.luscious.gif
 ----------------------
 Type

diff --git a/docs/gallery-dl.conf b/docs/gallery-dl.conf
@@ -50,6 +50,10 @@
         {
             "videos": true
         },
+        "cyberdrop":
+        {
+            "domain": "auto"
+        },
         "danbooru":
         {
             "username": null,

diff --git a/gallery_dl/extractor/lolisafe.py b/gallery_dl/extractor/lolisafe.py
@@ -63,6 +63,12 @@ def __init__(self, match):
         LolisafeExtractor.__init__(self, match)
         self.album_id = match.group(match.lastindex)
 
+        domain = self.config("domain")
+        if domain is None or domain == "auto":
+            self.root = text.root_from_url(match.group(0))
+        else:
+            self.root = text.ensure_http_scheme(domain)
+
     def items(self):
         files, data = self.fetch_album(self.album_id)