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

{taegis_sdk_python-1.2.18/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.18
+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.18 → 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
@@ -166,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
@@ -514,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.18 → 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.18 → 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")