1
0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-25 12:12:34 +01:00
gallery-dl/gallery_dl/extractor/recursive.py
2023-09-13 21:47:20 +02:00

32 lines
897 B
Python

# -*- coding: utf-8 -*-
# Copyright 2015-2023 Mike Fährmann
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
"""Recursive extractor"""
from .common import Extractor, Message
import re
class RecursiveExtractor(Extractor):
"""Extractor that fetches URLs from a remote or local source"""
category = "recursive"
pattern = r"r(?:ecursive)?:"
example = "recursive:https://pastebin.com/raw/FLwrCYsT"
def items(self):
url = self.url.partition(":")[2]
if url.startswith("file://"):
with open(url[7:]) as fp:
page = fp.read()
else:
page = self.request(url).text
for match in re.finditer(r"https?://[^\s\"']+", page):
yield Message.Queue, match.group(0), {}