taegis-sdk-python 1.6.9__tar.gz → 1.6.9.2__tar.gz

{taegis_sdk_python-1.6.9 → taegis_sdk_python-1.6.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taegis-sdk-python
-Version: 1.6.9
+Version: 1.6.9.2
 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.9 → taegis_sdk_python-1.6.9.2}/taegis_sdk_python/_version.py

@@ -1,3 +1,3 @@
 """Version idenitier."""
 
-__version__ = "1.6.9"
+__version__ = "1.6.9.2"

{taegis_sdk_python-1.6.9 → taegis_sdk_python-1.6.9.2}/taegis_sdk_python/services/authz/queries.py

@@ -201,7 +201,7 @@ class TaegisSDKAuthzQuery:
     def authz_check_tenants(
         self, input_: AuthzObjectActionInput, tenants: List[str]
     ) -> List[str]:
-        """Given a list of tenants and a permission, returns the IDs of tenants where the subject posesses the given permission."""
+        """Given a list of tenants and a permission, returns the IDs of tenants where the subject possesses the given permission."""
         endpoint = "authzCheckTenants"
 
         result = self.service.execute_query(
@@ -216,6 +216,26 @@ class TaegisSDKAuthzQuery:
             return result.get(endpoint)
         raise GraphQLNoRowsInResultSetError("for query authzCheckTenants")
 
+    def authz_permitted_tenants(
+        self, input_: AuthzPermittedTenantsInput
+    ) -> AuthzPermittedTenantsResponse:
+        """Returns all tenant IDs where the Central-native subject possesses the given permissions."""
+        endpoint = "authzPermittedTenants"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(
+                AuthzPermittedTenantsResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
+        )
+        if result.get(endpoint) is not None:
+            return AuthzPermittedTenantsResponse.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query authzPermittedTenants")
+
     def authz_permissions(self) -> List[AuthzPermission]:
         """Retrieve the full list of subjects permissions for the tenant context."""
         endpoint = "authzPermissions"

{taegis_sdk_python-1.6.9 → taegis_sdk_python-1.6.9.2}/taegis_sdk_python/services/authz/types.py

@@ -22,6 +22,13 @@ class Effect(str, Enum):
     DENY = "DENY"
 
 
+class PermissionOperator(str, Enum):
+    """PermissionOperator."""
+
+    AND = "AND"
+    OR = "OR"
+
+
 class AuthzRequestPermissionObject(str, Enum):
     """AuthzRequestPermissionObject."""
 
@@ -97,6 +104,20 @@ class AuthzActionTuple:
     effect: Optional[bool] = field(default=None, metadata=config(field_name="effect"))
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class AuthzPermittedTenantsResponse:
+    """AuthzPermittedTenantsResponse."""
+
+    total_count: Optional[int] = field(
+        default=None, metadata=config(field_name="totalCount")
+    )
+    count: Optional[int] = field(default=None, metadata=config(field_name="count"))
+    tenant_ids: Optional[List[str]] = field(
+        default=None, metadata=config(field_name="tenantIDs")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class TenantHierarchy:
@@ -296,6 +317,26 @@ class AuthzPolicy:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class AuthzPermittedTenantsInput:
+    """AuthzPermittedTenantsInput."""
+
+    count: Optional[int] = field(default=None, metadata=config(field_name="count"))
+    page: Optional[int] = field(default=None, metadata=config(field_name="page"))
+    permissions: Optional[List[AuthzObjectActionInput]] = field(
+        default=None, metadata=config(field_name="permissions")
+    )
+    operator: Optional[Union[PermissionOperator, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(PermissionOperator, x),
+            field_name="operator",
+        ),
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class AuthzCustomRoleUpdateInput:

{taegis_sdk_python-1.6.9 → taegis_sdk_python-1.6.9.2}/taegis_sdk_python/services/collector/queries.py

@@ -643,6 +643,15 @@ class TaegisSDKCollectorQuery:
             return Deployment.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query getRoleDeployment")
 
+    def get_all_roles(self) -> List[str]:
+        """Get all roles in the environment.."""
+        endpoint = "getAllRoles"
+
+        result = self.service.execute_query(endpoint=endpoint, variables={}, output="")
+        if result.get(endpoint) is not None:
+            return result.get(endpoint)
+        raise GraphQLNoRowsInResultSetError("for query getAllRoles")
+
     def get_all_collectors_overview(
         self, role: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> List[CollectorOverview]:

{taegis_sdk_python-1.6.9 → taegis_sdk_python-1.6.9.2}/taegis_sdk_python/services/tenants4/types.py

@@ -119,6 +119,28 @@ class CachedEntryType(str, Enum):
     ALL_LICENSES_FOR_TENANT = "ALL_LICENSES_FOR_TENANT"
 
 
+class AccountOrigin(str, Enum):
+    """AccountOrigin."""
+
+    CENTRAL = "CENTRAL"
+    TAEGIS = "TAEGIS"
+
+
+class AuthzProvider(str, Enum):
+    """AuthzProvider."""
+
+    CENTRAL = "CENTRAL"
+    TAEGIS = "TAEGIS"
+
+
+class XDRProvider(str, Enum):
+    """XDRProvider."""
+
+    CENTRAL = "CENTRAL"
+    TAEGIS = "TAEGIS"
+    SECURITY_OPERATIONS = "SECURITY_OPERATIONS"
+
+
 class RetentionMonths(str, Enum):
     """RetentionMonths."""
 
@@ -134,6 +156,17 @@ class RetentionMonths(str, Enum):
     SIXTY_MONTHS = "SIXTY_MONTHS"
 
 
+class LicenseLevel(str, Enum):
+    """LicenseLevel."""
+
+    NONE = "NONE"
+    MDR = "MDR"
+    XDR_ITDR = "XDR_ITDR"
+    TAEGIS_NATIVE = "TAEGIS_NATIVE"
+    ALL_XDR = "ALL_XDR"
+    ALL = "ALL"
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class MappedTenant:
@@ -500,6 +533,14 @@ class TenantV4:
     data_retention_policy: Optional[RetentionPolicy] = field(
         default=None, metadata=config(field_name="dataRetentionPolicy")
     )
+    license_level: Optional[Union[LicenseLevel, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(LicenseLevel, x),
+            field_name="licenseLevel",
+        ),
+    )
 
 
 @dataclass_json
@@ -605,6 +646,14 @@ class TenantsQuery:
             field_name="orderDir",
         ),
     )
+    license_level: Optional[Union[LicenseLevel, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(LicenseLevel, x),
+            field_name="licenseLevel",
+        ),
+    )
 
 
 @dataclass_json