aniani/toot
0
0
Fork 0
mirror of https://github.com/ihabunek/toot.git synced 2025-06-30 22:18:36 -04:00
Code

Add http.get_paged for paging lists

Browse Source
This commit is contained in:
Ivan Habunek 2025-01-12 14:13:08 +01:00
parent 38dfd747f6
commit d6096df299
No known key found for this signature in database
GPG Key ID: 01DB3DD0D824504C
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
toot/http.py
View File

@ -1,8 +1,10 @@
from urllib.parse import urlencode, urlparse
from requests import Request, Session from requests import Request, Session
from requests.exceptions import RequestException from requests.exceptions import RequestException
from toot import __version__ from toot import __version__
from toot.exceptions import NotFoundError, ApiError from toot.exceptions import ApiError, NotFoundError
from toot.logging import log_request, log_request_exception, log_response from toot.logging import log_request, log_request_exception, log_response
@ -65,6 +67,23 @@ def get(app, user, path, params=None, headers=None):
return process_response(response) return process_response(response)
def get_paged(app, user, path, params=None, headers=None):
if params:
path += f"?{urlencode(params)}"
while path:
response = get(app, user, path, params, headers)
yield response
path = _next_path(response)
def _next_path(response):
next_link = response.links.get("next")
if next_link:
next_url = urlparse(next_link["url"])
return "?".join([next_url.path, next_url.query])
def anon_get(url, params=None): def anon_get(url, params=None):
request = Request('GET', url, None, params=params) request = Request('GET', url, None, params=params)
response = send_request(request) response = send_request(request)