cli2 3.3.30__tar.gz → 3.3.31__tar.gz

{cli2-3.3.30/cli2.egg-info → cli2-3.3.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cli2
-Version: 3.3.30
+Version: 3.3.31
 Summary: image:: https://yourlabs.io/oss/cli2/badges/master/pipeline.svg
 Home-page: https://yourlabs.io/oss/cli2
 Author: James Pic

{cli2-3.3.30 → cli2-3.3.31}/cli2/client.py

@@ -75,9 +75,14 @@ class Paginator:
     .. py:attribute:: model
 
         :py:class:`Model` class or ``dict`` by default.
+
+    .. py:attribute:: callback
+
+        Callback called for every item.
     """
 
-    def __init__(self, client, url, params=None, model=None, expressions=None):
+    def __init__(self, client, url, params=None, model=None, expressions=None,
+                 callback=None):
         """
         Initialize a paginator object with a client on a URL with parameters.
 
@@ -93,6 +98,7 @@ class Paginator:
         self.page_start = 1
         self.per_page = None
         self.initialized = False
+        self.callback = callback
         self.expressions = []
         for expression in (expressions or []):
             if not isinstance(expression, Expression):
@@ -270,6 +276,8 @@ class Paginator:
         """
         Asynchronous iterator.
         """
+        callback = callback or self.callback
+
         if self._reverse and not self.total_pages:
             first_page_response = await self.page_response(1)
             page = self.total_pages
@@ -287,6 +295,8 @@ class Paginator:
 
             for item in items:
                 if not python_filter or python_filter.matches(item):
+                    if callback:
+                        callback(item)
                     yield item
 
             if self._reverse:
@@ -298,6 +308,8 @@ class Paginator:
                     items = self.response_items(first_page_response)
                     for item in reversed(items):
                         if not python_filter or python_filter.matches(item):
+                            if callback:
+                                callback(item)
                             yield item
                     break
             else:

{cli2-3.3.30 → cli2-3.3.31}/cli2/test_client.py

@@ -347,7 +347,13 @@ async def test_pagination(httpx_mock):
     httpx_mock.add_response(url='http://lol/?page=2', json=[dict(a=2)])
     httpx_mock.add_response(url='http://lol/?page=3', json=[])
     client = Client(base_url='http://lol')
-    assert await client.paginate('/').list() == [dict(a=1), dict(a=2)]
+    paginator = client.paginate('/')
+    paginator.callback = mock.Mock()
+    assert await paginator.list() == [dict(a=1), dict(a=2)]
+    assert paginator.callback.call_args_list == [
+        mock.call({'a': 1}),
+        mock.call({'a': 2})
+    ]
 
 
 @pytest.mark.asyncio

{cli2-3.3.30 → cli2-3.3.31/cli2.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cli2
-Version: 3.3.30
+Version: 3.3.31
 Summary: image:: https://yourlabs.io/oss/cli2/badges/master/pipeline.svg
 Home-page: https://yourlabs.io/oss/cli2
 Author: James Pic

{cli2-3.3.30 → cli2-3.3.31}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup
 
 setup(
     name='cli2',
-    version='3.3.30',
+    version='3.3.31',
     setup_requires='setupmeta',
     install_requires=[
         'docstring_parser',