From 2bfee316c493b1296182da1a9cf04f5ee7056aaf Mon Sep 17 00:00:00 2001
From: Mark Cotton <tifosigoon@gmail.com>
Date: Thu, 17 Aug 2023 16:28:25 -0500
Subject: [PATCH] adding pagination to accounts, deployed at 0.0.14

---
 src/EagleEyev3/__init__.py | 68 ++++++++++++++++++++++----------------
 1 file changed, 39 insertions(+), 29 deletions(-)

diff --git a/src/EagleEyev3/__init__.py b/src/EagleEyev3/__init__.py
index 708e5fc..d1e9053 100644
--- a/src/EagleEyev3/__init__.py
+++ b/src/EagleEyev3/__init__.py
@@ -1,5 +1,5 @@
 """ Python client for Eagle Eye Networks APIv3 """
-version = "0.0.13"
+version = "0.0.14"
 __version__ = version
 
 
@@ -427,39 +427,49 @@ class EagleEyev3():
         Returns:
             dict: Dictionary containing the success status, response HTTP status code, and data.
         """
-        url = f"https://{self.user_base_url}/api/v3.0/accounts"
-        headers = { 
-                    "Authorization": f"Bearer {self.access_token}", 
-                    "Accept": "application/json"
+        nextPageToken = None
+
+        # emulating a do while toop in order to handle pagination, remember to break out of this loop
+        while True:
+
+            if nextPageToken:
+                url = f"https://{self.user_base_url}/api/v3.0/accounts?pageToken={nextPageToken}"
+            else:
+                url = f"https://{self.user_base_url}/api/v3.0/accounts"
+
+            headers = { 
+                        "Authorization": f"Bearer {self.access_token}", 
+                        "Accept": "application/json"
+                    }
+
+            response = self._make_get_request(url=url, headers=headers, timeout='list_of_accounts')
+
+            if response:
+                response_json = response.json()
+
+                logging.info(f"{response.status_code} in get_list_of_accounts")
+
+            else:
+                return {
+                    "success": False,
+                    "response_http_status": 0,
+                    "data": None
                 }
 
-        try:
-            response = requests.get(url, headers=headers, timeout=self._get_timeout_values('list_of_accounts'))
-            response_json = response.json()
 
-            logging.info(f"{response.status_code} in get_list_of_accounts")
+            if response.status_code == 200:
+                success = True
+                self.accounts = [i for i in response_json['results'] if i not in self.accounts] + self.accounts
 
-        except requests.exceptions.Timeout:
-            logging.warn(f"timeout expired for get_list_of_accounts()")
-            return {
-                "success": False,
-                "response_http_status": 0,
-                "data": None
-            }
+                if 'nextPageToken' in response_json and len(response_json['nextPageToken']) > 0:
+                    nextPageToken = response_json['nextPageToken']
+                else:
+                    break
 
-        except requests.exceptions.RequestException as e:
-            logging.warn(e)
-            return {
-                "success": False,
-                "response_http_status": 0,
-                "data": None
-            }
+            else:
+                success = False
+                break
 
-        if response.status_code == 200:
-            success = True
-            self.accounts = [i for i in response_json['results'] if i not in self.accounts] + self.accounts
-        else:
-            success = False
 
         return {
             "success": success,
@@ -544,7 +554,7 @@ class EagleEyev3():
             return response
 
         except requests.exceptions.Timeout:
-            logging.warn(f"timeout expired get_list_of_cameras()")
+            logging.warn(f"timeout expired _make_get_request {timeout}")
             return None
 
         except requests.exceptions.RequestException as e: