taegis-sdk-python 1.2.3__tar.gz → 1.2.5__tar.gz

{taegis-sdk-python-1.2.3/taegis_sdk_python.egg-info → taegis-sdk-python-1.2.5}/PKG-INFO

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

taegis-sdk-python-1.2.5/taegis_sdk_python/_version.py

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

{taegis-sdk-python-1.2.3 → taegis-sdk-python-1.2.5}/taegis_sdk_python/services/authz/types.py

@@ -41,6 +41,7 @@ class AuthzSupportedFeatureTarget(str, Enum):
 
     TENANT = "TENANT"
     CHILD_TENANTS = "CHILD_TENANTS"
+    PARTNER_TENANTS = "PARTNER_TENANTS"
 
 
 @dataclass_json
@@ -185,6 +186,9 @@ class AuthzSupportedFeatureState:
     enabled_in_child_tenants: Optional[bool] = field(
         default=None, metadata=config(field_name="enabledInChildTenants")
     )
+    enabled_in_partner_tenants: Optional[bool] = field(
+        default=None, metadata=config(field_name="enabledInPartnerTenants")
+    )
 
 
 @dataclass_json

{taegis-sdk-python-1.2.3 → taegis-sdk-python-1.2.5}/taegis_sdk_python/services/clients/types.py

@@ -92,6 +92,19 @@ class ClientRoleAssignmentInput:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class CredentialValidationResult:
+    """CredentialValidationResult."""
+
+    is_valid: Optional[bool] = field(
+        default=None, metadata=config(field_name="isValid")
+    )
+    environments: Optional[List[str]] = field(
+        default=None, metadata=config(field_name="environments")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class Client:

{taegis-sdk-python-1.2.3 → taegis-sdk-python-1.2.5}/taegis_sdk_python/services/endpoint_command_manager/types.py

@@ -26,6 +26,7 @@ class FetchRequestPathTypeEnum(str, Enum):
     """FetchRequestPathTypeEnum."""
 
     FILE = "FILE"
+    DIAGNOSTICS = "DIAGNOSTICS"
 
 
 class FetchRequestReasonCodeEnum(str, Enum):
@@ -33,6 +34,7 @@ class FetchRequestReasonCodeEnum(str, Enum):
 
     REASONCODE_UNSPECIFIED = "REASONCODE_UNSPECIFIED"
     POSSIBLY_MALICIOUS = "POSSIBLY_MALICIOUS"
+    TROUBLESHOOTING = "TROUBLESHOOTING"
 
 
 class StatusEnum(str, Enum):

{taegis-sdk-python-1.2.3 → taegis-sdk-python-1.2.5}/taegis_sdk_python/services/endpoint_management_service/mutations.py

@@ -123,3 +123,65 @@ class TaegisSDKEndpointManagementServiceMutation:
         if result.get(endpoint) is not None:
             return EndpointGroup.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation rotateRegistrationKey")
+
+    def create_agent_setting(self, input_: CreateAgentSettingsInput) -> AgentSetting:
+        """create agent settings item."""
+        endpoint = "createAgentSetting"
+
+        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 createAgentSetting")
+
+    def update_agent_setting(self, input_: UpdateAgentSettingsInput) -> AgentSetting:
+        """update agent settings item."""
+        endpoint = "updateAgentSetting"
+
+        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 updateAgentSetting")
+
+    def update_default_agent_setting(
+        self, input_: UpdateDefaultAgentSettingInput
+    ) -> AgentSetting:
+        """update the default agent settings item."""
+        endpoint = "updateDefaultAgentSetting"
+
+        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 updateDefaultAgentSetting")
+
+    def delete_agent_setting(self, id_: str) -> bool:
+        """delete agent settings item."""
+        endpoint = "deleteAgentSetting"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "id": prepare_input(id_),
+            },
+            output="",
+        )
+        if result.get(endpoint) is not None:
+            return result.get(endpoint)
+        raise GraphQLNoRowsInResultSetError("for mutation deleteAgentSetting")

{taegis-sdk-python-1.2.3 → taegis-sdk-python-1.2.5}/taegis_sdk_python/services/endpoint_management_service/queries.py

@@ -143,3 +143,47 @@ class TaegisSDKEndpointManagementServiceQuery:
         if result.get(endpoint) is not None:
             return BulkAssignRequestOutput.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query bulkAssignmentStatusByID")
+
+    def default_agent_setting(self) -> AgentSetting:
+        """fetch the default agent settings item."""
+        endpoint = "defaultAgentSetting"
+
+        result = self.service.execute_query(
+            endpoint=endpoint, variables={}, output=build_output_string(AgentSetting)
+        )
+        if result.get(endpoint) is not None:
+            return AgentSetting.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query defaultAgentSetting")
+
+    def agent_setting(self, id_: str) -> AgentSetting:
+        """fetch a single agent settings item by id."""
+        endpoint = "agentSetting"
+
+        result = self.service.execute_query(
+            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 query agentSetting")
+
+    def agent_settings(
+        self, first: Optional[int] = None, after: Optional[str] = None
+    ) -> AgentSettings:
+        """fetch paginated list of agent settings items."""
+        endpoint = "agentSettings"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "first": prepare_input(first),
+                "after": prepare_input(after),
+            },
+            output=build_output_string(AgentSettings),
+        )
+        if result.get(endpoint) is not None:
+            return AgentSettings.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query agentSettings")