summaryrefslogtreecommitdiffstats
path: root/gallery_dl/extractor/kohlchan.py
blob: c96dedcffcfb485ec209840c20b4d2704887cf5f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
# -*- coding: utf-8 -*-

# 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.

"""Extractors for https://kohlchan.net/"""

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


class KohlchanThreadExtractor(Extractor):
    """Extractor for Kohlchan threads"""
    category = "kohlchan"
    subcategory = "thread"
    directory_fmt = ("{category}", "{boardUri}",
                     "{threadId} {subject|message[:50]}")
    filename_fmt = "{postId}{num:?-//} {filename}.{extension}"
    archive_fmt = "{boardUri}_{postId}_{num}"
    pattern = r"(?:https?://)?kohlchan\.net/([^/?#]+)/res/(\d+)"
    test = ("https://kohlchan.net/a/res/4594.html", {
        "pattern": r"https://kohlchan\.net/\.media/[0-9a-f]{64}(\.\w+)?$",
        "count": ">= 80",
    })

    def __init__(self, match):
        Extractor.__init__(self, match)
        self.board, self.thread = match.groups()

    def items(self):
        url = "https://kohlchan.net/{}/res/{}.json".format(
            self.board, self.thread)
        thread = self.request(url).json()
        thread["postId"] = thread["threadId"]
        posts = thread.pop("posts")

        yield Message.Directory, thread

        for post in itertools.chain((thread,), posts):
            files = post.pop("files", ())
            if files:
                thread.update(post)
                for num, file in enumerate(files):
                    file.update(thread)
                    file["num"] = num
                    url = "https://kohlchan.net" + file["path"]
                    text.nameext_from_url(file["originalName"], file)
                    yield Message.Url, url, file


class KohlchanBoardExtractor(Extractor):
    """Extractor for Kohlchan boards"""
    category = "kohlchan"
    subcategory = "board"
    pattern = (r"(?:https?://)?kohlchan\.net"
               r"/([^/?#]+)/(?:(?:catalog|\d+)\.html)?$")
    test = (
        ("https://kohlchan.net/a/", {
            "pattern": KohlchanThreadExtractor.pattern,
            "count": ">= 100",
        }),
        ("https://kohlchan.net/a/2.html"),
        ("https://kohlchan.net/a/catalog.html"),
    )

    def __init__(self, match):
        Extractor.__init__(self, match)
        self.board = match.group(1)

    def items(self):
        url = "https://kohlchan.net/{}/catalog.json".format(self.board)
        for thread in self.request(url).json():
            url = "https://kohlchan.net/{}/res/{}.html".format(
                self.board, thread["threadId"])
            thread["_extractor"] = KohlchanThreadExtractor
            yield Message.Queue, url, thread