mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-24 03:32:33 +01:00
137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
# -*- 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
|