1 files changed, 136 insertions, 0 deletions
diff --git a/gallery_dl/extractor/pillowfort.py b/gallery_dl/extractor/pillowfort.py
new file mode 100644
index 0000000..cbd65d7
--- /dev/null
+++ b/gallery_dl/extractor/pillowfort.py
@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2021 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.
+
+"""Extractors for https://www.pillowfort.social/"""
+
+from .common import Extractor, Message
+from .. import text
+
+BASE_PATTERN = r"(?:https?://)?www\.pillowfort\.social"
+
+
+class PillowfortExtractor(Extractor):
+    """Base class for pillowfort extractors"""
+    category = "pillowfort"
+    root = "https://www.pillowfort.social"
+    directory_fmt = ("{category}", "{username}")
+    filename_fmt = ("{post_id} {title|original_post[title]} "
+                    "{num:>02}.{extension}")
+    archive_fmt = "{id}"
+
+    def __init__(self, match):
+        Extractor.__init__(self, match)
+        self.item = match.group(1)
+        self.reblogs = self.config("reblogs", False)
+
+    def items(self):
+        for post in self.posts():
+
+            if "original_post" in post and not self.reblogs:
+                continue
+
+            files = post["media"]
+            del post["media"]
+
+            post["date"] = text.parse_datetime(
+                post["created_at"], "%Y-%m-%dT%H:%M:%S.%f%z")
+            yield Message.Directory, post
+
+            post["num"] = 0
+            for file in files:
+                url = file["url"]
+                if url:
+                    post.update(file)
+                    post["num"] += 1
+                    post["date"] = text.parse_datetime(
+                        file["created_at"], "%Y-%m-%dT%H:%M:%S.%f%z")
+                    yield Message.Url, url, text.nameext_from_url(url, post)
+
+
+class PillowfortPostExtractor(PillowfortExtractor):
+    """Extractor for a single pillowfort post"""
+    subcategory = "post"
+    pattern = BASE_PATTERN + r"/posts/(\d+)"
+    test = ("https://www.pillowfort.social/posts/27510", {
+        "pattern": r"https://img\d+\.pillowfort\.social/posts/\w+_out\d+\.png",
+        "count": 4,
+        "keyword": {
+            "avatar_url": str,
+            "col": 0,
+            "commentable": True,
+            "comments_count": int,
+            "community_id": None,
+            "content": str,
+            "created_at": str,
+            "date": "type:datetime",
+            "deleted": None,
+            "deleted_at": None,
+            "deleted_by_mod": None,
+            "deleted_for_flag_id": None,
+            "embed_code": None,
+            "id": int,
+            "last_activity": str,
+            "last_activity_elapsed": str,
+            "last_edited_at": None,
+            "likes_count": int,
+            "media_type": "picture",
+            "nsfw": False,
+            "num": int,
+            "original_post_id": None,
+            "original_post_user_id": None,
+            "picture_content_type": None,
+            "picture_file_name": None,
+            "picture_file_size": None,
+            "picture_updated_at": None,
+            "post_id": 27510,
+            "post_type": "picture",
+            "privacy": "public",
+            "reblog_copy_info": list,
+            "rebloggable": True,
+            "reblogged_from_post_id": None,
+            "reblogged_from_user_id": None,
+            "reblogs_count": int,
+            "row": int,
+            "small_image_url": None,
+            "tags": list,
+            "time_elapsed": str,
+            "timestamp": str,
+            "title": "What is Pillowfort.io? ",
+            "updated_at": str,
+            "url": r"re:https://img3.pillowfort.social/posts/.*\.png",
+            "user_id": 5,
+            "username": "Staff"
+        },
+    })
+
+    def posts(self):
+        url = "{}/posts/{}/json/".format(self.root, self.item)
+        return (self.request(url).json(),)
+
+
+class PillowfortUserExtractor(PillowfortExtractor):
+    """Extractor for all posts of a pillowfort user"""
+    subcategory = "user"
+    pattern = BASE_PATTERN + r"/(?!posts/)([^/?#]+)"
+    test = ("https://www.pillowfort.social/Pome", {
+        "pattern": r"https://img\d+\.pillowfort\.social/posts/",
+        "range": "1-15",
+        "count": 15,
+    })
+
+    def posts(self):
+        url = "{}/{}/json/".format(self.root, self.item)
+        params = {"p": 1}
+
+        while True:
+            posts = self.request(url, params=params).json()["posts"]
+            yield from posts
+
+            if len(posts) < 20:
+                return
+            params["p"] += 1