mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-23 03:02:50 +01:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2016-2018 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://twitter.com/"""
|
|
|
|
from .common import Extractor, Message
|
|
from .. import text
|
|
|
|
|
|
class TwitterTweetExtractor(Extractor):
|
|
"""Extractor for images from tweets on twitter.com"""
|
|
category = "twitter"
|
|
subcategory = "tweet"
|
|
directory_fmt = ["{category}", "{user}"]
|
|
filename_fmt = "{tweet_id}_{num}.{extension}"
|
|
archive_fmt = "{tweet_id}_{num}"
|
|
pattern = [r"(?:https?://)?(?:www\.|mobile\.)?twitter\.com/"
|
|
r"(([^/]+)/status/(\d+))"]
|
|
test = [
|
|
("https://twitter.com/PicturesEarth/status/672897688871018500", {
|
|
"url": "d9e68d41301d2fe382eb27711dea28366be03b1a",
|
|
"keyword": "7a6eac2bc88bbf16d0671ebb38e31f708d940ee8",
|
|
"content": "a1f2f04cb2d8df24b1afa7a39910afda23484342",
|
|
}),
|
|
("https://twitter.com/perrypumas/status/894001459754180609", {
|
|
"url": "c8a262a9698cb733fb27870f5a8f75faf77d79f6",
|
|
"keyword": "334cd0c1f85c3e66923b44740f17407ce444931e",
|
|
}),
|
|
]
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self)
|
|
self.path, self.user, self.tid = match.groups()
|
|
|
|
def items(self):
|
|
page = self.request("https://twitter.com/" + self.path).text
|
|
data = self.get_job_metadata()
|
|
imgs = self.get_image_urls(page)
|
|
data["count"] = len(imgs)
|
|
yield Message.Version, 1
|
|
yield Message.Directory, data
|
|
for data["num"], url in enumerate(imgs, 1):
|
|
yield Message.Url, url + ":orig", text.nameext_from_url(url, data)
|
|
|
|
def get_job_metadata(self):
|
|
"""Collect metadata for extractor-job"""
|
|
return {
|
|
"user": self.user,
|
|
"tweet_id": self.tid,
|
|
}
|
|
|
|
@staticmethod
|
|
def get_image_urls(page):
|
|
"""Extract and return a list of all image-urls"""
|
|
tweet = text.extract(
|
|
page, '<div class="follow-bar">', '<ul class="stats" ')[0]
|
|
return list(text.extract_iter(
|
|
tweet, '<img data-aria-label-part src="', '"'))
|