taegis-sdk-python 1.2.15__tar.gz → 1.2.16__tar.gz

{taegis_sdk_python-1.2.15/taegis_sdk_python.egg-info → taegis_sdk_python-1.2.16}/PKG-INFO

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

taegis_sdk_python-1.2.16/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/access_points/mutations.py

@@ -11,12 +11,13 @@ 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.access_points.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.access_points.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.access_points import AccessPointsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/access_points/subscriptions.py

@@ -11,12 +11,13 @@ 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.access_points.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.access_points.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.access_points import AccessPointsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/agent/mutations.py

@@ -11,12 +11,13 @@ 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.agent.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.agent.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.agent import AgentService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/agent/subscriptions.py

@@ -11,12 +11,13 @@ 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.agent.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.agent.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.agent import AgentService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/alerts/mutations.py

@@ -11,12 +11,13 @@ 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.alerts.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.alerts.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.alerts import AlertsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/alerts/subscriptions.py

@@ -11,12 +11,13 @@ 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.alerts.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.alerts.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.alerts import AlertsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/alerts/types.py

@@ -2104,11 +2104,11 @@ class AggregateAlertsBySeverityInputInput:
     latest: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="latest")
     )
-    excluded_severities: Optional[Union[List[AlertsSeverity], TaegisEnum]] = field(
+    excluded_severities: Optional[List[Union[AlertsSeverity, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[AlertsSeverity], x),
+            decoder=lambda x: decode_enum(AlertsSeverity, x),
             field_name="excluded_severities",
         ),
     )

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets/mutations.py

@@ -11,12 +11,13 @@ 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.assets.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets import AssetsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets/queries.py

@@ -113,9 +113,9 @@ class TaegisSDKAssetsQuery:
         self,
         offset: Optional[int] = None,
         limit: Optional[int] = None,
-        order_by: Optional[AssetsOrderByInput] = None,
-        order_direction: Optional[AssetsOrderDirectionInput] = None,
-        filter_asset_state: Optional[AssetStateFilter] = None,
+        order_by: Optional[Union[AssetsOrderByInput, TaegisEnum]] = None,
+        order_direction: Optional[Union[AssetsOrderDirectionInput, TaegisEnum]] = None,
+        filter_asset_state: Optional[Union[AssetStateFilter, TaegisEnum]] = None,
         only_most_recent: Optional[bool] = None,
     ) -> AssetsResult:
         """Get a list of assets."""
@@ -159,7 +159,9 @@ class TaegisSDKAssetsQuery:
             return AssetsResult.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query allAssetsExport")
 
-    def asset_count(self, endpoint_type: Optional[AgentType] = None) -> AssetCounts:
+    def asset_count(
+        self, endpoint_type: Optional[Union[AgentType, TaegisEnum]] = None
+    ) -> AssetCounts:
         """Count of assets of a specific endpoint_type."""
         endpoint = "assetCount"
 
@@ -323,10 +325,10 @@ class TaegisSDKAssetsQuery:
         tag: Optional[str] = None,
         host_id_partial_match: Optional[bool] = None,
         only_most_recent: Optional[bool] = None,
-        order_by: Optional[AssetsOrderByInput] = None,
-        order_direction: Optional[AssetsOrderDirectionInput] = None,
+        order_by: Optional[Union[AssetsOrderByInput, TaegisEnum]] = None,
+        order_direction: Optional[Union[AssetsOrderDirectionInput, TaegisEnum]] = None,
         or_search: Optional[bool] = None,
-        filter_asset_state: Optional[AssetStateFilter] = None,
+        filter_asset_state: Optional[Union[AssetStateFilter, TaegisEnum]] = None,
     ) -> AssetsResult:
         """search assets. Soon to be deprecated."""
         endpoint = "searchAssets"

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets/subscriptions.py

@@ -11,12 +11,13 @@ 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.assets.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets import AssetsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets2/mutations.py

@@ -11,12 +11,13 @@ 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.assets2.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets2.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets2 import Assets2Service

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets2/queries.py

@@ -31,7 +31,7 @@ class TaegisSDKAssets2Query:
         self.service = service
 
     def facets_v2(
-        self, endpoint_type: Optional[List[EndpointTypeV2]] = None
+        self, endpoint_type: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = None
     ) -> List[FacetV2]:
         """Retrieve a list of facets for a given endpoint type.."""
         endpoint = "facetsV2"
@@ -52,7 +52,7 @@ class TaegisSDKAssets2Query:
     def facet_info_v2(
         self,
         facets: List[str],
-        order_by: Optional[FacetInfoOrderByInputV2] = None,
+        order_by: Optional[Union[FacetInfoOrderByInputV2, TaegisEnum]] = None,
         filter_: Optional[AssetFilter] = None,
     ) -> List[FacetInfoV2]:
         """Get facet info based on a currently selected facet.."""
@@ -80,7 +80,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetFilter] = None,
-        order_by: Optional[AssetSearchOrderByInputV2] = None,
+        order_by: Optional[Union[AssetSearchOrderByInputV2, TaegisEnum]] = None,
     ) -> AssetsV2:
         """Retrieve assets.."""
         endpoint = "assetsV2"
@@ -106,7 +106,7 @@ class TaegisSDKAssets2Query:
         first: Optional[int] = None,
         after: Optional[str] = None,
         filter_: Optional[AssetFilter] = None,
-        order_by: Optional[AssetSearchOrderByInputV2] = None,
+        order_by: Optional[Union[AssetSearchOrderByInputV2, TaegisEnum]] = None,
     ) -> AssetsExportOutputV2:
         """Retrieve assets in a form coercible into CSV files.."""
         endpoint = "exportAssetsV2"
@@ -342,7 +342,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetHostnameFilter] = None,
-        order_by: Optional[AssetHostnameSearchOrderByInput] = None,
+        order_by: Optional[Union[AssetHostnameSearchOrderByInput, TaegisEnum]] = None,
     ) -> AssetsHostname:
         """ "Retrieve hostname details."""
         endpoint = "assetsHostname"
@@ -370,7 +370,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetIPAddressFilter] = None,
-        order_by: Optional[AssetIPAddressSearchOrderByInput] = None,
+        order_by: Optional[Union[AssetIPAddressSearchOrderByInput, TaegisEnum]] = None,
     ) -> AssetsIPAddress:
         """ "Retrieve ipAddress details."""
         endpoint = "assetsIPAddress"
@@ -398,7 +398,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetMacAddressFilter] = None,
-        order_by: Optional[AssetMacAddressSearchOrderByInput] = None,
+        order_by: Optional[Union[AssetMacAddressSearchOrderByInput, TaegisEnum]] = None,
     ) -> AssetsMacAddress:
         """ "Retrieve Mac details."""
         endpoint = "assetsMacAddress"

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets2/subscriptions.py

@@ -11,12 +11,13 @@ 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.assets2.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets2.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets2 import Assets2Service

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/assets2/types.py

@@ -616,11 +616,11 @@ class AssetsExportOutputV2:
 class TagFilter:
     """TagFilter."""
 
-    endpoint_types: Optional[Union[List[EndpointTypeV2], TaegisEnum]] = field(
+    endpoint_types: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[EndpointTypeV2], x),
+            decoder=lambda x: decode_enum(EndpointTypeV2, x),
             field_name="endpointTypes",
         ),
     )
@@ -1413,11 +1413,11 @@ class AssetsV2:
 class AssetHostnameFilter:
     """AssetHostnameFilter."""
 
-    endpoint_types: Optional[Union[List[EndpointTypeV2], TaegisEnum]] = field(
+    endpoint_types: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[EndpointTypeV2], x),
+            decoder=lambda x: decode_enum(EndpointTypeV2, x),
             field_name="endpointTypes",
         ),
     )
@@ -1431,11 +1431,11 @@ class AssetHostnameFilter:
 class AssetIPAddressFilter:
     """AssetIPAddressFilter."""
 
-    endpoint_types: Optional[Union[List[EndpointTypeV2], TaegisEnum]] = field(
+    endpoint_types: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[EndpointTypeV2], x),
+            decoder=lambda x: decode_enum(EndpointTypeV2, x),
             field_name="endpointTypes",
         ),
     )
@@ -1449,11 +1449,11 @@ class AssetIPAddressFilter:
 class AssetMacAddressFilter:
     """AssetMacAddressFilter."""
 
-    endpoint_types: Optional[Union[List[EndpointTypeV2], TaegisEnum]] = field(
+    endpoint_types: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[EndpointTypeV2], x),
+            decoder=lambda x: decode_enum(EndpointTypeV2, x),
             field_name="endpointTypes",
         ),
     )
@@ -1467,19 +1467,19 @@ class AssetMacAddressFilter:
 class AssetFilter:
     """AssetFilter."""
 
-    endpoint_types: Optional[Union[List[EndpointTypeV2], TaegisEnum]] = field(
+    endpoint_types: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[EndpointTypeV2], x),
+            decoder=lambda x: decode_enum(EndpointTypeV2, x),
             field_name="endpointTypes",
         ),
     )
-    asset_state: Optional[Union[List[AssetState], TaegisEnum]] = field(
+    asset_state: Optional[List[Union[AssetState, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[AssetState], x),
+            decoder=lambda x: decode_enum(AssetState, x),
             field_name="assetState",
         ),
     )

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/audits/mutations.py

@@ -11,12 +11,13 @@ 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.audits.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.audits.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.audits import AuditsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/audits/queries.py

@@ -77,7 +77,9 @@ class TaegisSDKAuditsQuery:
             return AuditResult.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query searchAudits")
 
-    def list_application_events(self, application: AuditEventEnum) -> AuditEventResult:
+    def list_application_events(
+        self, application: Union[AuditEventEnum, TaegisEnum]
+    ) -> AuditEventResult:
         """Get event name list."""
         endpoint = "ListApplicationEvents"
 

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/audits/subscriptions.py

@@ -11,12 +11,13 @@ 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.audits.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.audits.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.audits import AuditsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/authz/mutations.py

@@ -11,12 +11,13 @@ 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.authz.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.authz.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.authz import AuthzService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/authz/subscriptions.py

@@ -11,12 +11,13 @@ 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.authz.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.authz.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.authz import AuthzService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/byoti/mutations.py

@@ -11,12 +11,13 @@ 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.byoti.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.byoti.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.byoti import ByotiService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/byoti/subscriptions.py

@@ -11,12 +11,13 @@ 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.byoti.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.byoti.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.byoti import ByotiService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/clients/mutations.py

@@ -11,12 +11,13 @@ 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.clients.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.clients.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.clients import ClientsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/clients/subscriptions.py

@@ -11,12 +11,13 @@ 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.clients.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.clients.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.clients import ClientsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/collector/mutations.py

@@ -11,12 +11,13 @@ 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.collector.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.collector.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.collector import CollectorService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/collector/queries.py

@@ -46,7 +46,7 @@ class TaegisSDKCollectorQuery:
         raise GraphQLNoRowsInResultSetError("for query getCluster")
 
     def get_cluster_backlog_count(
-        self, cluster_id: str, time_range: TimeRange
+        self, cluster_id: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> List[ClusterNodeTimeSeries]:
         """None."""
         endpoint = "getClusterBacklogCount"
@@ -66,7 +66,7 @@ class TaegisSDKCollectorQuery:
         raise GraphQLNoRowsInResultSetError("for query getClusterBacklogCount")
 
     def get_cluster_backlog_age(
-        self, cluster_id: str, time_range: TimeRange
+        self, cluster_id: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> List[ClusterNodeTimeSeries]:
         """None."""
         endpoint = "getClusterBacklogAge"
@@ -86,7 +86,7 @@ class TaegisSDKCollectorQuery:
         raise GraphQLNoRowsInResultSetError("for query getClusterBacklogAge")
 
     def get_cluster_transmit_bytes(
-        self, cluster_id: str, time_range: TimeRange
+        self, cluster_id: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> List[ClusterNodeTimeSeries]:
         """None."""
         endpoint = "getClusterTransmitBytes"
@@ -106,7 +106,7 @@ class TaegisSDKCollectorQuery:
         raise GraphQLNoRowsInResultSetError("for query getClusterTransmitBytes")
 
     def get_cluster_receive_bytes(
-        self, cluster_id: str, time_range: TimeRange
+        self, cluster_id: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> List[ClusterNodeTimeSeries]:
         """None."""
         endpoint = "getClusterReceiveBytes"
@@ -210,7 +210,7 @@ class TaegisSDKCollectorQuery:
     def get_cluster_image(
         self,
         cluster_id: str,
-        image_type: ImageType,
+        image_type: Union[ImageType, TaegisEnum],
         launch_console: Optional[bool] = None,
         aws_details: Optional[AWSDetails] = None,
         gcp_details: Optional[GCPDetails] = None,
@@ -237,7 +237,9 @@ class TaegisSDKCollectorQuery:
             return Image.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query getClusterImage")
 
-    def get_cloud_zones(self, image_type: ImageType) -> List[CloudRegion]:
+    def get_cloud_zones(
+        self, image_type: Union[ImageType, TaegisEnum]
+    ) -> List[CloudRegion]:
         """Get a cloud service region and zones."""
         endpoint = "getCloudZones"
 
@@ -560,7 +562,7 @@ class TaegisSDKCollectorQuery:
         raise GraphQLNoRowsInResultSetError("for query getRoleDeployment")
 
     def get_all_collectors_overview(
-        self, role: str, time_range: TimeRange
+        self, role: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> List[CollectorOverview]:
         """Get all collector overview data for the given role and time range."""
         endpoint = "getAllCollectorsOverview"
@@ -579,7 +581,9 @@ class TaegisSDKCollectorQuery:
             )
         raise GraphQLNoRowsInResultSetError("for query getAllCollectorsOverview")
 
-    def get_collector_metrics(self, time_range: TimeRange) -> CollectorMetrics:
+    def get_collector_metrics(
+        self, time_range: Union[TimeRange, TaegisEnum]
+    ) -> CollectorMetrics:
         """Get collector data flow metrics over a given time range."""
         endpoint = "getCollectorMetrics"
 
@@ -595,7 +599,7 @@ class TaegisSDKCollectorQuery:
         raise GraphQLNoRowsInResultSetError("for query getCollectorMetrics")
 
     def get_aggregate_rate_by_collector(
-        self, cluster_id: str, time_range: TimeRange
+        self, cluster_id: str, time_range: Union[TimeRange, TaegisEnum]
     ) -> AggregateRateByCollector:
         """Get aggregated data flow rate metrics for a given collector over a given time range."""
         endpoint = "getAggregateRateByCollector"
@@ -612,7 +616,9 @@ class TaegisSDKCollectorQuery:
             return AggregateRateByCollector.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query getAggregateRateByCollector")
 
-    def get_flow_rate(self, cluster_id: str, time_range: TimeRange) -> FlowRate:
+    def get_flow_rate(
+        self, cluster_id: str, time_range: Union[TimeRange, TaegisEnum]
+    ) -> FlowRate:
         """Get flow rate metrics for a given collector over a given time range."""
         endpoint = "getFlowRate"
 

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/collector/subscriptions.py

@@ -11,12 +11,13 @@ 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.collector.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.collector.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.collector import CollectorService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.16}/taegis_sdk_python/services/comments/mutations.py

@@ -11,12 +11,13 @@ 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.comments.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.comments.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.comments import CommentsService