taegis-sdk-python 1.2.2.1__tar.gz → 1.2.4__tar.gz

{taegis-sdk-python-1.2.2.1/taegis_sdk_python.egg-info → taegis-sdk-python-1.2.4}/PKG-INFO

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

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

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

{taegis-sdk-python-1.2.2.1 → taegis-sdk-python-1.2.4}/taegis_sdk_python/authentication.py

@@ -2,6 +2,7 @@
 
 Authenication implementations for Taegis.
 """
+
 import logging
 import os
 from getpass import getpass
@@ -201,7 +202,7 @@ def get_token_by_password_grant(
     requests.HTTPError
         Any issue with retrieving token via HTTP
     """
-    auth_uri = "/auth/api/v2/auth/token"
+    auth_uri = "/universal-auth/token"
 
     if not password:
         password = getpass("Password: ")
@@ -218,10 +219,10 @@ def get_token_by_password_grant(
     try:
         mfa_token = response.json().get("mfa_token")
     except JSONDecodeError as exc:
-        raise HTTPError(response=response) from exc
+        raise HTTPError(response.text, response=response) from exc
 
     if not mfa_token:
-        raise HTTPError(response=response)
+        raise HTTPError(response.text, response=response)
 
     mfa_input = input("MFA Token: ")
 
@@ -238,13 +239,13 @@ def get_token_by_password_grant(
     try:
         access_token = response.json().get("access_token")
     except JSONDecodeError as exc:
-        raise HTTPError(response=response) from exc
+        raise HTTPError(response.text, response=response) from exc
 
     if not access_token:
         raise MissingAccessTokenError(
             message="Access token not found",
-            comments=["Check credentials provided to SSO provider."],
-            nested_exception=HTTPError(response=response),
+            comments=["Check credentials and MFA input."],
+            nested_exception=HTTPError(response.text, response=response),
         )
 
     return access_token
@@ -274,7 +275,7 @@ def get_token_by_sso_device_code(
     try:
         device_code_flow = response.json()
     except JSONDecodeError as exc:
-        raise HTTPError(response=response) from exc
+        raise HTTPError(response.text, response=response) from exc
 
     if device_code_flow.get("verification_uri_complete"):
         print(
@@ -303,13 +304,13 @@ def get_token_by_sso_device_code(
     try:
         access_token = response.json().get("access_token")
     except JSONDecodeError as exc:
-        raise HTTPError(response=response) from exc
+        raise HTTPError(response.text, response=response) from exc
 
     if not access_token:
         raise MissingAccessTokenError(
             message="Access token not found",
-            comments=["Check credentials and MFA input."],
-            nested_exception=HTTPError(response=response),
+            comments=["Check credentials provided to SSO provider."],
+            nested_exception=HTTPError(response.text, response=response),
         )
 
     return access_token

{taegis-sdk-python-1.2.2.1 → taegis-sdk-python-1.2.4}/taegis_sdk_python/services/alerts/types.py

@@ -1899,6 +1899,9 @@ class AggregateAlertsBySeverityInputInput:
     filter_custom_alerts: Optional[bool] = field(
         default=None, metadata=config(field_name="filter_custom_alerts")
     )
+    tenant_service_filters: Optional[List[str]] = field(
+        default=None, metadata=config(field_name="tenant_service_filters")
+    )
     group_by: Optional[AggregateAlertsBySeverityInput_GroupBy] = field(
         default=None, metadata=config(field_name="group_by")
     )

{taegis-sdk-python-1.2.2.1 → taegis-sdk-python-1.2.4}/taegis_sdk_python/services/byoti/queries.py

@@ -63,3 +63,20 @@ class TaegisSDKByotiQuery:
         if result.get(endpoint) is not None:
             return SearchIndicatorsResponse.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query getIndicators")
+
+    def paginate_indicators(
+        self, input_: Optional[PaginateIndicatorsInput] = None
+    ) -> SearchIndicatorsResponse:
+        """PaginateIndicators is similar to getIndicators but uses cursor based pagination."""
+        endpoint = "paginateIndicators"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(SearchIndicatorsResponse),
+        )
+        if result.get(endpoint) is not None:
+            return SearchIndicatorsResponse.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query paginateIndicators")

{taegis-sdk-python-1.2.2.1 → taegis-sdk-python-1.2.4}/taegis_sdk_python/services/byoti/types.py

@@ -141,6 +141,41 @@ class GetIndicatorsInput:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class PaginateIndicatorsInput:
+    """PaginateIndicatorsInput."""
+
+    updated_after: Optional[str] = field(
+        default=None, metadata=config(field_name="updated_after")
+    )
+    with_partner_tenants: Optional[bool] = field(
+        default=None, metadata=config(field_name="with_partner_tenants")
+    )
+    reference_url: Optional[str] = field(
+        default=None, metadata=config(field_name="reference_url")
+    )
+    source_name: Optional[str] = field(
+        default=None, metadata=config(field_name="source_name")
+    )
+    integration_id: Optional[int] = field(
+        default=None, metadata=config(field_name="integration_id")
+    )
+    exclude_deleted: Optional[bool] = field(
+        default=None, metadata=config(field_name="exclude_deleted")
+    )
+    per_page: Optional[int] = field(
+        default=None, metadata=config(field_name="per_page")
+    )
+    cursor: Optional[str] = field(default=None, metadata=config(field_name="cursor"))
+    object_type: Optional[ObjectType] = field(
+        default=None, metadata=config(field_name="object_type")
+    )
+    severity: Optional[Severity] = field(
+        default=None, metadata=config(field_name="severity")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class ByotiIndicator:
@@ -233,6 +268,7 @@ class SearchIndicatorsResponse:
     total_entries_size: Optional[int] = field(
         default=None, metadata=config(field_name="total_entries_size")
     )
+    cursor: Optional[str] = field(default=None, metadata=config(field_name="cursor"))
     indicators: Optional[List[ByotiIndicator]] = field(
         default=None, metadata=config(field_name="indicators")
     )

{taegis-sdk-python-1.2.2.1 → taegis-sdk-python-1.2.4}/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.2.1 → taegis-sdk-python-1.2.4}/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.2.1 → taegis-sdk-python-1.2.4}/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")