taegis-sdk-python 1.2.17__tar.gz → 1.2.19__tar.gz

{taegis_sdk_python-1.2.17/taegis_sdk_python.egg-info → taegis_sdk_python-1.2.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: taegis-sdk-python
-Version: 1.2.17
+Version: 1.2.19
 Summary: Taegis Python SDK
 Home-page: https://github.com/secureworks/taegis-sdk-python
 Author: Secureworks

taegis_sdk_python-1.2.19/taegis_sdk_python/_version.py

@@ -0,0 +1 @@
+__version__ = "v1.2.19"

{taegis_sdk_python-1.2.17 → taegis_sdk_python-1.2.19}/taegis_sdk_python/service_core.py

@@ -230,6 +230,7 @@ class ServiceCore:
                     )
                 },
             ),
+            execute_timeout=self.service.execute_timeout,
         )
 
         return client
@@ -261,6 +262,7 @@ class ServiceCore:
         client = Client(
             transport=transport,
             schema=self.schema,
+            execute_timeout=self.service.execute_timeout,
         )
 
         return client

{taegis_sdk_python-1.2.17 → taegis_sdk_python-1.2.19}/taegis_sdk_python/services/__init__.py

@@ -44,6 +44,7 @@ from taegis_sdk_python.services.file_info import FileInfoService
 from taegis_sdk_python.services.ingest_stats import IngestStatsService
 from taegis_sdk_python.services.investigations import InvestigationsService
 from taegis_sdk_python.services.investigations2 import Investigations2Service
+from taegis_sdk_python.services.isensor import IsensorService
 from taegis_sdk_python.services.mitre_attack_info import MitreAttackInfoService
 from taegis_sdk_python.services.multi_tenant_context import MultiTenantContextService
 from taegis_sdk_python.services.multi_tenant_ioc import MultiTenantIocService
@@ -86,6 +87,7 @@ class GraphQLService:
         proxy_headers: Optional[LooseHeaders] = None,
         trust_env: bool = False,
         ssl: Optional[Union[SSLContext, Literal[False], Fingerprint]] = None,
+        execute_timeout: Optional[Union[int, float]] = 30,
     ):  # pylint: disable=too-many-statements
         """
         GraphQLService
@@ -139,6 +141,8 @@ class GraphQLService:
         self._ssl = ssl
         self._trust_env = trust_env
 
+        self._execute_timeout = execute_timeout
+
         self._access_points = None
         self._agent = None
         self._alerts = None
@@ -163,6 +167,7 @@ class GraphQLService:
         self._ingest_stats = None
         self._investigations = None
         self._investigations2 = None
+        self._isensor = None
         self._mitre_attack_info = None
         self._multi_tenant_context = None
         self._multi_tenant_ioc = None
@@ -338,6 +343,11 @@ class GraphQLService:
 
         return value
 
+    @property
+    def execute_timeout(self):
+        """GraphQL Execute Timeout."""
+        return self._context_manager.get("execute_timeout", self._execute_timeout)
+
     @property
     def access_points(self):
         """Access Points Service Endpoint."""
@@ -506,6 +516,13 @@ class GraphQLService:
             self._investigations2 = Investigations2Service(self)
         return self._investigations2
 
+    @property
+    def isensor(self):
+        """Isensor Service Endpoint."""
+        if not self._isensor:
+            self._isensor = IsensorService(self)
+        return self._isensor
+
     @property
     def mitre_attack_info(self):
         """MitreAttackInfo Service Endpoint."""

{taegis_sdk_python-1.2.17 → taegis_sdk_python-1.2.19}/taegis_sdk_python/services/endpoint_management_service/mutations.py

@@ -125,7 +125,7 @@ class TaegisSDKEndpointManagementServiceMutation:
             return EndpointGroup.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation rotateRegistrationKey")
 
-    def create_agent_setting(self, input_: CreateAgentSettingsInput) -> AgentSetting:
+    def create_agent_setting(self, input_: AgentSettingsInput) -> AgentSetting:
         """create agent settings item."""
         endpoint = "createAgentSetting"
 
@@ -140,13 +140,16 @@ class TaegisSDKEndpointManagementServiceMutation:
             return AgentSetting.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation createAgentSetting")
 
-    def update_agent_setting(self, input_: UpdateAgentSettingsInput) -> AgentSetting:
+    def update_agent_setting(
+        self, id_: str, input_: AgentSettingsInput
+    ) -> AgentSetting:
         """update agent settings item."""
         endpoint = "updateAgentSetting"
 
         result = self.service.execute_mutation(
             endpoint=endpoint,
             variables={
+                "id": prepare_input(id_),
                 "input": prepare_input(input_),
             },
             output=build_output_string(AgentSetting),
@@ -155,9 +158,7 @@ class TaegisSDKEndpointManagementServiceMutation:
             return AgentSetting.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation updateAgentSetting")
 
-    def update_default_agent_setting(
-        self, input_: UpdateDefaultAgentSettingInput
-    ) -> AgentSetting:
+    def update_default_agent_setting(self, input_: AgentSettingsInput) -> AgentSetting:
         """update the default agent settings item."""
         endpoint = "updateDefaultAgentSetting"
 
@@ -186,3 +187,71 @@ class TaegisSDKEndpointManagementServiceMutation:
         if result.get(endpoint) is not None:
             return result.get(endpoint)
         raise GraphQLNoRowsInResultSetError("for mutation deleteAgentSetting")
+
+    def update_group_policy_override(
+        self, input_: UpdatePolicyOverrideInput
+    ) -> EndpointGroup:
+        """Update the policy override for an endpoint group."""
+        endpoint = "updateGroupPolicyOverride"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(EndpointGroup),
+        )
+        if result.get(endpoint) is not None:
+            return EndpointGroup.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for mutation updateGroupPolicyOverride")
+
+    def delete_group_policy_override(self, id_: str) -> EndpointGroup:
+        """Delete the policy override for an endpoint group."""
+        endpoint = "deleteGroupPolicyOverride"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "id": prepare_input(id_),
+            },
+            output=build_output_string(EndpointGroup),
+        )
+        if result.get(endpoint) is not None:
+            return EndpointGroup.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for mutation deleteGroupPolicyOverride")
+
+    def update_agent_setting_policy_override(
+        self, input_: UpdatePolicyOverrideInput
+    ) -> AgentSetting:
+        """Update the policy override for an agent setting."""
+        endpoint = "updateAgentSettingPolicyOverride"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(AgentSetting),
+        )
+        if result.get(endpoint) is not None:
+            return AgentSetting.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError(
+            "for mutation updateAgentSettingPolicyOverride"
+        )
+
+    def delete_agent_setting_policy_override(self, id_: str) -> AgentSetting:
+        """Delete the policy override for an agent setting."""
+        endpoint = "deleteAgentSettingPolicyOverride"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "id": prepare_input(id_),
+            },
+            output=build_output_string(AgentSetting),
+        )
+        if result.get(endpoint) is not None:
+            return AgentSetting.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError(
+            "for mutation deleteAgentSettingPolicyOverride"
+        )

{taegis_sdk_python-1.2.17 → taegis_sdk_python-1.2.19}/taegis_sdk_python/services/endpoint_management_service/queries.py

@@ -189,3 +189,20 @@ class TaegisSDKEndpointManagementServiceQuery:
         if result.get(endpoint) is not None:
             return AgentSettings.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query agentSettings")
+
+    def get_settings_for_profile(
+        self, profile_name: Optional[Union[AgentSettingsProfile, TaegisEnum]] = None
+    ) -> AgentSettingsDefaults:
+        """fetch the setting defaults for a given profile."""
+        endpoint = "getSettingsForProfile"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "profileName": prepare_input(profile_name),
+            },
+            output=build_output_string(AgentSettingsDefaults),
+        )
+        if result.get(endpoint) is not None:
+            return AgentSettingsDefaults.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query getSettingsForProfile")