taegis-sdk-python 1.6.11.1__tar.gz → 1.6.13__tar.gz

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/PKG-INFO

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

@@ -1,3 +1,3 @@
 """Version idenitier."""
 
-__version__ = "1.6.11.1"
+__version__ = "1.6.13"

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/commons/alerts/federated_search.py

@@ -110,8 +110,8 @@ def alerts_federated_search(
     *,
     limit: int = 10000,
     caller_name: str = "Taegis SDK Commons",
-    federated_call: Callable,
-    federated_poll_call: Callable,
+    federated_call: Callable = alerts_service_search_with_events,
+    federated_poll_call: Callable = alerts_service_poll_with_events,
 ) -> List[TaegisCommonsAlertsResponse]:
     """
     Search Taegis Alerts service.

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/commons/cases/federated_search.py

@@ -99,7 +99,7 @@ def cases_federated_search(
     query: str,
     *,
     limit: int = 10000,
-    federated_call: Callable,
+    federated_call: Callable = investigations_search_with_subjects,
 ) -> List[InvestigationsV2]:
     """Taegis Cases search with Subject federation."""
 

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/__init__.py

@@ -55,6 +55,7 @@ from taegis_sdk_python.services.ingest_stats import IngestStatsService
 from taegis_sdk_python.services.investigations import InvestigationsService
 from taegis_sdk_python.services.investigations2 import Investigations2Service
 from taegis_sdk_python.services.isensor import IsensorService
+from taegis_sdk_python.services.llm_service import LlmServiceService
 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
@@ -64,6 +65,7 @@ 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.ql_schemas import QlSchemasService
 from taegis_sdk_python.services.queries import QueriesService
 from taegis_sdk_python.services.roadrunner import RoadrunnerService
 from taegis_sdk_python.services.rules import RulesService
@@ -219,6 +221,7 @@ class GraphQLService:
         self._investigations = None
         self._investigations2 = None
         self._isensor = None
+        self._llm_service = None
         self._mitre_attack_info = None
         self._multi_tenant_context = None
         self._multi_tenant_ioc = None
@@ -228,6 +231,7 @@ class GraphQLService:
         self._notifications2 = None
         self._preferences = None
         self._process_trees = None
+        self._ql_schemas = None
         self._queries = None
         self._roadrunner = None
         self._rules = None
@@ -699,6 +703,13 @@ class GraphQLService:
             self._isensor = IsensorService(self)
         return self._isensor
 
+    @property
+    def llm_service(self):
+        """LLM Service Endpoint."""
+        if not self._llm_service:
+            self._llm_service = LlmServiceService(self)
+        return self._llm_service
+
     @property
     def mitre_attack_info(self):
         """MitreAttackInfo Service Endpoint."""
@@ -756,16 +767,11 @@ class GraphQLService:
         return self._preferences
 
     @property
-    def process_tress(self):
-        """Process Trees Service Endpoint."""
-
-        log.warning(
-            "Please use service.process_trees instead of service.process_tress."
-        )
-
-        if not self._process_trees:
-            self._process_trees = ProcessTreesService(self)
-        return self._process_trees
+    def ql_schemas(self):
+        """QL Schemas Service Endpoint."""
+        if not self._ql_schemas:
+            self._ql_schemas = QlSchemasService(self)
+        return self._ql_schemas
 
     @property
     def process_trees(self):

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/access_points/mutations.py

@@ -33,7 +33,7 @@ class TaegisSDKAccessPointsMutation:
         self.service = service
 
     def create_access_point(self, principal: str) -> AccessPoint:
-        """None."""
+        """No developer notes."""
         endpoint = "createAccessPoint"
 
         result = self.service.execute_mutation(
@@ -51,7 +51,7 @@ class TaegisSDKAccessPointsMutation:
         raise GraphQLNoRowsInResultSetError("for mutation createAccessPoint")
 
     def update_access_point(self, principal: str) -> AccessPoint:
-        """None."""
+        """No developer notes."""
         endpoint = "updateAccessPoint"
 
         result = self.service.execute_mutation(
@@ -69,7 +69,7 @@ class TaegisSDKAccessPointsMutation:
         raise GraphQLNoRowsInResultSetError("for mutation updateAccessPoint")
 
     def delete_access_point(self) -> AccessPoint:
-        """None."""
+        """No developer notes."""
         endpoint = "deleteAccessPoint"
 
         result = self.service.execute_mutation(

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/access_points/queries.py

@@ -33,7 +33,7 @@ class TaegisSDKAccessPointsQuery:
         self.service = service
 
     def get_access_point(self) -> AccessPoint:
-        """None."""
+        """No developer notes."""
         endpoint = "getAccessPoint"
 
         result = self.service.execute_query(
@@ -49,7 +49,7 @@ class TaegisSDKAccessPointsQuery:
         raise GraphQLNoRowsInResultSetError("for query getAccessPoint")
 
     def get_access_point_template(self) -> AccessPointCloudFormation:
-        """None."""
+        """No developer notes."""
         endpoint = "getAccessPointTemplate"
 
         result = self.service.execute_query(
@@ -65,7 +65,7 @@ class TaegisSDKAccessPointsQuery:
         raise GraphQLNoRowsInResultSetError("for query getAccessPointTemplate")
 
     def get_access_point_prefixes(self) -> AccessPointPrefixes:
-        """None."""
+        """No developer notes."""
         endpoint = "getAccessPointPrefixes"
 
         result = self.service.execute_query(

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/alerts/mutations.py

@@ -35,7 +35,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_update_investigation_info(
         self, in_: Optional[UpdateInvestigationRequestInput] = None
     ) -> UpdateInvestigationResponse:
-        """None."""
+        """No developer notes."""
         endpoint = "alertsServiceUpdateInvestigationInfo"
 
         result = self.service.execute_mutation(
@@ -57,7 +57,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_update_resolution_info(
         self, in_: Optional[UpdateResolutionRequestInput] = None
     ) -> UpdateResolutionResponse:
-        """Add a resolution or modify an existing resolution for a give list of alert IDs.."""
+        """Add a resolution or modify an existing resolution for a give list of alert IDs."""
         endpoint = "alertsServiceUpdateResolutionInfo"
 
         result = self.service.execute_mutation(
@@ -79,7 +79,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_bulk_investigations_processor(
         self, in_: Optional[BulkInvestigationsRequestInput] = None
     ) -> BulkInvestigationsResponse:
-        """Bulk add alerts to an existing investigation by providing either a query or list of alert IDs. If a query is provided, then all alerts matching the query will be added to the investigation.."""
+        """Bulk add alerts to an existing investigation by providing either a query or list of alert IDs. If a query is provided, then all alerts matching the query will be added to the investigation."""
         endpoint = "alertsServiceBulkInvestigationsProcessor"
 
         result = self.service.execute_mutation(
@@ -101,7 +101,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_evict(
         self, in_: Optional[EvictRequestInput] = None
     ) -> EvictResponse:
-        """DEPRECATED: Does not do anything other than to return OK. No replacement necessary.."""
+        """DEPRECATED: Does not do anything other than to return OK. No replacement necessary."""
         endpoint = "alertsServiceEvict"
 
         log.warning(
@@ -125,7 +125,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_update_threat_score(
         self, in_: UpdateThreatScoreRequestInput
     ) -> UpdateThreatScoreResponse:
-        """Update threat score for a given list of alert IDs.."""
+        """Update threat score for a given list of alert IDs."""
         endpoint = "alertsServiceUpdateThreatScore"
 
         result = self.service.execute_mutation(
@@ -147,7 +147,7 @@ class TaegisSDKAlertsMutation:
     def alerts_service_update_threat_score_v2(
         self, in_: UpdateThreatScoreV2RequestInput
     ) -> UpdateThreatScoreResponse:
-        """Update threat score v2 for a given list of alert IDs.."""
+        """Update threat score v2 for a given list of alert IDs."""
         endpoint = "alertsServiceUpdateThreatScoreV2"
 
         result = self.service.execute_mutation(

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/alerts/queries.py

@@ -35,7 +35,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_retrieve_alerts_by_id(
         self, in_: Optional[GetByIDRequestInput] = None
     ) -> AlertsResponse:
-        """Provide a list of Alert IDs to retrieve each alert’s detail.."""
+        """Provide a list of Alert IDs to retrieve each alert’s detail."""
         endpoint = "alertsServiceRetrieveAlertsById"
 
         result = self.service.execute_query(
@@ -55,7 +55,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_retrieve_alerts_by_host(
         self, in_: Optional[GetByIDRequestInput] = None
     ) -> AlertsResponse:
-        """Provide a list of Host IDs to retrieve alert details about each alert that contains those hosts.."""
+        """Provide a list of Host IDs to retrieve alert details about each alert that contains those hosts."""
         endpoint = "alertsServiceRetrieveAlertsByHost"
 
         result = self.service.execute_query(
@@ -77,7 +77,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_retrieve_alerts_by_entity(
         self, in_: Optional[GetByIDRequestInput] = None
     ) -> AlertsResponse:
-        """Provide a list of entities to retrieve alert details about each alert that contains those entities.."""
+        """Provide a list of entities to retrieve alert details about each alert that contains those entities."""
         endpoint = "alertsServiceRetrieveAlertsByEntity"
 
         result = self.service.execute_query(
@@ -99,7 +99,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_retrieve_alerts_by_group_key(
         self, in_: Optional[GetByIDRequestInput] = None
     ) -> AlertsResponse:
-        """Provide a list of entities to retrieve alert details about each alert that contains the group_key. This is used by the service to aid in alert deduplication. This would not commonly be used by a tenant of XDR.."""
+        """Provide a list of entities to retrieve alert details about each alert that contains the group_key. This is used by the service to aid in alert deduplication. This would not commonly be used by a tenant of XDR."""
         endpoint = "alertsServiceRetrieveAlertsByGroupKey"
 
         result = self.service.execute_query(
@@ -121,7 +121,7 @@ class TaegisSDKAlertsQuery:
     def alerts_count_by_tenant(
         self, in_: AlertsCountByTenantInput
     ) -> AlertsCountByTenantResponse:
-        """Returns the count of alerts per tenant. Allows a CQL query, but any aggregation or pipe will be ignored.."""
+        """Returns the count of alerts per tenant. Allows a CQL query, but any aggregation or pipe will be ignored."""
         endpoint = "alertsCountByTenant"
 
         result = self.service.execute_query(
@@ -141,7 +141,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_search(
         self, in_: Optional[SearchRequestInput] = None
     ) -> AlertsResponse:
-        """Search alerts using Query Language. This is the same query language provided in Advanced Search page in Taegis XDR.."""
+        """Search alerts using Query Language. This is the same query language provided in Advanced Search page in Taegis XDR."""
         endpoint = "alertsServiceSearch"
 
         result = self.service.execute_query(
@@ -161,7 +161,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_poll(
         self, in_: Optional[PollRequestInput] = None
     ) -> AlertsResponse:
-        """Poll for results for a specific `search_id`.."""
+        """Poll for results for a specific `search_id`."""
         endpoint = "alertsServicePoll"
 
         result = self.service.execute_query(
@@ -181,7 +181,7 @@ class TaegisSDKAlertsQuery:
     def alerts_service_aggregate_alerts_by_severity(
         self, in_: Optional[AggregateAlertsBySeverityInputInput] = None
     ) -> AlertsAggregateResponse:
-        """Pull alert severity aggregates based on `group_by` parameters: domain, watchlist, hostname, detector, user.."""
+        """Pull alert severity aggregates based on `group_by` parameters: domain, watchlist, hostname, detector, user."""
         endpoint = "alertsServiceAggregateAlertsBySeverity"
 
         result = self.service.execute_query(

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/alerts/subscriptions.py

@@ -35,7 +35,7 @@ class TaegisSDKAlertsSubscription:
     def alerts_service_bulk_resolution_processor(
         self, in_: Optional[BulkResolutionRequestInput] = None
     ) -> List[BulkResolutionResponse]:
-        """Add a resolution or modify an existing resolution for multiple alerts selected with a CQL query.."""
+        """Add a resolution or modify an existing resolution for multiple alerts selected with a CQL query."""
         endpoint = "alertsServiceBulkResolutionProcessor"
 
         result = self.service.execute_subscription(

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/alerts/types.py

@@ -3229,6 +3229,9 @@ class Alert2:
     parent_tenant_id: Optional[str] = field(
         default=None, metadata=config(field_name="parent_tenant_id")
     )
+    mdr_provider: Optional[str] = field(
+        default=None, metadata=config(field_name="mdr_provider")
+    )
     suppressed: Optional[bool] = field(
         default=None, metadata=config(field_name="suppressed")
     )

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/assets2/mutations.py

@@ -43,7 +43,7 @@ class TaegisSDKAssets2Mutation:
         tags for the endpoint. If the intention is to change/remove a single
         tag, query the asset first to get the current set of tags for the
         endpoint, then pass the complete set of tags desired with the changes
-        included.."""
+        included."""
         endpoint = "updateTagsForEndpointV2"
 
         result = self.service.execute_mutation(
@@ -68,7 +68,7 @@ class TaegisSDKAssets2Mutation:
         endpoints have tags with the same key, but a different value, the value
         will be updated with the value in the input. Use the task ID in the
         response to poll the bulkUpdateTagsForEndpointsStatusV2 query to
-        determine if the job succeeded.."""
+        determine if the job succeeded."""
         endpoint = "bulkUpdateTagsForEndpointsV2"
 
         result = self.service.execute_mutation(
@@ -88,7 +88,7 @@ class TaegisSDKAssets2Mutation:
     def delete_assets_v2(self, input_: DeleteAssetsInputV2) -> BulkOpPayloadV2:
         """Start a job to "soft" delete the assets matching the filter criteria.
         Use the task ID in the response to poll the deleteAssetsStatusV2
-        query to determine if the job succeeded.."""
+        query to determine if the job succeeded."""
         endpoint = "deleteAssetsV2"
 
         result = self.service.execute_mutation(
@@ -108,7 +108,7 @@ class TaegisSDKAssets2Mutation:
     def restore_assets_v2(self, input_: RestoreAssetsInputV2) -> BulkOpPayloadV2:
         """Start a job to restore assets that were previously deleted and that
         match the filter criteria. Use the task ID in the response to poll the
-        restoreAssetsStatusV2 query to determine if the job succeeded.."""
+        restoreAssetsStatusV2 query to determine if the job succeeded."""
         endpoint = "restoreAssetsV2"
 
         result = self.service.execute_mutation(
@@ -130,7 +130,7 @@ class TaegisSDKAssets2Mutation:
     ) -> BulkOpPayloadV2:
         """Start a job to assign the endpoints matching the filter criteria to the
         endpoint group in the input. Use the task ID in the response to poll the
-        assignBulkAssetsToGroupStatus query to determine if the job succeeded.."""
+        assignBulkAssetsToGroupStatus query to determine if the job succeeded."""
         endpoint = "assignBulkAssetsToGroup"
 
         result = self.service.execute_mutation(
@@ -153,7 +153,7 @@ class TaegisSDKAssets2Mutation:
         """Start a job to delete the provided tags from the endpoints matching the
         filter criteria in the input. Use the task ID in the response to poll
         the bulkDeleteTagsForEndpointsStatusV2 query to determine if the job
-        succeeded.."""
+        succeeded."""
         endpoint = "bulkDeleteTagsForEndpointsV2"
 
         result = self.service.execute_mutation(
@@ -173,7 +173,7 @@ class TaegisSDKAssets2Mutation:
     def bulk_reconnect_native_assets(
         self, input_: BulkReconnectNativeAssetsInput
     ) -> BulkOpPayloadV2:
-        """Start a job to reconnect taegis endpoints matching the filter criteria in the input.."""
+        """Start a job to reconnect taegis endpoints matching the filter criteria in the input."""
         endpoint = "bulkReconnectNativeAssets"
 
         result = self.service.execute_mutation(
@@ -193,7 +193,7 @@ class TaegisSDKAssets2Mutation:
     def bulk_uninstall_native_assets(
         self, input_: BulkUninstallNativeAssetsInput
     ) -> BulkOpPayloadV2:
-        """Start a job to uninstall taegis endpoints matching the filter criteria in the input.."""
+        """Start a job to uninstall taegis endpoints matching the filter criteria in the input."""
         endpoint = "bulkUninstallNativeAssets"
 
         result = self.service.execute_mutation(
@@ -209,3 +209,23 @@ class TaegisSDKAssets2Mutation:
         if result.get(endpoint) is not None:
             return BulkOpPayloadV2.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for mutation bulkUninstallNativeAssets")
+
+    def bulk_reconnect_by_scope(
+        self, input_: BulkReconnectByScopeInput
+    ) -> BulkOpPayloadV2:
+        """Start a job to reconnect taegis endpoints by scope (tenant or group level)."""
+        endpoint = "bulkReconnectByScope"
+
+        result = self.service.execute_mutation(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(
+                BulkOpPayloadV2,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
+        )
+        if result.get(endpoint) is not None:
+            return BulkOpPayloadV2.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for mutation bulkReconnectByScope")

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/assets2/queries.py

@@ -35,7 +35,7 @@ class TaegisSDKAssets2Query:
     def facets_v2(
         self, endpoint_type: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = None
     ) -> List[FacetV2]:
-        """Retrieve a list of facets for a given endpoint type.."""
+        """Retrieve a list of facets for a given endpoint type."""
         endpoint = "facetsV2"
 
         result = self.service.execute_query(
@@ -60,7 +60,7 @@ class TaegisSDKAssets2Query:
         order_by: Optional[Union[FacetInfoOrderByInputV2, TaegisEnum]] = None,
         filter_: Optional[AssetFilter] = None,
     ) -> List[FacetInfoV2]:
-        """Get facet info based on a currently selected facet.."""
+        """Get facet info based on a currently selected facet."""
         endpoint = "facetInfoV2"
 
         result = self.service.execute_query(
@@ -90,7 +90,7 @@ class TaegisSDKAssets2Query:
         filter_: Optional[AssetFilter] = None,
         order_by: Optional[Union[AssetSearchOrderByInputV2, TaegisEnum]] = None,
     ) -> AssetsV2:
-        """Retrieve assets.."""
+        """Retrieve assets."""
         endpoint = "assetsV2"
 
         result = self.service.execute_query(
@@ -119,7 +119,7 @@ class TaegisSDKAssets2Query:
         filter_: Optional[AssetFilter] = None,
         order_by: Optional[Union[AssetSearchOrderByInputV2, TaegisEnum]] = None,
     ) -> AssetsExportOutputV2:
-        """Retrieve assets in a form coercible into CSV files.."""
+        """Retrieve assets in a form coercible into CSV files."""
         endpoint = "exportAssetsV2"
 
         result = self.service.execute_query(
@@ -140,7 +140,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query exportAssetsV2")
 
     def tag_keys_v2(self, filter_: Optional[TagFilter] = None) -> List[str]:
-        """Fetch the tag keys matching the filter criteria.."""
+        """Fetch the tag keys matching the filter criteria."""
         endpoint = "tagKeysV2"
 
         result = self.service.execute_query(
@@ -155,7 +155,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query tagKeysV2")
 
     def tag_values_v2(self, filter_: Optional[TagFilter] = None) -> List[str]:
-        """Fetch the tag values matching the filter criteria.."""
+        """Fetch the tag values matching the filter criteria."""
         endpoint = "tagValuesV2"
 
         result = self.service.execute_query(
@@ -170,7 +170,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query tagValuesV2")
 
     def update_tags_for_endpoint_status_v2(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the updateTagsForEndpointV2 operation.."""
+        """Return the status of the updateTagsForEndpointV2 operation."""
         endpoint = "updateTagsForEndpointStatusV2"
 
         result = self.service.execute_query(
@@ -188,7 +188,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query updateTagsForEndpointStatusV2")
 
     def delete_assets_status_v2(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the deleteAssetsV2 operation.."""
+        """Return the status of the deleteAssetsV2 operation."""
         endpoint = "deleteAssetsStatusV2"
 
         result = self.service.execute_query(
@@ -206,7 +206,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query deleteAssetsStatusV2")
 
     def restore_assets_status_v2(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the restoreAssetsV2 operation.."""
+        """Return the status of the restoreAssetsV2 operation."""
         endpoint = "restoreAssetsStatusV2"
 
         result = self.service.execute_query(
@@ -224,7 +224,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query restoreAssetsStatusV2")
 
     def assign_bulk_assets_to_group_status(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the assignBulkAssetsToGroup operation.."""
+        """Return the status of the assignBulkAssetsToGroup operation."""
         endpoint = "assignBulkAssetsToGroupStatus"
 
         result = self.service.execute_query(
@@ -242,7 +242,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query assignBulkAssetsToGroupStatus")
 
     def bulk_update_tags_for_endpoints_status_v2(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the bulkUpdateTagsForEndpointsV2 operation.."""
+        """Return the status of the bulkUpdateTagsForEndpointsV2 operation."""
         endpoint = "bulkUpdateTagsForEndpointsStatusV2"
 
         result = self.service.execute_query(
@@ -262,7 +262,7 @@ class TaegisSDKAssets2Query:
         )
 
     def bulk_delete_tags_for_endpoints_status_v2(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the bulkDeleteTagsForEndpointsV2 operation.."""
+        """Return the status of the bulkDeleteTagsForEndpointsV2 operation."""
         endpoint = "bulkDeleteTagsForEndpointsStatusV2"
 
         result = self.service.execute_query(
@@ -282,7 +282,7 @@ class TaegisSDKAssets2Query:
         )
 
     def bulk_reconnect_native_assets_status(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the bulkReconnectNativeAssets operation.."""
+        """Return the status of the bulkReconnectNativeAssets operation."""
         endpoint = "bulkReconnectNativeAssetsStatus"
 
         result = self.service.execute_query(
@@ -300,7 +300,7 @@ class TaegisSDKAssets2Query:
         raise GraphQLNoRowsInResultSetError("for query bulkReconnectNativeAssetsStatus")
 
     def bulk_uninstall_native_assets_status(self, id_: str) -> TaskInfoPayload:
-        """Return the status of the bulkUninstallNativeAssets operation.."""
+        """Return the status of the bulkUninstallNativeAssets operation."""
         endpoint = "bulkUninstallNativeAssetsStatus"
 
         result = self.service.execute_query(
@@ -317,10 +317,28 @@ class TaegisSDKAssets2Query:
             return TaskInfoPayload.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query bulkUninstallNativeAssetsStatus")
 
+    def bulk_reconnect_by_scope_status(self, id_: str) -> TaskInfoPayload:
+        """Return the status of the bulkReconnectByScope operation."""
+        endpoint = "bulkReconnectByScopeStatus"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "id": prepare_input(id_),
+            },
+            output=build_output_string(
+                TaskInfoPayload,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
+        )
+        if result.get(endpoint) is not None:
+            return TaskInfoPayload.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query bulkReconnectByScopeStatus")
+
     def asset_dead_period(self) -> str:
         """Return the tenant's asset dead period threshold used by the api. The value
         from the tenants preference api is used if present, otherwise the default of
-        720h (30 days) is used.."""
+        720h (30 days) is used."""
         endpoint = "assetDeadPeriod"
 
         result = self.service.execute_query(endpoint=endpoint, variables={}, output="")

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/assets2/types.py

@@ -187,6 +187,14 @@ class BulkOpStatusV2(str, Enum):
     TASK_STATE_IN_PROGRESS = "TASK_STATE_IN_PROGRESS"
 
 
+class ReconnectScope(str, Enum):
+    """ReconnectScope."""
+
+    TENANT = "TENANT"
+    GROUPS = "GROUPS"
+    SETTING = "SETTING"
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class ConnectionDetails:
@@ -671,6 +679,28 @@ class AssetsExportOutputV2:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class BulkReconnectByScopeInput:
+    """BulkReconnectByScopeInput."""
+
+    group_ids: Optional[List[str]] = field(
+        default=None, metadata=config(field_name="groupIds")
+    )
+    setting_id: Optional[str] = field(
+        default=None, metadata=config(field_name="settingId")
+    )
+    reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
+    scope: Optional[Union[ReconnectScope, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ReconnectScope, x),
+            field_name="scope",
+        ),
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class TagFilter:

{taegis_sdk_python-1.6.11.1 → taegis_sdk_python-1.6.13}/taegis_sdk_python/services/authz/queries.py

@@ -289,8 +289,9 @@ class TaegisSDKAuthzQuery:
         self,
         subject_id: str,
         subject_type: Optional[Union[AuthzSubjectIDType, TaegisEnum]] = None,
+        include_mdr_provider: Optional[bool] = None,
     ) -> bool:
-        """Indicates if the input subject can assume the target tenant."""
+        """Indicates if the input subject can assume the target tenant, includeMDRProvider optionally also returns true if the subject can assume the tenant through an MDR provider relationship."""
         endpoint = "authzCanSubjectIDAssumeTenant"
 
         result = self.service.execute_query(
@@ -298,6 +299,7 @@ class TaegisSDKAuthzQuery:
             variables={
                 "subjectID": prepare_input(subject_id),
                 "subjectType": prepare_input(subject_type),
+                "includeMDRProvider": prepare_input(include_mdr_provider),
             },
             output="",
         )
@@ -472,3 +474,21 @@ class TaegisSDKAuthzQuery:
                 [r or {} for r in result.get(endpoint)], many=True
             )
         raise GraphQLNoRowsInResultSetError("for query authzTrustedRoles")
+
+    def authz_check_step_up(self, input_: AuthzStepUpInput) -> AuthzStepUpResponse:
+        """Step-up authorization check."""
+        endpoint = "authzCheckStepUp"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(
+                AuthzStepUpResponse,
+                exclude_deprecated_output=self.service.exclude_deprecated_output,
+            ),
+        )
+        if result.get(endpoint) is not None:
+            return AuthzStepUpResponse.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query authzCheckStepUp")