taegis-sdk-python 1.2.30__tar.gz → 1.2.32__tar.gz

{taegis_sdk_python-1.2.30/taegis_sdk_python.egg-info → taegis_sdk_python-1.2.32}/PKG-INFO

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

taegis_sdk_python-1.2.32/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.2.30 → taegis_sdk_python-1.2.32}/taegis_sdk_python/services/__init__.py

@@ -19,6 +19,7 @@ from taegis_sdk_python.service_core import ServiceCore
 from taegis_sdk_python.services.access_points import AccessPointsService
 from taegis_sdk_python.services.agent import AgentService
 from taegis_sdk_python.services.alerts import AlertsService
+from taegis_sdk_python.services.alerts_history import AlertsHistoryService
 from taegis_sdk_python.services.assets import AssetsService
 from taegis_sdk_python.services.assets2 import Assets2Service
 from taegis_sdk_python.services.audits import AuditsService
@@ -168,6 +169,7 @@ class GraphQLService:
         self._access_points = None
         self._agent = None
         self._alerts = None
+        self._alerts_history = None
         self._assets = None
         self._assets2 = None
         self._audits = None
@@ -398,6 +400,13 @@ class GraphQLService:
             self._alerts = AlertsService(self)
         return self._alerts
 
+    @property
+    def alerts_history(self):
+        """AlertsHistory Service Endpoint."""
+        if not self._alerts_history:
+            self._alerts_history = AlertsHistoryService(self)
+        return self._alerts_history
+
     @property
     def assets(self):
         """Assets Service Endpoint."""

{taegis_sdk_python-1.2.30 → taegis_sdk_python-1.2.32}/taegis_sdk_python/services/alerts/mutations.py

@@ -113,7 +113,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_update_threat_score(
         self, in_: UpdateThreatScoreRequestInput
     ) -> UpdateThreatScoreResponse:
-        """Update threat score for a give list of alert IDs.."""
+        """Update threat score for a given list of alert IDs.."""
         endpoint = "alertsServiceUpdateThreatScore"
 
         result = self.service.execute_mutation(
@@ -128,3 +128,22 @@ class TaegisSDKAlertsMutation:
         raise GraphQLNoRowsInResultSetError(
             "for mutation alertsServiceUpdateThreatScore"
         )
+
+    def alerts_service_update_threat_score_v2(
+        self, in_: UpdateThreatScoreV2RequestInput
+    ) -> UpdateThreatScoreResponse:
+        """Update threat score v2 for a given list of alert IDs.."""
+        endpoint = "alertsServiceUpdateThreatScoreV2"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "in": prepare_input(in_),
+            },
+            output=build_output_string(UpdateThreatScoreResponse),
+        )
+        if result.get(endpoint) is not None:
+            return UpdateThreatScoreResponse.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError(
+            "for mutation alertsServiceUpdateThreatScoreV2"
+        )

{taegis_sdk_python-1.2.30 → taegis_sdk_python-1.2.32}/taegis_sdk_python/services/alerts/types.py

@@ -280,6 +280,19 @@ class UpdateThreatScoreEntry:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class UpdateThreatScoreV2Entry:
+    """UpdateThreatScoreV2Entry."""
+
+    alert_id: Optional[str] = field(
+        default=None, metadata=config(field_name="alert_id")
+    )
+    threat_score_v2: Optional[float] = field(
+        default=None, metadata=config(field_name="threat_score_v2")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class AccountCompromiseDetectorDetailInput:
@@ -1763,6 +1776,19 @@ class UpdateThreatScoreRequestInput:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class UpdateThreatScoreV2RequestInput:
+    """UpdateThreatScoreV2RequestInput."""
+
+    threat_score_v2: Optional[float] = field(
+        default=None, metadata=config(field_name="threat_score_v2")
+    )
+    alert_ids: Optional[List[UpdateThreatScoreV2Entry]] = field(
+        default=None, metadata=config(field_name="alert_ids")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class PasswordSprayDetailInput:
@@ -3097,6 +3123,12 @@ class Alert2:
     threat_score: Optional[float] = field(
         default=None, metadata=config(field_name="threat_score")
     )
+    threat_score_v2: Optional[float] = field(
+        default=None, metadata=config(field_name="threat_score_v2")
+    )
+    observation: Optional[bool] = field(
+        default=None, metadata=config(field_name="observation")
+    )
     metadata: Optional[AlertsMetadata] = field(
         default=None, metadata=config(field_name="metadata")
     )

taegis_sdk_python-1.2.32/taegis_sdk_python/services/alerts_history/__init__.py

@@ -0,0 +1,34 @@
+"""AlertsHistory Service."""
+
+from taegis_sdk_python.service_core import ServiceCore
+from taegis_sdk_python.services.alerts_history.mutations import (
+    TaegisSDKAlertsHistoryMutation,
+)
+from taegis_sdk_python.services.alerts_history.queries import (
+    TaegisSDKAlertsHistoryQuery,
+)
+from taegis_sdk_python.services.alerts_history.subscriptions import (
+    TaegisSDKAlertsHistorySubscription,
+)
+
+
+class AlertsHistoryService(ServiceCore):
+    """Taegis AlertsHistory Service."""
+
+    @property
+    def query(self):
+        if not self._queries:
+            self._queries = TaegisSDKAlertsHistoryQuery(self)
+        return self._queries
+
+    @property
+    def mutation(self):
+        if not self._mutations:
+            self._mutations = TaegisSDKAlertsHistoryMutation(self)
+        return self._mutations
+
+    @property
+    def subscription(self):
+        if not self._subscriptions:
+            self._subscriptions = TaegisSDKAlertsHistorySubscription(self)
+        return self._subscriptions

taegis_sdk_python-1.2.32/taegis_sdk_python/services/alerts_history/mutations.py

@@ -0,0 +1,33 @@
+"""AlertsHistory Mutation."""
+
+# pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
+
+
+# Autogenerated
+# DO NOT MODIFY
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
+
+from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.alerts_history.types import *
+from taegis_sdk_python.utils import (
+    build_output_string,
+    parse_union_result,
+    prepare_input,
+)
+
+if TYPE_CHECKING:  # pragma: no cover
+    from taegis_sdk_python.services.alerts_history import AlertsHistoryService
+
+log = logging.getLogger(__name__)
+
+
+class TaegisSDKAlertsHistoryMutation:
+    """Taegis Alerts_history Mutation operations."""
+
+    def __init__(self, service: AlertsHistoryService):
+        self.service = service

taegis_sdk_python-1.2.32/taegis_sdk_python/services/alerts_history/queries.py

@@ -0,0 +1,50 @@
+"""AlertsHistory Query."""
+
+# pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
+
+
+# Autogenerated
+# DO NOT MODIFY
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
+
+from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python.utils import (
+    build_output_string,
+    parse_union_result,
+    prepare_input,
+)
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.alerts_history.types import *
+
+if TYPE_CHECKING:  # pragma: no cover
+    from taegis_sdk_python.services.alerts_history import AlertsHistoryService
+
+log = logging.getLogger(__name__)
+
+
+class TaegisSDKAlertsHistoryQuery:
+    """Taegis Alerts_history Query operations."""
+
+    def __init__(self, service: AlertsHistoryService):
+        self.service = service
+
+    def alert_history_by_id(self, id_: str) -> List[AlertChangeRecord]:
+        """Returns all recorded versions of the alert with the given ID, sorted by modification date."""
+        endpoint = "alertHistoryByID"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "id": prepare_input(id_),
+            },
+            output=build_output_string(AlertChangeRecord),
+        )
+        if result.get(endpoint) is not None:
+            return AlertChangeRecord.schema().load(
+                [r or {} for r in result.get(endpoint)], many=True
+            )
+        raise GraphQLNoRowsInResultSetError("for query alertHistoryByID")

taegis_sdk_python-1.2.32/taegis_sdk_python/services/alerts_history/subscriptions.py

@@ -0,0 +1,33 @@
+"""AlertsHistory Subscription."""
+
+# pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
+
+
+# Autogenerated
+# DO NOT MODIFY
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
+
+from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.alerts_history.types import *
+from taegis_sdk_python.utils import (
+    build_output_string,
+    parse_union_result,
+    prepare_input,
+)
+
+if TYPE_CHECKING:  # pragma: no cover
+    from taegis_sdk_python.services.alerts_history import AlertsHistoryService
+
+log = logging.getLogger(__name__)
+
+
+class TaegisSDKAlertsHistorySubscription:
+    """Taegis Alerts_history Subscription operations."""
+
+    def __init__(self, service: AlertsHistoryService):
+        self.service = service