Export rest_client.paginate to helpers.requests module

dlt-hub · Mar 25, 2024 · 4a31b34 · 4a31b34
1 parent a44c9b4
commit 4a31b34
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/dlt/sources/helpers/requests/__init__.py b/dlt/sources/helpers/requests/__init__.py
@@ -15,11 +15,12 @@
 from requests.exceptions import ChunkedEncodingError
 from dlt.sources.helpers.requests.retry import Client
 from dlt.sources.helpers.requests.session import Session
+from dlt.sources.helpers.rest_client import paginate
 from dlt.common.configuration.specs import RunConfiguration
 
 client = Client()
 
-get, post, put, patch, delete, options, head, request = (
+get, post, put, patch, delete, options, head, request, paginate = (
     client.get,
     client.post,
     client.put,
@@ -28,6 +29,7 @@
     client.options,
     client.head,
     client.request,
+    paginate,
 )
 
 

diff --git a/tests/sources/helpers/rest_client/test_client.py b/tests/sources/helpers/rest_client/test_client.py
@@ -2,8 +2,8 @@
 import pytest
 from typing import Any, cast
 from dlt.common.typing import TSecretStrValue
-from dlt.sources.helpers.requests import Response, Request
-from dlt.sources.helpers.rest_client import RESTClient, paginate
+from dlt.sources.helpers.requests import Response, Request, paginate
+from dlt.sources.helpers.rest_client import RESTClient
 from dlt.sources.helpers.rest_client.client import Hooks
 from dlt.sources.helpers.rest_client.paginators import JSONResponsePaginator