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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: taegis-sdk-python
-Version: 1.2.16
+Version: 1.2.18
 Summary: Taegis Python SDK
 Home-page: https://github.com/secureworks/taegis-sdk-python
 Author: Secureworks
@@ -22,6 +22,7 @@ Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: requests-toolbelt
 Requires-Dist: websockets
 Requires-Dist: filelock>=3.12.4
+Requires-Dist: aiohttp
 
 # Taegis SDK for Python
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/requirements.txt

@@ -8,4 +8,5 @@ dataclasses-json
 pyjwt>=2.8.0
 requests-toolbelt
 websockets
-filelock>=3.12.4
+filelock>=3.12.4
+aiohttp

taegis_sdk_python-1.2.18/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/service_core.py

@@ -6,17 +6,18 @@ Taegis ServiceCore manager.
 from __future__ import annotations
 
 import logging
+import threading
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
-import threading
+
+from graphql import GraphQLError, GraphQLField, GraphQLSchema
 
 from gql import Client, gql
 from gql.transport import Transport
-from gql.transport.requests import RequestsHTTPTransport
-from gql.transport.websockets import WebsocketsTransport
-from graphql import GraphQLError, GraphQLField, GraphQLSchema
+from gql.transport.aiohttp import AIOHTTPTransport
 from taegis_sdk_python._version import __version__
 from taegis_sdk_python.errors import InvalidGraphQLEndpoint
+from taegis_sdk_python.transport.aiohttp_websockets import AIOHTTPWebsocketsTransport
 from taegis_sdk_python.utils import async_block, prepare_variables, remove_node
 
 if TYPE_CHECKING:  # pragma: no cover
@@ -129,7 +130,8 @@ class SchemaCacheMap:
                     del self._map[transport.url]
 
     @staticmethod
-    def retrieve_schema(
+    @async_block
+    async def retrieve_schema(
         transport: Transport, introspection_args: Dict[str, Any]
     ) -> GraphQLSchema:
         """Retrieve the schema from a gql Client with the provided transport and
@@ -152,7 +154,7 @@ class SchemaCacheMap:
             introspection_args=introspection_args,
         )
 
-        with client:
+        async with client:
             schema = client.schema
 
         return schema
@@ -207,9 +209,13 @@ class ServiceCore:
     @property
     def sync_client(self) -> Client:
         """GraphQL Synchronous Transport with Client."""
-        transport = RequestsHTTPTransport(
+        client_session_args = {"trust_env": self.service.trust_env}
+
+        transport = AIOHTTPTransport(
             f"{self.sync_url}{self.gateway}",
             headers=self.service.headers,
+            ssl=self.service.ssl,
+            client_session_args=client_session_args,
         )
 
         client = Client(
@@ -224,6 +230,7 @@ class ServiceCore:
                     )
                 },
             ),
+            execute_timeout=self.service.execute_timeout,
         )
 
         return client
@@ -239,16 +246,23 @@ class ServiceCore:
         if self.service.tenant_id:
             subprotocols.append(f"x-tenant-context-{self.service.tenant_id}")
 
-        transport = WebsocketsTransport(
+        client_session_args = {"trust_env": self.service.trust_env}
+
+        transport = AIOHTTPWebsocketsTransport(
             f"{self.wss_url}{self.gateway}",
             headers=self.service.headers,
             subprotocols=subprotocols,
-            connect_args={"max_size": None},
+            ssl=self.service.ssl,
+            client_session_args=client_session_args,
+            proxy=self.service.proxy,
+            proxy_auth=self.service.proxy_auth,
+            proxy_headers=self.service.proxy_headers,
         )
 
         client = Client(
             transport=transport,
             schema=self.schema,
+            execute_timeout=self.service.execute_timeout,
         )
 
         return client
@@ -448,7 +462,8 @@ class ServiceCore:
 
         return self.subscribe(query_string, variables)
 
-    def execute(
+    @async_block
+    async def execute(
         self, query_string: str, variables: Optional[Dict[str, Any]] = None
     ) -> Any:
         """Execute a GraphQL string.
@@ -465,10 +480,19 @@ class ServiceCore:
         Any
             Query/Mutation results
         """
+        extra_args = {}
+        if self.service.proxy:
+            extra_args = {
+                "proxy": self.service.proxy,
+                "proxy_auth": self.service.proxy_auth,
+                "proxy_headers": self.service.proxy_headers,
+            }
+
         query = gql(query_string)
-        return self.sync_client.execute(
+        return await self.sync_client.execute_async(
             query,
             variable_values=prepare_variables(variables),
+            extra_args=extra_args,
         )
 
     @async_block
@@ -500,7 +524,8 @@ class ServiceCore:
 
         return results[:-1]
 
-    def _build_validated_query(
+    @async_block
+    async def _build_validated_query(
         self,
         operation_type: str,
         endpoint: str,
@@ -537,7 +562,7 @@ class ServiceCore:
         query_string = " ".join(query_string.split())
 
         # open a connection for introspection and download schema
-        with self.sync_client as session:
+        async with self.sync_client as session:
             # if multiple fields are invalid, we want to iterate until the
             # output string is valid
             for _ in range(10000):

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/__init__.py

@@ -2,8 +2,14 @@
 This needs to be a generated file.  Need to make jinja template.
 """
 
-from typing import Dict, Optional, Any
+import logging
 import threading
+from ssl import SSLContext
+from typing import Any, Dict, Optional, Union, Literal
+
+import aiohttp
+from aiohttp.client_reqrep import Fingerprint
+from aiohttp.typedefs import LooseHeaders
 
 from taegis_sdk_python._consts import TAEGIS_ENVIRONMENT_URLS
 from taegis_sdk_python._version import __version__
@@ -60,6 +66,8 @@ from taegis_sdk_python.services.vdr import VDRService
 
 __all__ = ["GraphQLService"]
 
+log = logging.getLogger(__name__)
+
 
 class GraphQLService:
     """Taegis GraphQL Service manager."""
@@ -73,6 +81,12 @@ class GraphQLService:
         gateway: Optional[str] = None,
         extra_headers: Optional[Dict[str, Any]] = None,
         schema_expiry: int = 5,
+        proxy: Optional[str] = None,
+        proxy_auth: Optional[aiohttp.BasicAuth] = None,
+        proxy_headers: Optional[LooseHeaders] = None,
+        trust_env: bool = False,
+        ssl: Optional[Union[SSLContext, Literal[False], Fingerprint]] = None,
+        execute_timeout: Optional[Union[int, float]] = 30,
     ):  # pylint: disable=too-many-statements
         """
         GraphQLService
@@ -114,6 +128,20 @@ class GraphQLService:
         self._schema_expiry = schema_expiry
         self._input_value_deprecation = None
 
+        if trust_env and proxy:
+            raise ValueError("trust_env and proxy cannot be used together")
+
+        if proxy_auth and not proxy:
+            raise ValueError("proxy_auth requires proxy to be set")
+
+        self._proxy = proxy
+        self._proxy_auth = proxy_auth
+        self._proxy_headers = proxy_headers
+        self._ssl = ssl
+        self._trust_env = trust_env
+
+        self._execute_timeout = execute_timeout
+
         self._access_points = None
         self._agent = None
         self._alerts = None
@@ -270,6 +298,54 @@ class GraphQLService:
             "input_value_deprecation", self._input_value_deprecation
         )
 
+    @property
+    def proxy(self):
+        """Proxy URL for API requests."""
+        return self._context_manager.get("proxy", self._proxy)
+
+    @property
+    def proxy_auth(self):
+        """Proxy Auth for API requests."""
+        value = self._context_manager.get("proxy_auth", self._proxy_auth)
+
+        if value and not self.proxy:
+            log.warning("proxy_auth is ignored when proxy is not set")
+            return None
+
+        return value
+
+    @property
+    def proxy_headers(self):
+        """Proxy Headers for API requests."""
+        value = self._context_manager.get("proxy_headers", self._proxy_headers)
+
+        if value and not self.proxy:
+            log.warning("proxy_headers is ignored when proxy is not set")
+            return None
+
+        return value
+
+    @property
+    def ssl(self):
+        """SSL Context for API requests."""
+        return self._context_manager.get("ssl", self._ssl)
+
+    @property
+    def trust_env(self):
+        """Trust Environment for Proxy Authentication."""
+        value = self._context_manager.get("trust_env", self._trust_env)
+
+        if value and self.proxy:
+            log.warning("trust_env is ignored when proxy is set")
+            return False
+
+        return value
+
+    @property
+    def execute_timeout(self):
+        """GraphQL Execute Timeout."""
+        return self._context_manager.get("execute_timeout", self._execute_timeout)
+
     @property
     def access_points(self):
         """Access Points Service Endpoint."""

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/access_points/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/agent/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/alerts/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

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

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

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

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

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

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/authz/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/byoti/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/clients/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

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

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/comments/queries.py

@@ -11,16 +11,18 @@ 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
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/contracted_endpoint/queries.py

@@ -11,16 +11,18 @@ 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.contracted_endpoint.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.contracted_endpoint.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.contracted_endpoint import ContractedEndpointService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/datasources/queries.py

@@ -11,16 +11,18 @@ 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.datasources.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.datasources.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.datasources import DatasourcesService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/detector_registry/queries.py

@@ -11,16 +11,18 @@ 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.detector_registry.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.detector_registry.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.detector_registry import DetectorRegistryService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/endpoint_command_manager/queries.py

@@ -11,18 +11,20 @@ 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.endpoint_command_manager.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.endpoint_command_manager.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.endpoint_command_manager import (
         EndpointCommandManagerService,
     )
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/endpoint_management_service/queries.py

@@ -11,18 +11,20 @@ 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.endpoint_management_service.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.endpoint_management_service.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.endpoint_management_service import (
         EndpointManagementServiceService,
     )
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/event_search/queries.py

@@ -11,16 +11,18 @@ 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.event_search.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.event_search.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.event_search import EventSearchService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/events/queries.py

@@ -11,16 +11,18 @@ 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.events.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.events.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.events import EventsService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.16 → taegis_sdk_python-1.2.18}/taegis_sdk_python/services/exports/queries.py

@@ -11,16 +11,18 @@ 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.exports.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.exports.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.exports import ExportsService
 
+
 log = logging.getLogger(__name__)