zscaler-sdk-python 0.10.0__tar.gz → 0.10.2__tar.gz

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zscaler-sdk-python
-Version: 0.10.0
+Version: 0.10.2
 Summary: Official Python SDK for the Zscaler Products (Beta)
 Home-page: https://github.com/zscaler/zscaler-sdk-python
 License: MIT
@@ -22,7 +22,7 @@ Classifier: Topic :: Security
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Provides-Extra: dev
 Requires-Dist: aenum ; extra == "dev"
-Requires-Dist: aiohttp (>=3.10.2)
+Requires-Dist: aiohttp (>=3.11.11)
 Requires-Dist: arrow
 Requires-Dist: black (>=24.3.0) ; extra == "dev"
 Requires-Dist: certifi (>=2024.2.2)
@@ -34,16 +34,16 @@ Requires-Dist: idna (>=3.10)
 Requires-Dist: okta (>=2.9.7)
 Requires-Dist: pycryptodomex (>=3.20.0)
 Requires-Dist: pydash (>=8.0.3) ; extra == "dev"
-Requires-Dist: python-box (>=7.2.0)
+Requires-Dist: python-box (>=7.3.0)
 Requires-Dist: python-dateutil
 Requires-Dist: pytz (>=2024.2)
 Requires-Dist: pyyaml
 Requires-Dist: requests (>=2.32.3)
 Requires-Dist: responses (>=0.25.3)
 Requires-Dist: restfly (>=1.5.0)
-Requires-Dist: six
+Requires-Dist: six (>=1.17.0)
 Requires-Dist: xmltodict (>=0.14.2)
-Requires-Dist: yarl (>=1.14.0)
+Requires-Dist: yarl (>=1.17.0)
 Project-URL: Bug Tracker, https://github.com/zscaler/zscaler-sdk-python/issues
 Project-URL: Documentation, https://zscaler-sdk-python.readthedocs.io
 Project-URL: Repository, https://github.com/zscaler/zscaler-sdk-python

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zscaler-sdk-python"
-version = "0.10.0"
+version = "0.10.2"
 description = "Official Python SDK for the Zscaler Products (Beta)"
 authors = ["Zscaler, Inc. <devrel@zscaler.com>"]
 license = "MIT"
@@ -33,16 +33,16 @@ arrow = "*"
 certifi = ">=2024.2.2"
 charset-normalizer = "*"
 idna = ">=3.10"
-python-box = ">=7.2.0"
+python-box = ">=7.3.0"
 python-dateutil = "*"
 requests = ">=2.32.3"
 responses = ">=0.25.3"
 restfly = ">=1.5.0"
-six = "*"
+six = ">=1.17.0"
 flatdict = "*"
 pyyaml = "*"
 xmltodict = ">=0.14.2"
-yarl = ">=1.14.0"
+yarl = ">=1.17.0"
 pycryptodomex = ">=3.20.0"
 aenum = "*"
 pydash = ">=8.0.3"
@@ -51,7 +51,7 @@ pytz = ">=2024.2"
 black = ">=24.3.0"
 cryptography = ">=43.0.0"
 okta = ">=2.9.7"
-aiohttp = ">=3.10.2"
+aiohttp = ">=3.11.11"
 
 [tool.poetry.dev-dependencies]
 black = ">=24.3.0"
@@ -62,7 +62,7 @@ pytest-recording = "^0.13.2"
 pytest-cov = "*"
 pyfakefs = ">=5.6.0"
 isort = "*"
-wheel = "*"
+wheel = ">=0.45.1"
 sphinx = "^7.4.7"
 sphinx-autobuild = "*"
 sphinx_rtd_theme = "*"

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/zscaler/__init__.py

@@ -29,7 +29,7 @@ __license__ = "MIT"
 __contributors__ = [
     "William Guilherme",
 ]
-__version__ = "0.10.0"
+__version__ = "0.10.2"
 
 from zscaler.zdx import ZDXClientHelper  # noqa
 from zscaler.zia import ZIAClientHelper  # noqa

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/zscaler/zia/__init__.py

@@ -407,7 +407,6 @@ class ZIAClientHelper(ZIAClient):
         page=None,
         pagesize=None,
         search=None,
-        max_page_size=1000,  # Default to 1000, can be adjusted based on endpoint constraints
         max_items=None,  # Maximum number of items to retrieve across pages
         max_pages=None,  # Maximum number of pages to retrieve
         type=None,  # Specify type of VPN credentials (CN, IP, UFQDN, XAUTH)
@@ -423,7 +422,7 @@ class ZIAClientHelper(ZIAClient):
             path (str): The API endpoint path to send requests to.
             expected_status_code (int): The expected HTTP status code for a successful request. Defaults to 200.
             page (int): Specific page number to fetch. Defaults to 1 if not provided.
-            pagesize (int): Number of items per page, default is 100, with a maximum of 1000.
+            pagesize (int): Number of items per page, default is 100, with a maximum of 10000.
             search (str): Search query to filter the results.
             max_items (int): Maximum number of items to retrieve.
             max_pages (int): Maximum number of pages to fetch.
@@ -431,7 +430,7 @@ class ZIAClientHelper(ZIAClient):
             include_only_without_location (bool, optional): Filter to include only VPN credentials not associated with a location.
             location_id (int, optional): Retrieve VPN credentials for the specified location ID.
             managed_by (int, optional): Retrieve VPN credentials managed by the specified partner.
-            prefix (int, optional): Retrieve VPN credentials managed by the specified partner.
+            prefix (int, optional): Retrieve VPN credentials managed by a given partner.
 
         Returns:
             tuple: A tuple containing:
@@ -445,10 +444,10 @@ class ZIAClientHelper(ZIAClient):
             "EMPTY_RESULTS": "No results found for page {page}.",
         }
 
-        # Initialize pagination parameters
+        # Initialize parameters
         params = {
             "page": page if page is not None else 1,  # Start at page 1 if not specified
-            "pagesize": min(pagesize if pagesize is not None else 100, max_page_size),  # Apply max_page_size limit
+            "pageSize": pagesize if pagesize is not None else 100,  # Allow any user-defined pagesize
         }
 
         # Add optional filters to the params if provided
@@ -465,12 +464,30 @@ class ZIAClientHelper(ZIAClient):
         if prefix:
             params["prefix"] = prefix
 
+        # If the user specifies a page, fetch only that page
+        if page is not None:
+            response = self.send("GET", path=path, params=params)
+            if response.status_code != expected_status_code:
+                error_msg = ERROR_MESSAGES["UNEXPECTED_STATUS"].format(
+                    status_code=response.status_code, page=params["page"]
+                )
+                logger.error(error_msg)
+                return BoxList([]), error_msg
+
+            response_data = response.json()
+            if not isinstance(response_data, list):
+                error_msg = ERROR_MESSAGES["EMPTY_RESULTS"].format(page=params["page"])
+                logger.warn(error_msg)
+                return BoxList([]), error_msg
+
+            data = convert_keys_to_snake(response_data)
+            return BoxList(data), None
+
+        # If no page is specified, iterate through pages to fetch all items
         ret_data = []
         total_collected = 0
-
         try:
             while True:
-                # Apply rate-limiting if necessary
                 should_wait, delay = self.rate_limiter.wait("GET")
                 if should_wait:
                     time.sleep(delay)
@@ -495,13 +512,6 @@ class ZIAClientHelper(ZIAClient):
 
                 data = convert_keys_to_snake(response_data)
 
-                # If searching for a specific item, stop if we find a match
-                if search:
-                    for item in data:
-                        if item.get("name") == search:
-                            ret_data.append(item)
-                            return BoxList(ret_data), None
-
                 # Limit data collection based on max_items
                 if max_items is not None:
                     data = data[: max_items - total_collected]  # Limit items on the current page
@@ -514,8 +524,8 @@ class ZIAClientHelper(ZIAClient):
                 ):
                     break
 
-                # Stop if we've processed all available pages (i.e., less than requested page size)
-                if len(data) < params["pagesize"]:
+                # Stop if fewer items than pageSize are returned
+                if len(data) < params["pageSize"]:
                     break
 
                 # Move to the next page
@@ -531,6 +541,142 @@ class ZIAClientHelper(ZIAClient):
 
         return BoxList(ret_data), None
 
+
+    # def get_paginated_data(
+    #     self,
+    #     path=None,
+    #     expected_status_code=200,
+    #     page=None,
+    #     pagesize=None,
+    #     search=None,
+    #     max_items=None,  # Maximum number of items to retrieve across pages
+    #     max_pages=None,  # Maximum number of pages to retrieve
+    #     type=None,  # Specify type of VPN credentials (CN, IP, UFQDN, XAUTH)
+    #     include_only_without_location=None,  # Include only VPN credentials not associated with any location
+    #     location_id=None,  # VPN credentials for a specific location ID
+    #     managed_by=None,  # VPN credentials managed by a given partner
+    #     prefix=None,  # VPN credentials managed by a given partner
+    # ):
+    #     """
+    #     Fetches paginated data from the API based on specified parameters and handles pagination.
+
+    #     Args:
+    #         path (str): The API endpoint path to send requests to.
+    #         expected_status_code (int): The expected HTTP status code for a successful request. Defaults to 200.
+    #         page (int): Specific page number to fetch. Defaults to 1 if not provided.
+    #         pagesize (int): Number of items per page, default is 100, with a maximum of 1000.
+    #         search (str): Search query to filter the results.
+    #         max_items (int): Maximum number of items to retrieve.
+    #         max_pages (int): Maximum number of pages to fetch.
+    #         type (str, optional): Type of VPN credentials (e.g., CN, IP, UFQDN, XAUTH).
+    #         include_only_without_location (bool, optional): Filter to include only VPN credentials not associated with a location.
+    #         location_id (int, optional): Retrieve VPN credentials for the specified location ID.
+    #         managed_by (int, optional): Retrieve VPN credentials managed by the specified partner.
+    #         prefix (int, optional): Retrieve VPN credentials managed by the specified partner.
+
+    #     Returns:
+    #         tuple: A tuple containing:
+    #             - BoxList: A list of fetched items wrapped in a BoxList for easy access.
+    #             - str: An error message if any occurred during the data fetching process.
+    #     """
+    #     logger = logging.getLogger(__name__)
+
+    #     ERROR_MESSAGES = {
+    #         "UNEXPECTED_STATUS": "Unexpected status code {status_code} received for page {page}.",
+    #         "EMPTY_RESULTS": "No results found for page {page}.",
+    #     }
+
+    #     # Initialize pagination parameters
+    #     # params = {
+    #     #     "page": page if page is not None else 1,  # Start at page 1 if not specified
+    #     #     "pagesize": min(pagesize if pagesize is not None else 100, max_page_size),  # Apply max_page_size limit
+    #     # }
+
+    #     params = {
+    #         "page": page if page is not None else 1,  # Start at page 1 if not specified
+    #         "pagesize": max(100, min(pagesize or 100, 10000)),  # Ensure pagesize is within API limits
+    #     }
+        
+    #     # Add optional filters to the params if provided
+    #     if search:
+    #         params["search"] = search
+    #     if type:
+    #         params["type"] = type
+    #     if include_only_without_location is not None:
+    #         params["includeOnlyWithoutLocation"] = include_only_without_location
+    #     if location_id:
+    #         params["locationId"] = location_id
+    #     if managed_by:
+    #         params["managedBy"] = managed_by
+    #     if prefix:
+    #         params["prefix"] = prefix
+
+    #     ret_data = []
+    #     total_collected = 0
+
+    #     try:
+    #         while True:
+    #             # Apply rate-limiting if necessary
+    #             should_wait, delay = self.rate_limiter.wait("GET")
+    #             if should_wait:
+    #                 time.sleep(delay)
+
+    #             # Send the request to the API
+    #             response = self.send("GET", path=path, params=params)
+
+    #             # Check for unexpected status code
+    #             if response.status_code != expected_status_code:
+    #                 error_msg = ERROR_MESSAGES["UNEXPECTED_STATUS"].format(
+    #                     status_code=response.status_code, page=params["page"]
+    #                 )
+    #                 logger.error(error_msg)
+    #                 return BoxList([]), error_msg
+
+    #             # Parse the response as a flat list of items
+    #             response_data = response.json()
+    #             if not isinstance(response_data, list):
+    #                 error_msg = ERROR_MESSAGES["EMPTY_RESULTS"].format(page=params["page"])
+    #                 logger.warn(error_msg)
+    #                 return BoxList([]), error_msg
+
+    #             data = convert_keys_to_snake(response_data)
+
+    #             # If searching for a specific item, stop if we find a match
+    #             if search:
+    #                 for item in data:
+    #                     if item.get("name") == search:
+    #                         ret_data.append(item)
+    #                         return BoxList(ret_data), None
+
+    #             # Limit data collection based on max_items
+    #             if max_items is not None:
+    #                 data = data[: max_items - total_collected]  # Limit items on the current page
+    #             ret_data.extend(data)
+    #             total_collected += len(data)
+
+    #             # Check if we've reached max_items or max_pages limits
+    #             if (max_items is not None and total_collected >= max_items) or (
+    #                 max_pages is not None and params["page"] >= max_pages
+    #             ):
+    #                 break
+
+    #             # Stop if we've processed all available pages (i.e., less than requested page size)
+    #             if len(data) < params["pagesize"]:
+    #                 break
+
+    #             # Move to the next page
+    #             params["page"] += 1
+
+    #     finally:
+    #         time.sleep(2)  # Ensure a delay between requests regardless of outcome
+
+    #     if not ret_data:
+    #         error_msg = ERROR_MESSAGES["EMPTY_RESULTS"].format(page=params["page"])
+    #         logger.warn(error_msg)
+    #         return BoxList([]), error_msg
+
+    #     return BoxList(ret_data), None
+
     @property
     def admin_and_role_management(self):
         """
@@ -619,7 +765,7 @@ class ZIAClientHelper(ZIAClient):
 
         """
         return CloudSandboxAPI(self)
-
+    
     @property
     def security(self):
         """

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/zscaler/zia/pac_files.py

@@ -39,23 +39,13 @@ class PacFilesAPI:
                 Specifies the page size. The default size is 100, but the maximum size is 1000.
 
         Returns:
-            :obj:`BoxList`: The list of Rule Labels configured in ZIA.
+            :obj:`BoxList`: The list of PAC Files configured in ZIA.
 
         Examples:
-            List Rule Labels using default settings:
+            List PAC Files using default settings:
 
-            >>> for label in zia.labels.list_labels():
-            ...   print(label)
-
-            List labels, limiting to a maximum of 10 items:
-
-            >>> for label in zia.labels.list_labels(max_items=10):
-            ...    print(label)
-
-            List labels, returning 200 items per page for a maximum of 2 pages:
-
-            >>> for label in zia.labels.list_labels(page_size=200, max_pages=2):
-            ...    print(label)
+            >>> for pac_files in zia.pac_files.list_pac_files():
+            ...   print(pac_files)
 
         """
         list, _ = self.rest.get_paginated_data(path="/pacFiles", **kwargs)

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/zscaler/zia/users.py

@@ -170,7 +170,7 @@ class UserManagementAPI:
             **name (str, optional):
                 Filters by user name. This is a `partial` match.
             **page_size (int, optional):
-                Specifies the page size. The default size is 100, but the maximum size is 1000.
+                Specifies the page size. The default size is 100, but the maximum size is 10000.
             **sort_by (str):
                 The field name to sort by, supported values: id, name, creationTime or modifiedTime (default to name)
             **sort_order (str):

{zscaler_sdk_python-0.10.0 → zscaler_sdk_python-0.10.2}/zscaler/zpa/policies.py

@@ -15,8 +15,6 @@
 # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
 
-import functools
-
 from box import Box, BoxList
 from requests import Response
 
@@ -61,6 +59,7 @@ class PolicySetsAPI:
         """
         template = []
         app_and_app_group_operands = []
+        scim_and_scim_group_operands = []
         object_types_to_operands = {
             "CONSOLE": [],
             "MACHINE_GRP": [],
@@ -100,6 +99,8 @@ class PolicySetsAPI:
 
                 if object_type in ["APP", "APP_GROUP"]:
                     app_and_app_group_operands.append(operand)
+                elif object_type in ["SCIM","SCIM_GROUP"]:
+                    scim_and_scim_group_operands.append(operand)
                 elif object_type in object_types_to_operands:
                     object_types_to_operands[object_type].append(operand)
 
@@ -131,6 +132,11 @@ class PolicySetsAPI:
             app_group_operator = operators_for_types.get("APP", "OR")
             template.append({"operator": app_group_operator, "operands": app_and_app_group_operands})
 
+        # Combine SCIM and SCIM_GROUP operands with their specific operator
+        if scim_and_scim_group_operands:
+            scim_group_operator = operators_for_types.get("SCIM_GROUP", "OR")
+            template.append({"operator": scim_group_operator, "operands": scim_and_scim_group_operands})
+
         # Combine other object types into their blocks with their respective operator
         for object_type, operands in object_types_to_operands.items():
             if operands: