Merge pull request #44 from ParclLabs/autopaginate

autopaginate param

CHANGELOG.md

@@ -1,3 +1,6 @@
+### v0.4.2
+- Add `auto_paginate` support for all endpoints (retrieve & retrieve_many)
+
 ### v0.4.1
 - Backend Parcl Labs Services refactor
 

parcllabs/__version__.py

@@ -1 +1 @@
-VERSION = "0.4.1"
+VERSION = "0.4.2"

parcllabs/services/parcllabs_service.py

@@ -151,6 +151,7 @@ def retrieve_many_items(
         self,
         parcl_ids: List[int],
         params: Optional[Mapping[str, Any]] = None,
+        auto_paginate: bool = False,
         get_key_on_last_request: List[str] = [],
         **kwargs,
     ) -> Dict[str, Any]:
@@ -171,7 +172,12 @@ def retrieve_many_items(
         with alive_bar(len(parcl_ids)) as bar:
             for parcl_id in parcl_ids:
                 try:
-                    output = self.retrieve(parcl_id=parcl_id, params=params, **kwargs)
+                    output = self.retrieve(
+                        parcl_id=parcl_id,
+                        params=params,
+                        auto_paginate=auto_paginate,
+                        **kwargs,
+                    )
                     results[parcl_id] = output.get("items")
                 except RequestException as e:
                     # continue if no data is found for the parcl_id
@@ -189,6 +195,7 @@ def retrieve_many(
         end_date: str = None,
         params: Optional[Mapping[str, Any]] = None,
         as_dataframe: bool = False,
+        auto_paginate: bool = False,
         get_key_on_last_request: List[str] = [],
     ):
         start_date = self.validate_date(start_date)
@@ -203,6 +210,7 @@ def retrieve_many(
         results, additional_output = self.retrieve_many_items(
             parcl_ids=parcl_ids,
             params=params,
+            auto_paginate=auto_paginate,
             get_key_on_last_request=get_key_on_last_request,
         )
 

parcllabs/services/portfolio_size_service.py

@@ -38,6 +38,7 @@ def retrieve_many(
         portfolio_size: str = None,
         params: Optional[Mapping[str, Any]] = {},
         as_dataframe: bool = False,
+        auto_paginate: bool = False,
     ):
         portfolio_size = self.validate_portfolio_size(portfolio_size)
 
@@ -50,5 +51,6 @@ def retrieve_many(
             end_date=end_date,
             params=params,
             as_dataframe=as_dataframe,
+            auto_paginate=auto_paginate,
             get_key_on_last_request=["portfolio_size"],
         )

parcllabs/services/property_type_service.py

@@ -38,6 +38,7 @@ def retrieve_many(
         property_type: str = None,
         params: Optional[Mapping[str, Any]] = {},
         as_dataframe: bool = False,
+        auto_paginate: bool = False,
     ):
         property_type = self.validate_property_type(property_type)
 
@@ -50,5 +51,6 @@ def retrieve_many(
             end_date=end_date,
             params=params,
             as_dataframe=as_dataframe,
+            auto_paginate=auto_paginate,
             get_key_on_last_request=["property_type"],
         )