1
0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-22 18:53:21 +01:00

[bluesky] add 'following' extractor (#4438)

This commit is contained in:
Mike Fährmann 2024-02-08 21:58:17 +01:00
parent 86ce35d6a1
commit 9c10be54fb
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88

View File

@ -149,6 +149,17 @@ class BlueskyListExtractor(BlueskyExtractor):
return self.api.get_list_feed(self.user, self.list)
class BlueskyFollowingExtractor(BlueskyExtractor):
subcategory = "following"
pattern = USER_PATTERN + r"/follows"
example = "https://bsky.app/profile/HANDLE/follows"
def items(self):
for user in self.api.get_follows(self.user):
url = "https://bsky.app/profile/" + user["did"]
yield Message.Queue, url, user
class BlueskyPostExtractor(BlueskyExtractor):
subcategory = "post"
pattern = USER_PATTERN + r"/post/([^/?#]+)"
@ -206,6 +217,14 @@ class BlueskyAPI():
}
return self._pagination(endpoint, params)
def get_follows(self, actor):
endpoint = "app.bsky.graph.getFollows"
params = {
"actor": self._did_from_actor(actor),
"limit": "100",
}
return self._pagination(endpoint, params, "follows")
def get_list_feed(self, actor, list):
endpoint = "app.bsky.feed.getListFeed"
params = {
@ -293,10 +312,10 @@ class BlueskyAPI():
"API request failed (%s %s)",
response.status_code, response.reason)
def _pagination(self, endpoint, params):
def _pagination(self, endpoint, params, key="feed"):
while True:
data = self._call(endpoint, params)
yield from data["feed"]
yield from data[key]
cursor = data.get("cursor")
if not cursor: