taegis-sdk-python 1.2.22__tar.gz → 1.2.23__tar.gz

{taegis_sdk_python-1.2.22/taegis_sdk_python.egg-info → taegis_sdk_python-1.2.23}/PKG-INFO

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

taegis_sdk_python-1.2.23/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python/services/__init__.py

@@ -48,6 +48,7 @@ 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
+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.preferences import PreferencesService
@@ -192,6 +193,7 @@ class GraphQLService:
         self._mitre_attack_info = None
         self._multi_tenant_context = None
         self._multi_tenant_ioc = None
+        self._nl_search = None
         self._notebooks = None
         self._notifications = None
         self._preferences = None
@@ -571,6 +573,13 @@ class GraphQLService:
             self._multi_tenant_ioc = MultiTenantIocService(self)
         return self._multi_tenant_ioc
 
+    @property
+    def nl_search(self):
+        """NlSearch Service Endpoint."""
+        if not self._nl_search:
+            self._nl_search = NlSearchService(self)
+        return self._nl_search
+
     @property
     def notebooks(self):
         """Notebooks Service Endpoint."""

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python/services/datasources/types.py

@@ -221,6 +221,9 @@ class LastSeenAsset:
     last_seen: Optional[int] = field(
         default=None, metadata=config(field_name="lastSeen")
     )
+    tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="tenantID")
+    )
     health: Optional[Union[HealthState, TaegisEnum]] = field(
         default=None,
         metadata=config(

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python/services/isensor/types.py

@@ -569,6 +569,9 @@ class IsensorDetail:
     tenant_id: Optional[str] = field(
         default=None, metadata=config(field_name="tenantId")
     )
+    is_partner: Optional[bool] = field(
+        default=None, metadata=config(field_name="isPartner")
+    )
     mode: Optional[Union[ISensorMode, TaegisEnum]] = field(
         default=None,
         metadata=config(

taegis_sdk_python-1.2.23/taegis_sdk_python/services/nl_search/__init__.py

@@ -0,0 +1,30 @@
+"""NlSearch Service."""
+
+from taegis_sdk_python.service_core import ServiceCore
+from taegis_sdk_python.services.nl_search.mutations import TaegisSDKNlSearchMutation
+from taegis_sdk_python.services.nl_search.queries import TaegisSDKNlSearchQuery
+from taegis_sdk_python.services.nl_search.subscriptions import (
+    TaegisSDKNlSearchSubscription,
+)
+
+
+class NlSearchService(ServiceCore):
+    """Taegis NlSearch Service."""
+
+    @property
+    def query(self):
+        if not self._queries:
+            self._queries = TaegisSDKNlSearchQuery(self)
+        return self._queries
+
+    @property
+    def mutation(self):
+        if not self._mutations:
+            self._mutations = TaegisSDKNlSearchMutation(self)
+        return self._mutations
+
+    @property
+    def subscription(self):
+        if not self._subscriptions:
+            self._subscriptions = TaegisSDKNlSearchSubscription(self)
+        return self._subscriptions

taegis_sdk_python-1.2.23/taegis_sdk_python/services/nl_search/mutations.py

@@ -0,0 +1,47 @@
+"""NlSearch 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.nl_search.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.nl_search import NlSearchService
+
+log = logging.getLogger(__name__)
+
+
+class TaegisSDKNlSearchMutation:
+    """Taegis Nl_search Mutation operations."""
+
+    def __init__(self, service: NlSearchService):
+        self.service = service
+
+    def nl_search_feedback(self, in_: NLSearchFeedback) -> NLSearchFeedbackOutput:
+        """nlSearchFeedback records user feedback on efficacy of the generated output."""
+        endpoint = "nlSearchFeedback"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "in": prepare_input(in_),
+            },
+            output=build_output_string(NLSearchFeedbackOutput),
+        )
+        if result.get(endpoint) is not None:
+            return NLSearchFeedbackOutput.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for mutation nlSearchFeedback")

taegis_sdk_python-1.2.23/taegis_sdk_python/services/nl_search/queries.py

@@ -0,0 +1,50 @@
+"""NlSearch 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._consts import TaegisEnum
+from taegis_sdk_python.services.nl_search.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.nl_search import NlSearchService
+
+
+log = logging.getLogger(__name__)
+
+
+class TaegisSDKNlSearchQuery:
+    """Taegis Nl_search Query operations."""
+
+    def __init__(self, service: NlSearchService):
+        self.service = service
+
+    def nl_search(self, in_: NLSearchInputs) -> List[NLSearchOutputs]:
+        """nlSearch takes a natural language query input and converts to QL as the output. If many results are returned as output, the best result is first at the top of the list.."""
+        endpoint = "nlSearch"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "in": prepare_input(in_),
+            },
+            output=build_output_string(NLSearchOutputs),
+        )
+        if result.get(endpoint) is not None:
+            return NLSearchOutputs.schema().load(
+                [r or {} for r in result.get(endpoint)], many=True
+            )
+        raise GraphQLNoRowsInResultSetError("for query nlSearch")

taegis_sdk_python-1.2.23/taegis_sdk_python/services/nl_search/subscriptions.py

@@ -0,0 +1,32 @@
+"""NlSearch 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.nl_search.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.nl_search import NlSearchService
+
+log = logging.getLogger(__name__)
+
+
+class TaegisSDKNlSearchSubscription:
+    """Taegis Nl_search Subscription operations."""
+
+    def __init__(self, service: NlSearchService):
+        self.service = service

taegis_sdk_python-1.2.23/taegis_sdk_python/services/nl_search/types.py

@@ -0,0 +1,89 @@
+"""NlSearch Types and Enums."""
+# pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
+
+# Autogenerated
+# DO NOT MODIFY
+
+from dataclasses import dataclass, field
+
+from enum import Enum
+
+from typing import Any, Dict, List, Optional, Tuple, Union
+
+from dataclasses_json import config, dataclass_json
+
+
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.utils import encode_enum, decode_enum
+
+
+class NLSearchStatus(str, Enum):
+    """NLSearchStatus."""
+
+    COMPLETE = "COMPLETE"
+    PARTIAL = "PARTIAL"
+    FAILED = "FAILED"
+
+
+class NLSearchRating(str, Enum):
+    """NLSearchRating."""
+
+    THUMBS_UP = "THUMBS_UP"
+    THUMBS_DOWN = "THUMBS_DOWN"
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class NLSearchInputs:
+    """NLSearchInputs."""
+
+    query: Optional[str] = field(default=None, metadata=config(field_name="query"))
+    limit: Optional[int] = field(default=None, metadata=config(field_name="limit"))
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class NLSearchFeedbackOutput:
+    """NLSearchFeedbackOutput."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class NLSearchOutputs:
+    """NLSearchOutputs."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+    ql: Optional[str] = field(default=None, metadata=config(field_name="ql"))
+    reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
+    status: Optional[Union[NLSearchStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(NLSearchStatus, x),
+            field_name="status",
+        ),
+    )
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class NLSearchFeedback:
+    """NLSearchFeedback."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+    modified: Optional[str] = field(
+        default=None, metadata=config(field_name="modified")
+    )
+    feedback_comment: Optional[str] = field(
+        default=None, metadata=config(field_name="feedbackComment")
+    )
+    rating: Optional[Union[NLSearchRating, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(NLSearchRating, x),
+            field_name="rating",
+        ),
+    )

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python/services/users/mutations.py

@@ -224,7 +224,9 @@ class TaegisSDKUsersMutation:
             return TDRUser.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation registerPartnerUser")
 
-    def forgot_password(self, email: str) -> ForgotPasswordResponse:
+    def forgot_password(
+        self, email: str, app: Optional[str] = None
+    ) -> ForgotPasswordResponse:
         """Initiate the password reset workflow requested by a user."""
         endpoint = "forgotPassword"
 
@@ -232,6 +234,7 @@ class TaegisSDKUsersMutation:
             endpoint=endpoint,
             variables={
                 "email": prepare_input(email),
+                "app": prepare_input(app),
             },
             output=build_output_string(ForgotPasswordResponse),
         )

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python/services/users/types.py

@@ -279,6 +279,9 @@ class MFAResetResponse:
     challenge_session: Optional[str] = field(
         default=None, metadata=config(field_name="challenge_session")
     )
+    challenge_name: Optional[str] = field(
+        default=None, metadata=config(field_name="challenge_name")
+    )
 
 
 @dataclass_json
@@ -327,8 +330,8 @@ class CognitoAuthEventPageDetails:
 
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
-class CognitoAuthEventType:
-    """CognitoAuthEventType."""
+class CognitoAuthChallengeType:
+    """CognitoAuthChallengeType."""
 
     challenge_name: Optional[str] = field(
         default=None, metadata=config(field_name="challengeName")
@@ -449,8 +452,11 @@ class CognitoAuthLogEntry:
     event_response: Optional[str] = field(
         default=None, metadata=config(field_name="eventResponse")
     )
-    auth_event_type: Optional[List[CognitoAuthEventType]] = field(
-        default=None, metadata=config(field_name="authEventType")
+    event_type: Optional[str] = field(
+        default=None, metadata=config(field_name="eventType")
+    )
+    auth_challenge_type: Optional[List[CognitoAuthChallengeType]] = field(
+        default=None, metadata=config(field_name="authChallengeType")
     )
 
 
@@ -540,6 +546,9 @@ class TDRUsersSearchInput:
     omit_role_assignments: Optional[bool] = field(
         default=None, metadata=config(field_name="omitRoleAssignments")
     )
+    read_from_writer: Optional[bool] = field(
+        default=None, metadata=config(field_name="readFromWriter")
+    )
     sort_by: Optional[Union[TDRUsersSearchSortBy, TaegisEnum]] = field(
         default=None,
         metadata=config(

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python/services/vdr/types.py

@@ -184,6 +184,19 @@ class VdrStatistics:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class VdrInspectScore:
+    """VdrInspectScore."""
+
+    step_base: Optional[float] = field(
+        default=None, metadata=config(field_name="stepBase")
+    )
+    step_final: Optional[float] = field(
+        default=None, metadata=config(field_name="stepFinal")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class VdrAffectedHost:
@@ -406,6 +419,36 @@ class VdrAssetsFiltersInputArgs:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class VdrMatchingVulnerabilityGroup:
+    """VdrMatchingVulnerabilityGroup."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+    classification: Optional[str] = field(
+        default=None, metadata=config(field_name="classification")
+    )
+    cvenumber: Optional[str] = field(
+        default=None, metadata=config(field_name="cvenumber")
+    )
+    description: Optional[str] = field(
+        default=None, metadata=config(field_name="description")
+    )
+    differentiator: Optional[str] = field(
+        default=None, metadata=config(field_name="differentiator")
+    )
+    host: Optional[str] = field(default=None, metadata=config(field_name="host"))
+    host_id: Optional[List[str]] = field(
+        default=None, metadata=config(field_name="hostId")
+    )
+    severity: Optional[str] = field(
+        default=None, metadata=config(field_name="severity")
+    )
+    scores: Optional[VdrInspectScore] = field(
+        default=None, metadata=config(field_name="scores")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class VdrDefinition:
@@ -572,9 +615,9 @@ class VdrInspectHost:
     definitions: Optional[List[VdrDefinition]] = field(
         default=None, metadata=config(field_name="definitions")
     )
-    matching_vulnerability_groups: Optional[List[VdrVulnerability]] = field(
-        default=None, metadata=config(field_name="matchingVulnerabilityGroups")
-    )
+    matching_vulnerability_groups: Optional[
+        List[VdrMatchingVulnerabilityGroup]
+    ] = field(default=None, metadata=config(field_name="matchingVulnerabilityGroups"))
 
 
 @dataclass_json

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23/taegis_sdk_python.egg-info}/PKG-INFO

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

{taegis_sdk_python-1.2.22 → taegis_sdk_python-1.2.23}/taegis_sdk_python.egg-info/SOURCES.txt

@@ -160,6 +160,11 @@ taegis_sdk_python/services/multi_tenant_ioc/mutations.py
 taegis_sdk_python/services/multi_tenant_ioc/queries.py
 taegis_sdk_python/services/multi_tenant_ioc/subscriptions.py
 taegis_sdk_python/services/multi_tenant_ioc/types.py
+taegis_sdk_python/services/nl_search/__init__.py
+taegis_sdk_python/services/nl_search/mutations.py
+taegis_sdk_python/services/nl_search/queries.py
+taegis_sdk_python/services/nl_search/subscriptions.py
+taegis_sdk_python/services/nl_search/types.py
 taegis_sdk_python/services/notebooks/__init__.py
 taegis_sdk_python/services/notebooks/mutations.py
 taegis_sdk_python/services/notebooks/queries.py

taegis_sdk_python-1.2.22/taegis_sdk_python/_version.py

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