57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2014-2017 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.
|
|
|
|
"""Extract images from https://danbooru.donmai.us/"""
|
|
|
|
from . import booru
|
|
|
|
|
|
class DanbooruExtractor(booru.JSONBooruExtractor):
|
|
"""Base class for danbooru extractors"""
|
|
category = "danbooru"
|
|
api_url = "https://danbooru.donmai.us/posts.json"
|
|
|
|
|
|
class DanbooruTagExtractor(DanbooruExtractor, booru.BooruTagExtractor):
|
|
"""Extractor for images from danbooru based on search-tags"""
|
|
pattern = [r"(?:https?://)?danbooru\.donmai\.us/posts"
|
|
r"\?(?:utf8=%E2%9C%93&)?tags=([^&]+)"]
|
|
test = [("https://danbooru.donmai.us/posts?tags=bonocho", {
|
|
"content": "b196fb9f1668109d7774a0a82efea3ffdda07746",
|
|
})]
|
|
|
|
|
|
class DanbooruPoolExtractor(DanbooruExtractor, booru.BooruPoolExtractor):
|
|
"""Extractor for image-pools from danbooru"""
|
|
pattern = [r"(?:https?://)?danbooru\.donmai\.us/pools/(\d+)"]
|
|
test = [("https://danbooru.donmai.us/pools/7659", {
|
|
"content": "b16bab12bea5f7ea9e0a836bf8045f280e113d99",
|
|
})]
|
|
|
|
|
|
class DanbooruPostExtractor(DanbooruExtractor, booru.BooruPostExtractor):
|
|
"""Extractor for single images from danbooru"""
|
|
pattern = [r"(?:https?://)?danbooru\.donmai\.us/posts/(\d+)"]
|
|
test = [("https://danbooru.donmai.us/posts/294929", {
|
|
"content": "5e255713cbf0a8e0801dc423563c34d896bb9229",
|
|
})]
|
|
|
|
|
|
class DanbooruPopularExtractor(DanbooruExtractor, booru.BooruPopularExtractor):
|
|
"""Extractor for popular images from danbooru"""
|
|
pattern = [r"(?:https?://)?danbooru\.donmai\.us/"
|
|
r"explore/posts/popular()(?:\?([^#]*))?"]
|
|
test = [
|
|
("https://danbooru.donmai.us/explore/posts/popular", None),
|
|
(("https://danbooru.donmai.us/explore/posts/popular"
|
|
"?date=2017-07-17+14%3A13%3A05+-0400&scale=week"), {
|
|
"count": 20,
|
|
}),
|
|
]
|
|
api_url = "https://danbooru.donmai.us/explore/posts/popular.json"
|