adding count of API requests

tmunzer · Sep 25, 2023 · 0122a08 · 0122a08
1 parent 673553d
commit 0122a08
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/mistapi/__api_request.py b/src/mistapi/__api_request.py
@@ -28,7 +28,10 @@ def __init__(self):
         self._cloud_uri = ""
         self._session = requests.session()
         self.privileges = ""
+        self._count = 0
 
+    def get_request_count(self):
+        return self._count
 
     def _url(self, uri) -> str:
         """
@@ -113,6 +116,7 @@ def mist_get(self, uri:str, query:dict=None) -> APIResponse:
             logger.error(f'apirequest:mist_get: Other error occurred: {err}')  # Python 3.6
             logger.error("apirequest:mist_get: Exception occurred", exc_info=True)
         finally:
+            self._count += 1
             return APIResponse(url=url, response=resp)
 
     def mist_post(self, uri:str,  body:dict=None) -> APIResponse:
@@ -149,6 +153,7 @@ def mist_post(self, uri:str,  body:dict=None) -> APIResponse:
             logger.error(f'apirequest:mist_post: Other error occurred: {err}')  # Python 3.6
             logger.error("apirequest:mist_post: Exception occurred", exc_info=True)
         finally: 
+            self._count += 1
             return APIResponse(url=url, response=resp)
 
     def mist_put(self, uri:str, body:dict=None) -> APIResponse:
@@ -185,6 +190,7 @@ def mist_put(self, uri:str, body:dict=None) -> APIResponse:
             logger.error(f'apirequest:mist_put: Other error occurred: {err}')  # Python 3.6
             logger.error("apirequest:mist_put: Exception occurred", exc_info=True)
         finally: 
+            self._count += 1
             return APIResponse(url=url, response=resp)
 
     def mist_delete(self, uri:str, query:dict=None) -> APIResponse:
@@ -213,6 +219,7 @@ def mist_delete(self, uri:str, query:dict=None) -> APIResponse:
             logger.error(f'apirequest:mist_delete: Other error occurred: {err}')  # Python 3.6
             logger.error("apirequest:mist_delete: Exception occurred", exc_info=True)
         else: 
+            self._count += 1
             return APIResponse(url=url, response=resp)
 
 
@@ -263,4 +270,5 @@ def mist_post_file(self, uri:str, multipart_form_data:dict={}) -> APIResponse:
             logger.error(f'apirequest:mist_post_file: Other error occurred: {err}')  # Python 3.6
             logger.error("apirequest:mist_post_file: Exception occurred", exc_info=True)
         else: 
+            self._count += 1
             return APIResponse(url=url, response=resp)
diff --git a/src/mistapi/__api_session.py b/src/mistapi/__api_session.py
@@ -93,6 +93,7 @@ def __init__(
         self._apitoken = None
         self._csrftoken = None
         self._authenticated = False
+        self._count=0
         self._session = requests.session()
         self._console_log_level = console_log_level
         self._logging_log_level = logging_log_level