zscaler-sdk-python 1.9.11__tar.gz → 1.9.12__tar.gz

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zscaler-sdk-python
-Version: 1.9.11
+Version: 1.9.12
 Summary: Official Python SDK for the Zscaler Products
 Home-page: https://github.com/zscaler/zscaler-sdk-python
 License: MIT

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zscaler-sdk-python"
-version = "1.9.11"
+version = "1.9.12"
 description = "Official Python SDK for the Zscaler Products"
 authors = ["Zscaler, Inc. <devrel@zscaler.com>"]
 license = "MIT"
@@ -87,3 +87,24 @@ max-line-length = 127
 
 [tool.isort]
 profile = "black"
+
+[tool.coverage.run]
+omit = [
+    "*/models/*",
+    "zscaler/*/models/*",
+    "tests/*",
+    "*/__init__.py",
+]
+
+[tool.coverage.report]
+omit = [
+    "*/models/*",
+    "zscaler/*/models/*",
+    "tests/*",
+    "*/__init__.py",
+]
+exclude_lines = [
+    "pragma: no cover",
+    "def __repr__",
+    "raise NotImplementedError",
+]

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/__init__.py

@@ -29,7 +29,7 @@ __license__ = "MIT"
 __contributors__ = [
     "William Guilherme",
 ]
-__version__ = "1.9.11"
+__version__ = "1.9.12"
 
 
 from zscaler.oneapi_client import Client as ZscalerClient  # noqa

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/utils.py

@@ -91,6 +91,7 @@ reformat_params = [
     ("entity_group_ids", "entityGroups"),
     ("scope_entity_ids", "adminScopeScopeEntities"),
     ("supported_region_ids", "supportedRegions"),
+    ("src_workload_groups_ids", "srcWorkloadGroups"),
 ]
 
 

zscaler_sdk_python-1.9.12/zscaler/zia/models/sandbox.py

@@ -0,0 +1,86 @@
+"""
+Copyright (c) 2023, Zscaler Inc.
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+"""
+
+from typing import Dict, List, Optional, Any, Union
+from zscaler.oneapi_object import ZscalerObject
+from zscaler.oneapi_collection import ZscalerCollection
+from zscaler.zia.models import common
+
+
+class BehavioralAnalysisAdvancedSettings(ZscalerObject):
+    """
+    A class for Behavioral Analysis Advanced Settings objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        """
+        Initialize the Behavioral Analysis Advanced Settings model based on API response.
+
+        Args:
+            config (dict): A dictionary representing the configuration.
+        """
+        super().__init__(config)
+
+        if config:
+            self.md5_hash_value_list = ZscalerCollection.form_list(
+                config["md5HashValueList"] if "md5HashValueList" in config else [], MD5HashValueList
+            )
+        else:
+            self.md5_hash_value_list = []
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        parent_req_format = super().request_format()
+        current_obj_format = {
+            "md5HashValueList": self.md5_hash_value_list,
+        }
+        parent_req_format.update(current_obj_format)
+        return parent_req_format
+
+
+class MD5HashValueList(ZscalerObject):
+    """
+    A class for MD5 Hash Value List objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        """
+        Initialize the MD5 Hash Value List model based on API response.
+        """
+        super().__init__(config)
+        if config:
+            self.url = config["url"] if "url" in config else None
+            self.url_comment = config["urlComment"] if "urlComment" in config else None
+            self.type = config["type"] if "type" in config else None
+        else:
+            self.url = None
+            self.url_comment = None
+            self.type = None
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        parent_req_format = super().request_format()
+        current_obj_format = {
+            "url": self.url,
+            "urlComment": self.url_comment,
+            "type": self.type,
+        }
+        parent_req_format.update(current_obj_format)
+        return parent_req_format

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/zia/sandbox.py

@@ -19,10 +19,12 @@ import mimetypes
 import time
 from zscaler.request_executor import RequestExecutor
 from zscaler.utils import format_url
+from zscaler.api_client import APIClient
 from zscaler.types import APIResult
+from zscaler.zia.models.sandbox import BehavioralAnalysisAdvancedSettings
 
 
-class CloudSandboxAPI:
+class CloudSandboxAPI(APIClient):
     """
     A Client object for the Cloud Sandbox resource.
     """
@@ -31,6 +33,7 @@ class CloudSandboxAPI:
     _zia_base_endpoint = "/zia/api/v1"
 
     def __init__(self, request_executor: "RequestExecutor") -> None:
+        super().__init__()
         self._request_executor: RequestExecutor = request_executor
 
     def submit_file(self, file_path: str, force: bool = False) -> APIResult[dict]:
@@ -268,10 +271,9 @@ class CloudSandboxAPI:
             return (None, response, error)
 
         try:
-            result = response.get_body()
+            result = BehavioralAnalysisAdvancedSettings(self.form_response_body(response.get_body()))
         except Exception as error:
             return (None, response, error)
-
         return (result, response, None)
 
     def get_file_hash_count(self) -> APIResult[dict]:
@@ -314,16 +316,37 @@ class CloudSandboxAPI:
 
         return (result, response, None)
 
-    def add_hash_to_custom_list(self, file_hashes_to_be_blocked: list) -> APIResult[dict]:
+    def add_hash_to_custom_list(
+        self,
+        md5_hash_value_list: List[Dict[str, str]],
+    ) -> APIResult[BehavioralAnalysisAdvancedSettings]:
         """
         Updates the custom list of MD5 file hashes that are blocked by Sandbox.
 
         Args:
-            file_hashes_to_be_blocked (:obj:`list` of :obj:`str`):
-                The list of MD5 Hashes to be added. Pass an empty list to clear the blocklist.
+            md5_hash_value_list (list[dict]): A list of MD5 hash entries to be blocked.
+                Each entry should be a dictionary with the following keys:
+                    - url (str): The MD5 hash value.
+                    - urlComment (str): A comment describing the hash.
+                    - type (str): The type of threat, e.g., "MALWARE".
+                Pass an empty list to clear the blocklist.
 
         Returns:
-            tuple: A tuple containing the result, response, and error.
+            tuple: A tuple containing (BehavioralAnalysisAdvancedSettings, Response, error).
+
+        Examples:
+            Add MD5 hashes to the sandbox blocklist:
+
+            >>> hash_list = [
+            ...     {
+            ...         "url": "42914d6d213a20a2684064be5c80ffa9",
+            ...         "urlComment": "Malicious file detected",
+            ...         "type": "MALWARE"
+            ...     }
+            ... ]
+            >>> result, response, error = client.zia.sandbox.add_hash_to_custom_list(
+            ...     md5_hash_value_list=hash_list
+            ... )
         """
         http_method = "put".upper()
         api_url = format_url(
@@ -333,7 +356,7 @@ class CloudSandboxAPI:
             """
         )
 
-        payload = {"fileHashesToBeBlocked": file_hashes_to_be_blocked}
+        payload = {"md5HashValueList": md5_hash_value_list}
 
         request, error = self._request_executor.create_request(method=http_method, endpoint=api_url, body=payload)
 
@@ -344,5 +367,10 @@ class CloudSandboxAPI:
 
         if error:
             return (None, response, error)
-        time.sleep(2)
-        return self.get_behavioral_analysis()
+
+        try:
+            result = BehavioralAnalysisAdvancedSettings(self.form_response_body(response.get_body()))
+        except Exception as error:
+            return (None, response, error)
+
+        return (result, response, None)

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/zia/url_filtering.py

@@ -62,11 +62,18 @@ class URLFilteringAPI(APIClient):
             query_params {dict}: Map of query parameters for the request.
                 ``[query_params.search]`` {str}: Search string for filtering results by rule name.
 
+                ``[query_params.page]`` (int): Specifies the page offset.
+
+                ``[query_params.page_size]`` (int): Specifies the page size.
+                    The default size is 100, but the maximum size is 100.
+
         Returns:
             tuple: A tuple containing (list of url filtering rules instances, Response, error)
 
         Examples:
-        >>> rules_list, _, error = client.zia.url_filtering.list_rules()
+        >>> rules_list, _, error = client.zia.url_filtering.list_rules(
+        ... query_params={'page': 1, 'page_size': 10}
+        )
         >>> if error:
         ...     print(f"Error listing url filtering rules: {error}")
         ...     return
@@ -84,31 +91,26 @@ class URLFilteringAPI(APIClient):
 
         query_params = query_params or {}
 
-        local_search = query_params.pop("search", None)
-
         body = {}
         headers = {}
 
         request, error = self._request_executor.create_request(http_method, api_url, body, headers, params=query_params)
+
         if error:
             return (None, None, error)
 
         response, error = self._request_executor.execute(request)
+
         if error:
             return (None, response, error)
 
         try:
-            results = []
+            result = []
             for item in response.get_results():
-                results.append(URLFilteringRule(self.form_response_body(item)))
-        except Exception as exc:
-            return (None, response, exc)
-
-        if local_search:
-            lower_search = local_search.lower()
-            results = [r for r in results if lower_search in (r.name.lower() if r.name else "")]
-
-        return (results, response, None)
+                result.append(URLFilteringRule(self.form_response_body(item)))
+        except Exception as error:
+            return (None, response, error)
+        return (result, response, None)
 
     def get_rule(
         self,

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/ztw/forwarding_rules.py

@@ -141,6 +141,7 @@ class ForwardingControlRulesAPI(APIClient):
             nw_application_groups (list): IDs for network application groups.
             device_groups (list): Device groups managed using Zscaler Client Connector.
             devices (list): Devices managed using Zscaler Client Connector.
+            src_workload_groups (list): IDs for source workload groups.
 
             zpa_app_segments (list[dict]): **ZPA Application Segments applicable to the rule.**
                 - `external_id` (str): Indicates the external ID. Applicable only when this reference is of an external entity.
@@ -170,6 +171,7 @@ class ForwardingControlRulesAPI(APIClient):
             ...    dest_addresses=["192.168.255.1"],
             ...    dest_ip_categories=["ZSPROXY_IPS"],
             ...    dest_countries=["COUNTRY_CA", "COUNTRY_US"],
+            ...    src_workload_groups_ids=["1234567890", "1234567891"],
             ... )
 
             Add a ZPA forwarding control rule:
@@ -274,6 +276,7 @@ class ForwardingControlRulesAPI(APIClient):
             nw_application_groups (list): IDs for network application groups.
             device_groups (list): Device groups managed using Zscaler Client Connector.
             devices (list): Devices managed using Zscaler Client Connector.
+            src_workload_groups (list): IDs for source workload groups.
 
             zpa_app_segments (list[dict]): **ZPA Application Segments applicable to the rule.**
                 - `external_id` (str): Indicates the external ID. Applicable only when this reference is of an external entity.
@@ -304,6 +307,7 @@ class ForwardingControlRulesAPI(APIClient):
             ...    dest_addresses=["192.168.255.1"],
             ...    dest_ip_categories=["ZSPROXY_IPS"],
             ...    dest_countries=["COUNTRY_CA", "COUNTRY_US"],
+            ...    src_workload_groups_ids=["1234567890", "1234567891"],
             ... )
 
             Update a ZPA forwarding control rule:
@@ -318,6 +322,7 @@ class ForwardingControlRulesAPI(APIClient):
             ...    dest_addresses=["192.168.255.1"],
             ...    dest_ip_categories=["ZSPROXY_IPS"],
             ...    dest_countries=["COUNTRY_CA", "COUNTRY_US"],
+            ...    src_workload_groups_ids=["1234567890", "1234567891"],
             ...    zpa_gateway={
             ...     "name": "ZPAGW01",
             ...     "external_id": "2"

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/ztw/legacy.py

@@ -506,6 +506,16 @@ class LegacyZTWClientHelper:
         from zscaler.ztw.discovery_service import DiscoveryServiceAPI
         return DiscoveryServiceAPI(self.request_executor)
 
+    @property
+    def workload_groups(self) -> "WorkloadGroupsAPI":
+        """
+        The interface object for the :ref:`ZTW Workload Groups <ztw-workload_groups>`.
+
+        """
+        from zscaler.ztw.workload_groups import WorkloadGroupsAPI
+
+        return WorkloadGroupsAPI(self.request_executor)
+
     """
     Misc
     """

{zscaler_sdk_python-1.9.11 → zscaler_sdk_python-1.9.12}/zscaler/ztw/models/forwarding_rules.py

@@ -125,6 +125,10 @@ class ForwardingControlRule(ZscalerObject):
                 config["labels"] if "labels" in config else [], common.CommonIDNameExternalID
             )
 
+            self.src_workload_groups = ZscalerCollection.form_list(
+                config["srcWorkloadGroups"] if "srcWorkloadGroups" in config else [], common.CommonIDNameExternalID
+            )
+
             self.devices = ZscalerCollection.form_list(
                 config["devices"] if "devices" in config else [], common.CommonIDNameExternalID
             )
@@ -205,6 +209,7 @@ class ForwardingControlRule(ZscalerObject):
             self.zpa_app_segments = []
             self.zpa_application_segments = []
             self.zpa_application_segment_groups = []
+            self.src_workload_groups = []
             self.proxy_gateway = None
             self.zpa_gateway = None
             self.zpa_broker_rule = None
@@ -252,6 +257,7 @@ class ForwardingControlRule(ZscalerObject):
             "labels": [label.request_format() for label in (self.labels or [])],
             "devices": [device.request_format() for device in (self.devices or [])],
             "deviceGroups": [dg.request_format() for dg in (self.device_groups or [])],
+            "srcWorkloadGroups": [wg.request_format() for wg in (self.src_workload_groups or [])],
             "zpaAppSegments": [zpa.request_format() for zpa in (self.zpa_app_segments or [])],
             "zpaApplicationSegments": [zpa_app.request_format() for zpa_app in (self.zpa_application_segments or [])],
             "zpaApplicationSegmentGroups": [

zscaler_sdk_python-1.9.12/zscaler/ztw/models/workload_groups.py

@@ -0,0 +1,210 @@
+"""
+Copyright (c) 2023, Zscaler Inc.
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+"""
+
+from typing import Dict, List, Optional, Any, Union
+from zscaler.oneapi_object import ZscalerObject
+from zscaler.oneapi_collection import ZscalerCollection
+from zscaler.ztw.models import common as common
+
+
+class WorkloadGroups(ZscalerObject):
+    """
+    A class for WorkloadGroup objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        super().__init__(config)
+        if config:
+            self.id = config["id"] if "id" in config else None
+            self.name = config["name"] if "name" in config else None
+            self.description = config["description"] if "description" in config else None
+            self.expression = config["expression"] if "expression" in config else None
+            self.last_modified_time = config["lastModifiedTime"] if "lastModifiedTime" in config else None
+
+            if "expressionJson" in config:
+                if isinstance(config["expressionJson"], ExpressionJson):
+                    self.expression_json = config["expressionJson"]
+                elif config["expressionJson"] is not None:
+                    self.expression_json = ExpressionJson(config["expressionJson"])
+                else:
+                    self.expression_json = None
+            else:
+                self.expression_json = None
+
+            if "lastModifiedBy" in config:
+                if isinstance(config["lastModifiedBy"], common.CommonBlocks):
+                    self.last_modified_by = config["lastModifiedBy"]
+                elif config["lastModifiedBy"] is not None:
+                    self.last_modified_by = common.CommonBlocks(config["lastModifiedBy"])
+                else:
+                    self.last_modified_by = None
+            else:
+                self.last_modified_by = None
+        else:
+            self.id = None
+            self.name = None
+            self.description = None
+            self.expression = None
+            self.last_modified_time = None
+            self.last_modified_by = None
+            self.expression_json = None
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        return {
+            "id": self.id,
+            "name": self.name,
+            "description": self.description,
+            "expression": self.expression,
+            "lastModifiedTime": self.last_modified_time,
+            "lastModifiedBy": self.last_modified_by,
+            "expressionJson": self.expression_json
+        }
+
+
+class ExpressionJson(ZscalerObject):
+    """
+    A class for ExpressionJson objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        super().__init__(config)
+        if config:
+            self.tag_type = config["tagType"] if "tagType" in config else None
+            self.operator = config["operator"] if "operator" in config else None
+
+            self.expression_containers = ZscalerCollection.form_list(
+                config["expressionContainers"] if "expressionContainers" in config else [], ExpressionContainers
+            )
+
+        else:
+            self.tag_type = None
+            self.operator = None
+            self.tag_container = None
+            self.expression_containers = []
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        parent_req_format = super().request_format()
+        current_obj_format = {
+            "tagType": self.tag_type,
+            "operator": self.operator,
+            "expressionContainers": self.expression_containers
+        }
+        parent_req_format.update(current_obj_format)
+        return parent_req_format
+
+
+class ExpressionContainers(ZscalerObject):
+    """
+    A class for ExpressionContainers objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        super().__init__(config)
+        if config:
+            self.tag_type = config["tagType"] if "tagType" in config else None
+            self.operator = config["operator"] if "operator" in config else None
+
+            if "tagContainer" in config:
+                if isinstance(config["tagContainer"], TagContainer):
+                    self.tag_container = config["tagContainer"]
+                elif config["tagContainer"] is not None:
+                    self.tag_container = TagContainer(config["tagContainer"])
+                else:
+                    self.tag_container = None
+            else:
+                self.tag_container = None
+
+        else:
+            self.tag_type = None
+            self.operator = None
+            self.tag_container = None
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        parent_req_format = super().request_format()
+        current_obj_format = {
+            "tagType": self.tag_type,
+            "operator": self.operator,
+            "tagContainer": self.tag_container
+        }
+        parent_req_format.update(current_obj_format)
+        return parent_req_format
+
+
+class TagContainer(ZscalerObject):
+    """
+    A class for TagContainer objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        super().__init__(config)
+        if config:
+            self.operator = config["operator"] if "operator" in config else None
+            self.tags = ZscalerCollection.form_list(
+                config["tags"] if "tags" in config else [], Tags
+            )
+
+        else:
+            self.operator = None
+            self.tags = None
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        parent_req_format = super().request_format()
+        current_obj_format = {
+            "operator": self.operator,
+            "tags": self.tags,
+        }
+        parent_req_format.update(current_obj_format)
+        return parent_req_format
+
+
+class Tags(ZscalerObject):
+    """
+    A class for Tags objects.
+    """
+
+    def __init__(self, config: Optional[Dict[str, Any]] = None) -> None:
+        super().__init__(config)
+        if config:
+            self.key = config["key"] if "key" in config else None
+            self.value = config["value"] if "value" in config else None
+
+        else:
+            self.key = None
+            self.value = None
+
+    def request_format(self) -> Dict[str, Any]:
+        """
+        Return the object as a dictionary in the format expected for API requests.
+        """
+        parent_req_format = super().request_format()
+        current_obj_format = {
+            "key": self.key,
+            "value": self.value,
+        }
+        parent_req_format.update(current_obj_format)
+        return parent_req_format