taegis-sdk-python 1.6.6__tar.gz → 1.6.8__tar.gz

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taegis-sdk-python
-Version: 1.6.6
+Version: 1.6.8
 Summary: Taegis Python SDK
 Project-URL: Homepage, https://github.com/secureworks/taegis-sdk-python
 Project-URL: Bug Reports, https://github.com/secureworks/taegis-sdk-python/issues

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/_consts.py

@@ -34,6 +34,8 @@ TAEGIS_TENANTS_URLS = {
 UNIVERSAL_ENVIRONMENT = "universal"
 UNIVERSAL_AUTHENTICATION_URL = "https://api.taegis.secureworks.com"
 
+DEFAULT_GATEWAY = "/graphql"
+
 
 class TaegisEnum(str, Enum):
     """Handler for Unknown Enums."""

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/_version.py

@@ -1,3 +1,3 @@
 """Version idenitier."""
 
-__version__ = "1.6.6"
+__version__ = "1.6.8"

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/middlewares/utils.py

@@ -22,7 +22,7 @@ def scrub_dict(d: Mapping):
     """Remove sensitive data from dictionary keys and values."""
     access_tokens = get_access_tokens()
 
-    if "authorization" in d:
+    if "Authorization" in d:
         token = d["Authorization"].replace("Bearer ", "")
         access_tokens.append(token)
 

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/service_core.py

@@ -16,6 +16,7 @@ from gql.transport.aiohttp import AIOHTTPTransport
 from gql.transport.aiohttp_websockets import AIOHTTPWebsocketsTransport
 from graphql import GraphQLError, GraphQLField, GraphQLSchema
 
+from taegis_sdk_python._consts import DEFAULT_GATEWAY
 from taegis_sdk_python._version import __version__
 from taegis_sdk_python.errors import InvalidGraphQLEndpoint
 from taegis_sdk_python.utils import async_block, prepare_variables, remove_node
@@ -167,7 +168,7 @@ class ServiceCore:
         self.service = service
 
         self._urls = self.service._environments
-        self._gateway = self.service._gateway
+        self._gateway = None
         self._input_value_deprecation = True
 
         self._cache_map = SchemaCacheMap(expires=self.service.schema_expiry)
@@ -181,6 +182,11 @@ class ServiceCore:
         """Syncronous URL."""
         return self.service.url or self._urls.get(self.service.environment)
 
+    @property
+    def exclude_deprecated_output(self):
+        """GraphQL Query Deprecated Output Fields in Schema."""
+        return self.service.exclude_deprecated_output
+
     @property
     def wss_url(self) -> str:
         """WebSockets URL."""
@@ -189,7 +195,7 @@ class ServiceCore:
     @property
     def gateway(self) -> str:
         """GraphQL Gateway"""
-        return self.service.gateway or self._gateway
+        return self.service.gateway or self._gateway or DEFAULT_GATEWAY
 
     @property
     def query(self):  # pragma: no cover

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/__init__.py

@@ -6,7 +6,7 @@ import logging
 import threading
 from collections.abc import Mapping
 from ssl import SSLContext
-from typing import Any, Dict, Literal, Optional, Union
+from typing import Any, Callable, Dict, Literal, Optional, Tuple, Union
 
 import aiohttp
 from aiohttp.client_reqrep import Fingerprint
@@ -45,6 +45,7 @@ from taegis_sdk_python.services.endpoint_management_service import (
     EndpointManagementServiceService,
 )
 from taegis_sdk_python.services.entity_context import EntityContextService
+from taegis_sdk_python.services.escalation_policies import EscalationPoliciesService
 from taegis_sdk_python.services.event_search import EventSearchService
 from taegis_sdk_python.services.events import EventsService
 from taegis_sdk_python.services.exports import ExportsService
@@ -60,6 +61,7 @@ from taegis_sdk_python.services.multi_tenant_ioc import MultiTenantIocService
 from taegis_sdk_python.services.nl_search import NlSearchService
 from taegis_sdk_python.services.notebooks import NotebooksService
 from taegis_sdk_python.services.notifications import NotificationsService
+from taegis_sdk_python.services.notifications2 import Notifications2Service
 from taegis_sdk_python.services.preferences import PreferencesService
 from taegis_sdk_python.services.process_trees import ProcessTreesService
 from taegis_sdk_python.services.queries import QueriesService
@@ -103,8 +105,9 @@ class GraphQLService:
         execute_timeout: Optional[Union[int, float]] = 300,
         max_message_size: int = 0,
         use_universal_authentication: bool = False,
-        middlewares=None,
-        reuse_request_id=False,
+        middlewares: Optional[Tuple[Callable]] = None,
+        reuse_request_id: bool = False,
+        exclude_deprecated_output: bool = True,
     ):  # pylint: disable=too-many-statements
         """
         GraphQLService
@@ -157,7 +160,7 @@ class GraphQLService:
             raise ValueError(f"environment must be in {self._environments.keys()}")
 
         self._tenant_id = tenant_id
-        self._gateway = gateway or "/graphql"
+        self._gateway = gateway
         self._thread_id = threading.get_ident()
         self._context_kwargs = {}
         if not extra_headers:
@@ -184,6 +187,8 @@ class GraphQLService:
         self._middlewares = middlewares or ()
         self._reuse_request_id = reuse_request_id
 
+        self._exclude_deprecated_output = exclude_deprecated_output
+
         self._access_points = None
         self._agent = None
         self._alerts = None
@@ -204,6 +209,7 @@ class GraphQLService:
         self._endpoint_management_service = None
         self._entity_context = None
         self._event_search = None
+        self._escalation_policies = None
         self._events = None
         self._exports = None
         self._fast_ioc = None
@@ -218,6 +224,7 @@ class GraphQLService:
         self._nl_search = None
         self._notebooks = None
         self._notifications = None
+        self._notifications2 = None
         self._preferences = None
         self._process_trees = None
         self._queries = None
@@ -343,7 +350,7 @@ class GraphQLService:
     @property
     def gateway(self):
         """Taegis GraphQL Gateway."""
-        return self._context_manager.get("gateway")
+        return self._context_manager.get("gateway", self._gateway)
 
     @property
     def tenant_id(self):
@@ -420,6 +427,13 @@ class GraphQLService:
         """Max Message Size for Subscriptions."""
         return self._context_manager.get("max_message_size", self._max_message_size)
 
+    @property
+    def exclude_deprecated_output(self):
+        """GraphQL Query Deprecated Output Fields in Schema."""
+        return self._context_manager.get(
+            "exclude_deprecated_output", self._exclude_deprecated_output
+        )
+
     @property
     def input_value_deprecation(self):
         """GraphQL Query Deprecated Input Fields in Schema."""
@@ -613,6 +627,13 @@ class GraphQLService:
             self._entity_context = EntityContextService(self)
         return self._entity_context
 
+    @property
+    def escalation_policies(self):
+        """Escalation Policies Service Endpoint."""
+        if not self._escalation_policies:
+            self._escalation_policies = EscalationPoliciesService(self)
+        return self._escalation_policies
+
     @property
     def event_search(self):
         """Events Search Service Endpoint."""
@@ -718,6 +739,13 @@ class GraphQLService:
             self._notifications = NotificationsService(self)
         return self._notifications
 
+    @property
+    def notifications2(self):
+        """Notifications2 Service Endpoint."""
+        if not self._notifications2:
+            self._notifications2 = Notifications2Service(self)
+        return self._notifications2
+
     @property
     def preferences(self):
         """Preferences Service Endpoint."""

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/access_points/mutations.py

@@ -41,7 +41,10 @@ class TaegisSDKAccessPointsMutation:
             variables={
                 "principal": prepare_input(principal),
             },
-            output=build_output_string(AccessPoint),
+            output=build_output_string(
+                AccessPoint,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AccessPoint.from_dict(result.get(endpoint))
@@ -56,7 +59,10 @@ class TaegisSDKAccessPointsMutation:
             variables={
                 "principal": prepare_input(principal),
             },
-            output=build_output_string(AccessPoint),
+            output=build_output_string(
+                AccessPoint,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AccessPoint.from_dict(result.get(endpoint))
@@ -67,7 +73,12 @@ class TaegisSDKAccessPointsMutation:
         endpoint = "deleteAccessPoint"
 
         result = self.service.execute_mutation(
-            endpoint=endpoint, variables={}, output=build_output_string(AccessPoint)
+            endpoint=endpoint,
+            variables={},
+            output=build_output_string(
+                AccessPoint,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AccessPoint.from_dict(result.get(endpoint))

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/access_points/queries.py

@@ -37,7 +37,12 @@ class TaegisSDKAccessPointsQuery:
         endpoint = "getAccessPoint"
 
         result = self.service.execute_query(
-            endpoint=endpoint, variables={}, output=build_output_string(AccessPoint)
+            endpoint=endpoint,
+            variables={},
+            output=build_output_string(
+                AccessPoint,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AccessPoint.from_dict(result.get(endpoint))
@@ -50,7 +55,10 @@ class TaegisSDKAccessPointsQuery:
         result = self.service.execute_query(
             endpoint=endpoint,
             variables={},
-            output=build_output_string(AccessPointCloudFormation),
+            output=build_output_string(
+                AccessPointCloudFormation,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AccessPointCloudFormation.from_dict(result.get(endpoint))
@@ -63,7 +71,10 @@ class TaegisSDKAccessPointsQuery:
         result = self.service.execute_query(
             endpoint=endpoint,
             variables={},
-            output=build_output_string(AccessPointPrefixes),
+            output=build_output_string(
+                AccessPointPrefixes,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AccessPointPrefixes.from_dict(result.get(endpoint))

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/agent/queries.py

@@ -43,7 +43,10 @@ class TaegisSDKAgentQuery:
             variables={
                 "args": prepare_input(args),
             },
-            output=build_output_string(Package),
+            output=build_output_string(
+                Package,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return Package.schema().load(
@@ -62,7 +65,10 @@ class TaegisSDKAgentQuery:
             variables={
                 "args": prepare_input(args),
             },
-            output=build_output_string(PackageSignedUrl),
+            output=build_output_string(
+                PackageSignedUrl,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return PackageSignedUrl.from_dict(result.get(endpoint))
@@ -77,7 +83,10 @@ class TaegisSDKAgentQuery:
             variables={
                 "id": prepare_input(id_),
             },
-            output=build_output_string(PackageSignedUrl),
+            output=build_output_string(
+                PackageSignedUrl,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return PackageSignedUrl.from_dict(result.get(endpoint))
@@ -90,7 +99,10 @@ class TaegisSDKAgentQuery:
         result = self.service.execute_query(
             endpoint=endpoint,
             variables={},
-            output=build_output_string(ReleaseChannelsConfig),
+            output=build_output_string(
+                ReleaseChannelsConfig,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return ReleaseChannelsConfig.schema().load(

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/alerts/mutations.py

@@ -43,7 +43,10 @@ class TaegisSDKAlertsMutation:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(UpdateInvestigationResponse),
+            output=build_output_string(
+                UpdateInvestigationResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return UpdateInvestigationResponse.from_dict(result.get(endpoint))
@@ -62,7 +65,10 @@ class TaegisSDKAlertsMutation:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(UpdateResolutionResponse),
+            output=build_output_string(
+                UpdateResolutionResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return UpdateResolutionResponse.from_dict(result.get(endpoint))
@@ -81,7 +87,10 @@ class TaegisSDKAlertsMutation:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(BulkInvestigationsResponse),
+            output=build_output_string(
+                BulkInvestigationsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return BulkInvestigationsResponse.from_dict(result.get(endpoint))
@@ -104,7 +113,10 @@ class TaegisSDKAlertsMutation:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(EvictResponse),
+            output=build_output_string(
+                EvictResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return EvictResponse.from_dict(result.get(endpoint))
@@ -121,7 +133,10 @@ class TaegisSDKAlertsMutation:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(UpdateThreatScoreResponse),
+            output=build_output_string(
+                UpdateThreatScoreResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return UpdateThreatScoreResponse.from_dict(result.get(endpoint))
@@ -140,7 +155,10 @@ class TaegisSDKAlertsMutation:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(UpdateThreatScoreResponse),
+            output=build_output_string(
+                UpdateThreatScoreResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return UpdateThreatScoreResponse.from_dict(result.get(endpoint))

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/alerts/queries.py

@@ -43,7 +43,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsResponse),
+            output=build_output_string(
+                AlertsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsResponse.from_dict(result.get(endpoint))
@@ -60,7 +63,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsResponse),
+            output=build_output_string(
+                AlertsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsResponse.from_dict(result.get(endpoint))
@@ -79,7 +85,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsResponse),
+            output=build_output_string(
+                AlertsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsResponse.from_dict(result.get(endpoint))
@@ -98,7 +107,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsResponse),
+            output=build_output_string(
+                AlertsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsResponse.from_dict(result.get(endpoint))
@@ -117,7 +129,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsCountByTenantResponse),
+            output=build_output_string(
+                AlertsCountByTenantResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsCountByTenantResponse.from_dict(result.get(endpoint))
@@ -134,7 +149,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsResponse),
+            output=build_output_string(
+                AlertsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsResponse.from_dict(result.get(endpoint))
@@ -151,7 +169,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsResponse),
+            output=build_output_string(
+                AlertsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsResponse.from_dict(result.get(endpoint))
@@ -168,7 +189,10 @@ class TaegisSDKAlertsQuery:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(AlertsAggregateResponse),
+            output=build_output_string(
+                AlertsAggregateResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertsAggregateResponse.from_dict(result.get(endpoint))

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/alerts/subscriptions.py

@@ -43,7 +43,10 @@ class TaegisSDKAlertsSubscription:
             variables={
                 "in": prepare_input(in_),
             },
-            output=build_output_string(BulkResolutionResponse),
+            output=build_output_string(
+                BulkResolutionResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if any(r.get(endpoint) for r in result):
             return BulkResolutionResponse.schema().load(

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/alerts_history/queries.py

@@ -41,7 +41,10 @@ class TaegisSDKAlertsHistoryQuery:
             variables={
                 "id": prepare_input(id_),
             },
-            output=build_output_string(AlertChangeRecord),
+            output=build_output_string(
+                AlertChangeRecord,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
         )
         if result.get(endpoint) is not None:
             return AlertChangeRecord.schema().load(

{taegis_sdk_python-1.6.6 → taegis_sdk_python-1.6.8}/taegis_sdk_python/services/assets/mutations.py

@@ -42,7 +42,9 @@ class TaegisSDKAssetsMutation:
                 "id": prepare_input(id_),
                 "reason": prepare_input(reason),
             },
-            output=build_output_string(Asset),
+            output=build_output_string(
+                Asset, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Asset.from_dict(result.get(endpoint))
@@ -58,7 +60,9 @@ class TaegisSDKAssetsMutation:
                 "id": prepare_input(id_),
                 "reason": prepare_input(reason),
             },
-            output=build_output_string(Asset),
+            output=build_output_string(
+                Asset, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Asset.from_dict(result.get(endpoint))
@@ -98,7 +102,9 @@ class TaegisSDKAssetsMutation:
                 "hostId": prepare_input(host_id),
                 "tag": prepare_input(tag),
             },
-            output=build_output_string(Tag),
+            output=build_output_string(
+                Tag, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Tag.from_dict(result.get(endpoint))
@@ -118,7 +124,9 @@ class TaegisSDKAssetsMutation:
                 "id": prepare_input(id_),
                 "tag": prepare_input(tag),
             },
-            output=build_output_string(Tag),
+            output=build_output_string(
+                Tag, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Tag.from_dict(result.get(endpoint))
@@ -137,7 +145,9 @@ class TaegisSDKAssetsMutation:
             variables={
                 "id": prepare_input(id_),
             },
-            output=build_output_string(Tag),
+            output=build_output_string(
+                Tag, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Tag.from_dict(result.get(endpoint))
@@ -156,7 +166,9 @@ class TaegisSDKAssetsMutation:
             variables={
                 "assetInput": prepare_input(asset_input),
             },
-            output=build_output_string(Asset),
+            output=build_output_string(
+                Asset, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Asset.from_dict(result.get(endpoint))
@@ -182,7 +194,9 @@ class TaegisSDKAssetsMutation:
                 "assetIds": prepare_input(asset_ids),
                 "investigationId": prepare_input(investigation_id),
             },
-            output=build_output_string(Asset),
+            output=build_output_string(
+                Asset, exclude_deprecated_output=self.service.exclude_deprecated_output
+            ),
         )
         if result.get(endpoint) is not None:
             return Asset.schema().load(