Fixing azure_monitor_driver for deprecated httpx API (#809)

* Fixing azure_monitor_driver to handle deprecated
httpx API call.
Also fixed a doc bug in process_tree_utils and initializer in host_logon_session.

* Update typing

* Replace hardcoded value

---------

Co-authored-by: Florian BRACQ <[email protected]>
Co-authored-by: FlorianBracq <[email protected]>

Author: ianhelle

msticpy/data/drivers/azure_monitor_driver.py

@@ -24,7 +24,7 @@
 import pandas as pd
 from azure.core.exceptions import HttpResponseError
 from azure.core.pipeline.policies import UserAgentPolicy
-from packaging.version import parse as parse_version
+from packaging.version import Version, parse as parse_version
 
 from ..._version import VERSION
 from ...auth.azure_auth import AzureCloudConfig, az_connect
@@ -605,13 +605,25 @@ def _get_schema(self) -> Dict[str, Dict]:
         token = credentials.modern.get_token(f"{mgmt_endpoint}/.default")
         headers = {"Authorization": f"Bearer {token.token}", **mp_ua_header()}
         logger.info("Schema request to %s", fmt_url)
-        response = httpx.get(
-            fmt_url,
-            headers=headers,
+
+        # Handle proxies (parameter changes in httpx 0.25.0)
+        httpx_version: Version = parse_version(httpx.__version__)
+        proxies: dict[str, str] = self._def_proxies or {}
+        httpx_proxy_kwargs: dict[str, Any] = {}
+        if proxies:
+            if httpx_version < parse_version("0.25.0"):
+                httpx_proxy_kwargs = {"proxies": proxies}
+            else:
+                httpx_proxy_kwargs = {"mounts": proxies}
+        with httpx.Client(
             timeout=get_http_timeout(),
-            proxies=self._def_proxies or {},
-        )
-        if response.status_code != 200:
+            **httpx_proxy_kwargs,
+        ) as httpx_client:
+            response: httpx.Response = httpx_client.get(
+                fmt_url,
+                headers=headers,
+            )
+        if response.status_code != httpx.codes.OK:
             logger.info("Schema request failed. Status code: %d", response.status_code)
             return {}
         tables = response.json()

msticpy/datamodel/entities/host_logon_session.py

@@ -75,7 +75,7 @@ def __init__(
         self.StartTimeUtc: datetime = datetime.min
         self.EndTimeUtc: datetime = datetime.min
         self.Host: Optional[Host] = None
-        self.SessionId: str | None = ""
+        self.SessionId: str | None = None
         super().__init__(src_entity=src_entity, **kwargs)
 
         if src_event is not None:

msticpy/transform/process_tree_utils.py

@@ -423,7 +423,7 @@ def tree_to_text(
     Raises
     ------
     ValueError
-        If neither of
+        If neither schema nor template are supplied as parameters.
 
     """
     if not schema and not template: