[twitter] changes and improvements

- rename User- to TimelineExtractor
- rename 'userid' to 'user_id' to conform to the other ..._id values
- adjust archive_fmt to deal with retweets
- emulate browser behavior for API calls
This commit is contained in:
Mike Fährmann
2018-08-18 18:58:10 +02:00
parent 10365394d7
commit 9b1c39032c
2 changed files with 32 additions and 21 deletions

View File

@@ -17,25 +17,28 @@ class TwitterExtractor(Extractor):
category = "twitter"
directory_fmt = ["{category}", "{user}"]
filename_fmt = "{tweet_id}_{num}.{extension}"
archive_fmt = "{tweet_id}_{num}"
archive_fmt = "{tweet_id}_{retweet_id}_{num}"
root = "https://twitter.com"
def __init__(self):
Extractor.__init__(self)
self.user = None
self.retweets = self.config("retweets", True)
self.retweets = True
def items(self):
yield Message.Version, 1
yield Message.Directory, self.metadata()
for tweet in self.tweets():
images = list(text.extract_iter(
tweet, 'data-image-url="', '"'))
if not images:
continue
data = self._data_from_tweet(tweet)
if not self.retweets and data["retweet_id"]:
continue
images = text.extract_iter(
tweet, 'data-image-url="', '"')
for data["num"], url in enumerate(images, 1):
text.nameext_from_url(url, data)
yield Message.Url, url + ":orig", data
@@ -54,29 +57,29 @@ class TwitterExtractor(Extractor):
("retweeter" , 'data-retweeter="' , '"'),
("user" , 'data-screen-name="', '"'),
("username" , 'data-name="' , '"'),
("userid" , 'data-user-id="' , '"'),
("user_id" , 'data-user-id="' , '"'),
))[0]
for key in ("tweet_id", "retweet_id", "userid"):
for key in ("tweet_id", "retweet_id", "user_id"):
data[key] = text.parse_int(data[key])
data["retweeter"] = data["retweeter"] or ""
return data
class TwitterUserExtractor(TwitterExtractor):
"""Extractor for all tweeted images of a user"""
subcategory = "user"
archive_fmt = "{tweet_id}_{num}"
class TwitterTimelineExtractor(TwitterExtractor):
"""Extractor for all tweeted images from a user's timeline"""
subcategory = "timeline"
pattern = [r"(?:https?://)?(?:www\.|mobile\.)?twitter\.com"
r"/([^/?&#]+)/?$"]
test = [("https://twitter.com/PicturesEarth", {
"range": (1, 40),
"url": "2f4d51cbba81e56c1c755677b3ad58fc167c9771",
"keyword": "611066e488c233e0b1bd2ab45d5f7fca1335f691",
"keyword": "cbae53b6f4ba133078bb13c95dbd3cbb4fa40b9f",
})]
def __init__(self, match):
TwitterExtractor.__init__(self)
self.user = match.group(1)
self.retweets = self.config("retweets", True)
def metadata(self):
return {"user": self.user}
@@ -84,37 +87,45 @@ class TwitterUserExtractor(TwitterExtractor):
def tweets(self):
url = "{}/i/profiles/show/{}/timeline/tweets".format(
self.root, self.user)
params = {}
tweet = None
params = {
"include_available_features": "1",
"include_entities": "1",
"reset_error_state": "false",
"lang": "en",
}
headers = {
"X-Requested-With": "XMLHttpRequest",
"X-Twitter-Active-User": "yes",
"Referer": "{}/{}".format(self.root, self.user)
}
while True:
data = self.request(url, params=params).json()
html = data["items_html"]
data = self.request(url, params=params, headers=headers).json()
for tweet in text.extract_iter(
html, '<div class="tweet ', '\n</li>'):
data["items_html"], '<div class="tweet ', '\n</li>'):
yield tweet
if not tweet or not data["has_more_items"]:
if not data["has_more_items"]:
return
params["max_position"] = text.extract(
tweet, 'data-tweet-id="', '"')[0]
class TwitterTweetExtractor(TwitterExtractor):
"""Extractor for images from tweets on twitter.com"""
"""Extractor for images from individual tweets"""
subcategory = "tweet"
pattern = [r"(?:https?://)?(?:www\.|mobile\.)?twitter\.com"
r"/([^/?&#]+)/status/(\d+)"]
test = [
("https://twitter.com/PicturesEarth/status/672897688871018500", {
"url": "d9e68d41301d2fe382eb27711dea28366be03b1a",
"keyword": "6ea5cdc97e4e8b2b133f7dfb2048b85a608ef041",
"keyword": "46c8e739a892000848a8a2184da91346c9cbe4bf",
"content": "a1f2f04cb2d8df24b1afa7a39910afda23484342",
}),
("https://twitter.com/perrypumas/status/894001459754180609", {
"url": "c8a262a9698cb733fb27870f5a8f75faf77d79f6",
"keyword": "9d0960bc4e1b7407319f8826a43bccf2df60abcf",
"keyword": "7729cd3ff16a5647b0b5ffdec9d428c91eedafbe",
}),
]