fix: Resolve mypy type checking errors for Sentinel plugin

Author: SinsBre

graphistry/client_session.py

@@ -9,6 +9,7 @@
 from . import util
 from .plugins_types.spanner_types import SpannerConfig
 from .plugins_types.kusto_types import KustoConfig
+from .plugins_types.sentinel_types import SentinelConfig
 
 
 
@@ -85,6 +86,7 @@ def __init__(self) -> None:
         # NOTE: These are dataclasses, so we shallow copy them
         self.kusto: Optional[KustoConfig] = None
         self.spanner: Optional[SpannerConfig] = None
+        self.sentinel: Optional[SentinelConfig] = None
 
         # TODO: Migrate to a pattern like Kusto or Spanner
         self._bolt_driver: Optional[Any] = None

graphistry/plugins/sentinel.py

@@ -1,14 +1,20 @@
 import time
 import pandas as pd
-from typing import Any, List, Optional, TYPE_CHECKING, Union, overload, Literal
+from typing import Any, List, Optional, TYPE_CHECKING, Union, overload, Literal, Tuple
 from datetime import datetime, timedelta
 
 if TYPE_CHECKING:
     from azure.monitor.query import LogsQueryClient
     from azure.core.credentials import TokenCredential
+    from azure.core.exceptions import HttpResponseError
+    from azure.identity import DefaultAzureCredential, ClientSecretCredential, DeviceCodeCredential
 else:
     LogsQueryClient = Any
     TokenCredential = Any
+    HttpResponseError = Any
+    DefaultAzureCredential = Any
+    ClientSecretCredential = Any
+    DeviceCodeCredential = Any
 
 from graphistry.Plottable import Plottable
 from graphistry.util import setup_logger
@@ -228,7 +234,7 @@ def kql(
         self,
         query: str,
         *,
-        timespan: Optional[Union[timedelta, tuple[datetime, datetime]]] = None,
+        timespan: Optional[Union[timedelta, Tuple[datetime, datetime]]] = None,
         unwrap_nested: Optional[bool] = None,
         single_table: Literal[True] = True,
         include_statistics: bool = False
@@ -240,7 +246,7 @@ def kql(
         self,
         query: str,
         *,
-        timespan: Optional[Union[timedelta, tuple[datetime, datetime]]] = None,
+        timespan: Optional[Union[timedelta, Tuple[datetime, datetime]]] = None,
         unwrap_nested: Optional[bool] = None,
         single_table: Literal[False],
         include_statistics: bool = False
@@ -252,7 +258,7 @@ def kql(
         self,
         query: str,
         *,
-        timespan: Optional[Union[timedelta, tuple[datetime, datetime]]] = None,
+        timespan: Optional[Union[timedelta, Tuple[datetime, datetime]]] = None,
         unwrap_nested: Optional[bool] = None,
         single_table: bool = True,
         include_statistics: bool = False
@@ -263,7 +269,7 @@ def kql(
         self,
         query: str,
         *,
-        timespan: Optional[Union[timedelta, tuple[datetime, datetime]]] = None,
+        timespan: Optional[Union[timedelta, Tuple[datetime, datetime]]] = None,
         unwrap_nested: Optional[bool] = None,
         single_table: bool = True,
         include_statistics: bool = False
@@ -479,7 +485,7 @@ def sentinel_schema(self, table: str) -> pd.DataFrame:
     def _sentinel_query(
         self,
         query: str,
-        timespan: Optional[Union[timedelta, tuple[datetime, datetime]]] = None
+        timespan: Optional[Union[timedelta, Tuple[datetime, datetime]]] = None
     ) -> List[SentinelQueryResult]:
         """Execute KQL query and return raw results.
 

mypy.ini

@@ -110,5 +110,14 @@ ignore_missing_imports = True
 [mypy-azure.kusto.*]
 ignore_missing_imports = True
 
+[mypy-azure.monitor.*]
+ignore_missing_imports = True
+
+[mypy-azure.identity.*]
+ignore_missing_imports = True
+
+[mypy-azure.core.*]
+ignore_missing_imports = True
+
 [mypy-requests.*]
 ignore_missing_imports = True