taegis-sdk-python 1.4.2__tar.gz → 1.4.4__tar.gz

{taegis_sdk_python-1.4.2/taegis_sdk_python.egg-info → taegis_sdk_python-1.4.4}/PKG-INFO

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

taegis_sdk_python-1.4.4/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/__init__.py

@@ -2,7 +2,6 @@
 Taegis SDK Python GraphQL Service manager.
 """
 
-
 import logging
 import threading
 from ssl import SSLContext
@@ -147,8 +146,7 @@ class GraphQLService:
         self._use_universal_authentication = use_universal_authentication
 
         if self._environment not in self._environments:
-            raise ValueError(
-                f"environment must be in {self._environments.keys()}")
+            raise ValueError(f"environment must be in {self._environments.keys()}")
 
         self._tenant_id = tenant_id
         self._gateway = gateway or "/graphql"
@@ -538,16 +536,14 @@ class GraphQLService:
     def endpoint_command_manager(self):
         """Endpoint Command Manager Service Endpoint."""
         if not self._endpoint_command_manager:
-            self._endpoint_command_manager = EndpointCommandManagerService(
-                self)
+            self._endpoint_command_manager = EndpointCommandManagerService(self)
         return self._endpoint_command_manager
 
     @property
     def endpoint_management_service(self):
         """Endpoint Management Service Endpoint."""
         if not self._endpoint_management_service:
-            self._endpoint_management_service = EndpointManagementServiceService(
-                self)
+            self._endpoint_management_service = EndpointManagementServiceService(self)
         return self._endpoint_management_service
 
     @property

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/alerts/types.py

@@ -905,6 +905,12 @@ class EntityHost:
     )
     os: Optional[str] = field(default=None, metadata=config(field_name="os"))
     os_arch: Optional[str] = field(default=None, metadata=config(field_name="os_arch"))
+    sensor_id: Optional[str] = field(
+        default=None, metadata=config(field_name="sensor_id")
+    )
+    sensor_type: Optional[str] = field(
+        default=None, metadata=config(field_name="sensor_type")
+    )
     vendor_agent_device_id: Optional[str] = field(
         default=None, metadata=config(field_name="vendor_agent_device_id")
     )

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/authz/queries.py

@@ -342,6 +342,17 @@ class TaegisSDKAuthzQuery:
             )
         raise GraphQLNoRowsInResultSetError("for query authzSupportedFeatures")
 
+    def authz_subject_central_role(self) -> AuthzRole:
+        """Identify the equivalent central role for the requesting subject in the current tenant context."""
+        endpoint = "authzSubjectCentralRole"
+
+        result = self.service.execute_query(
+            endpoint=endpoint, variables={}, output=build_output_string(AuthzRole)
+        )
+        if result.get(endpoint) is not None:
+            return AuthzRole.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query authzSubjectCentralRole")
+
     def authz_can_user_assume_tenant(self) -> bool:
         """DEPRECATED: Please use authzCanSubjectAssumeTenant."""
         endpoint = "authzCanUserAssumeTenant"

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/clients/mutations.py

@@ -32,7 +32,12 @@ class TaegisSDKClientsMutation:
     def __init__(self, service: ClientsService):
         self.service = service
 
-    def create_client(self, name: str, roles: Optional[List[str]] = None) -> NewClient:
+    def create_client(
+        self,
+        name: str,
+        roles: Optional[List[str]] = None,
+        input_: Optional[CreateClientInput] = None,
+    ) -> NewClient:
         """Create a new client (SCWX (tenant 5000) clients are disallowed). The new client will be assigned the TenantAnalyst role if the `roles` parameter is omitted."""
         endpoint = "createClient"
 
@@ -41,6 +46,7 @@ class TaegisSDKClientsMutation:
             variables={
                 "name": prepare_input(name),
                 "roles": prepare_input(roles),
+                "input": prepare_input(input_),
             },
             output=build_output_string(NewClient),
         )

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/clients/types.py

@@ -93,6 +93,26 @@ class ClientRoleAssignmentInput:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class CreateClientInput:
+    """CreateClientInput."""
+
+    owner_tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="ownerTenantID")
+    )
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class CreateClientInternalInput:
+    """CreateClientInternalInput."""
+
+    owner_tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="ownerTenantID")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class CredentialValidationResult:
@@ -134,6 +154,9 @@ class Client:
     tenant_id: Optional[str] = field(
         default=None, metadata=config(field_name="tenantID")
     )
+    owner_tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="ownerTenantID")
+    )
     created_at: Optional[str] = field(
         default=None,
         metadata=config(

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/cql_metadata/__init__.py

@@ -1,5 +1,9 @@
 """"CqlMetadata Service."""
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from taegis_sdk_python.service_core import ServiceCore
 from taegis_sdk_python.services.cql_metadata.mutations import (
     TaegisSDKCqlMetadataMutation,
@@ -9,10 +13,19 @@ from taegis_sdk_python.services.cql_metadata.subscriptions import (
     TaegisSDKCqlMetadataSubscription,
 )
 
+if TYPE_CHECKING:  # pragma: no cover
+    from taegis_sdk_python.services import GraphQLService
+
 
 class CqlMetadataService(ServiceCore):
     """Taegis CqlMetadata Service."""
 
+    def __init__(self, service: GraphQLService):
+        """Initialize the CqlMetadata Service."""
+        super().__init__(service)
+        self._gateway = "/cql-metadata/query"
+        self._input_value_deprecation = False
+
     @property
     def query(self):
         if not self._queries:

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/process_trees/types.py

@@ -69,6 +69,9 @@ class ProcessEvent:
     user_is_admin: Optional[bool] = field(
         default=None, metadata=config(field_name="userIsAdmin")
     )
+    was_blocked: Optional[bool] = field(
+        default=None, metadata=config(field_name="wasBlocked")
+    )
     normalizer: Optional[str] = field(
         default=None, metadata=config(field_name="normalizer")
     )

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/tenants/mutations.py

@@ -486,3 +486,18 @@ class TaegisSDKTenantsMutation:
         if result.get(endpoint) is not None:
             return RegisteredDomain.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation verifyRegisteredDomain")
+
+    def delete_registered_domain(self, domain: str) -> RegisteredDomain:
+        """Delete an existing registered domain from the tenant."""
+        endpoint = "deleteRegisteredDomain"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "domain": prepare_input(domain),
+            },
+            output=build_output_string(RegisteredDomain),
+        )
+        if result.get(endpoint) is not None:
+            return RegisteredDomain.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for mutation deleteRegisteredDomain")

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/tenants/types.py

@@ -46,6 +46,7 @@ class AuthzAction(str, Enum):
     APPROVE = "Approve"
     EXECUTE = "Execute"
     REGISTER_DOMAIN = "RegisterDomain"
+    DELETE_REGISTERED_DOMAIN = "DeleteRegisteredDomain"
 
 
 class TenantType(str, Enum):
@@ -150,6 +151,9 @@ class AuditAction(str, Enum):
     CREATE_SERVICE = "create_service"
     DELETE_SERVICE = "delete_service"
     UPDATE_SERVICE = "update_service"
+    ADD_REGISTERED_DOMAIN = "add_registered_domain"
+    VERIFY_REGISTERED_DOMAIN = "verify_registered_domain"
+    DELETE_REGISTERED_DOMAIN = "delete_registered_domain"
 
 
 class TenantDecommissionRequestStatus(str, Enum):

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/tenants4/types.py

@@ -51,6 +51,36 @@ class OrderDir(str, Enum):
     DESC = "DESC"
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class MappedTenant:
+    """MappedTenant."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+    endpoint: Optional[str] = field(
+        default=None, metadata=config(field_name="endpoint")
+    )
+    region: Optional[str] = field(default=None, metadata=config(field_name="region"))
+    data_region: Optional[str] = field(
+        default=None, metadata=config(field_name="dataRegion")
+    )
+    account_type: Optional[str] = field(
+        default=None, metadata=config(field_name="accountType")
+    )
+    account_origin: Optional[str] = field(
+        default=None, metadata=config(field_name="accountOrigin")
+    )
+    authz_ownership: Optional[str] = field(
+        default=None, metadata=config(field_name="authzOwnership")
+    )
+    xdr_ownership: Optional[str] = field(
+        default=None, metadata=config(field_name="xdrOwnership")
+    )
+    last_refresh: Optional[str] = field(
+        default=None, metadata=config(field_name="lastRefresh")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class TenantLabel:
@@ -133,6 +163,72 @@ class TenantSubscriptions:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class TenantV4:
+    """TenantV4."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+    parent_tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="parentTenantID")
+    )
+    partner_tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="partnerTenantID")
+    )
+    partner_name: Optional[str] = field(
+        default=None, metadata=config(field_name="partnerName")
+    )
+    organization_tenant_id: Optional[str] = field(
+        default=None, metadata=config(field_name="organizationTenantID")
+    )
+    organization_name: Optional[str] = field(
+        default=None, metadata=config(field_name="organizationName")
+    )
+    name: Optional[str] = field(default=None, metadata=config(field_name="name"))
+    enabled: Optional[bool] = field(default=None, metadata=config(field_name="enabled"))
+    hierarchy_path: Optional[str] = field(
+        default=None, metadata=config(field_name="hierarchyPath")
+    )
+    is_organization: Optional[bool] = field(
+        default=None, metadata=config(field_name="isOrganization")
+    )
+    is_partner: Optional[bool] = field(
+        default=None, metadata=config(field_name="isPartner")
+    )
+    has_support_enabled: Optional[bool] = field(
+        default=None, metadata=config(field_name="hasSupportEnabled")
+    )
+    support_inherited: Optional[bool] = field(
+        default=None, metadata=config(field_name="supportInherited")
+    )
+    expires_at: Optional[str] = field(
+        default=None, metadata=config(field_name="expiresAt")
+    )
+    created_at: Optional[str] = field(
+        default=None, metadata=config(field_name="createdAt")
+    )
+    updated_at: Optional[str] = field(
+        default=None, metadata=config(field_name="updatedAt")
+    )
+    enabled_at_regions: Optional[List[Union[TenantRegion, TaegisEnum]]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(TenantRegion, x),
+            field_name="enabledAtRegions",
+        ),
+    )
+    labels: Optional[List[TenantLabel]] = field(
+        default=None, metadata=config(field_name="labels")
+    )
+    subscriptions: Optional[TenantSubscriptions] = field(
+        default=None, metadata=config(field_name="subscriptions")
+    )
+    central_tenant: Optional[MappedTenant] = field(
+        default=None, metadata=config(field_name="centralTenant")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class TenantsQuery:
@@ -207,69 +303,6 @@ class TenantsQuery:
     )
 
 
-@dataclass_json
-@dataclass(order=True, eq=True, frozen=True)
-class TenantV4:
-    """TenantV4."""
-
-    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
-    parent_tenant_id: Optional[str] = field(
-        default=None, metadata=config(field_name="parentTenantID")
-    )
-    partner_tenant_id: Optional[str] = field(
-        default=None, metadata=config(field_name="partnerTenantID")
-    )
-    partner_name: Optional[str] = field(
-        default=None, metadata=config(field_name="partnerName")
-    )
-    organization_tenant_id: Optional[str] = field(
-        default=None, metadata=config(field_name="organizationTenantID")
-    )
-    organization_name: Optional[str] = field(
-        default=None, metadata=config(field_name="organizationName")
-    )
-    name: Optional[str] = field(default=None, metadata=config(field_name="name"))
-    enabled: Optional[bool] = field(default=None, metadata=config(field_name="enabled"))
-    hierarchy_path: Optional[str] = field(
-        default=None, metadata=config(field_name="hierarchyPath")
-    )
-    is_organization: Optional[bool] = field(
-        default=None, metadata=config(field_name="isOrganization")
-    )
-    is_partner: Optional[bool] = field(
-        default=None, metadata=config(field_name="isPartner")
-    )
-    has_support_enabled: Optional[bool] = field(
-        default=None, metadata=config(field_name="hasSupportEnabled")
-    )
-    support_inherited: Optional[bool] = field(
-        default=None, metadata=config(field_name="supportInherited")
-    )
-    expires_at: Optional[str] = field(
-        default=None, metadata=config(field_name="expiresAt")
-    )
-    created_at: Optional[str] = field(
-        default=None, metadata=config(field_name="createdAt")
-    )
-    updated_at: Optional[str] = field(
-        default=None, metadata=config(field_name="updatedAt")
-    )
-    enabled_at_regions: Optional[List[Union[TenantRegion, TaegisEnum]]] = field(
-        default=None,
-        metadata=config(
-            encoder=encode_enum,
-            decoder=lambda x: decode_enum(TenantRegion, x),
-            field_name="enabledAtRegions",
-        ),
-    )
-    labels: Optional[List[TenantLabel]] = field(
-        default=None, metadata=config(field_name="labels")
-    )
-    subscriptions: Optional[TenantSubscriptions] = field(
-        default=None, metadata=config(field_name="subscriptions")
-    )
-
-
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class TenantResults:

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4}/taegis_sdk_python/services/threat/queries.py

@@ -366,7 +366,6 @@ class TaegisSDKThreatQuery:
         self,
         object_type: Union[ThreatFacetObject, TaegisEnum],
         filters: Optional[ThreatFilter] = None,
-        facets: Optional[List[str]] = None,
         facet_objs: Optional[List[str]] = None,
     ) -> List[FacetInfo]:
         """Retrieves facet count based on object type and filters."""
@@ -377,7 +376,6 @@ class TaegisSDKThreatQuery:
             variables={
                 "objectType": prepare_input(object_type),
                 "filters": prepare_input(filters),
-                "facets": prepare_input(facets),
                 "facetObjs": prepare_input(facet_objs),
             },
             output=build_output_string(FacetInfo),

{taegis_sdk_python-1.4.2 → taegis_sdk_python-1.4.4/taegis_sdk_python.egg-info}/PKG-INFO

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

taegis_sdk_python-1.4.2/taegis_sdk_python/_version.py

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