wizata-dsapi 2.0.0.dev8__tar.gz → 2.0.0.dev9__tar.gz

{wizata_dsapi-2.0.0.dev8/wizata_dsapi.egg-info → wizata_dsapi-2.0.0.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev8
+Version: 2.0.0.dev9
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.0.0.dev8 → wizata_dsapi-2.0.0.dev9}/wizata_dsapi/__init__.py

@@ -52,6 +52,7 @@ from .streamlit_utils import (get_streamlit_token, get_streamlit_domain, get_str
 from .search import FilterOperator, SearchQuery, SortOption, GroupSearchQuery
 from .notification import NotificationScope, DeviceToken, AlertRule, AlertActive, MonitoredDatapoint
 from .edge_device import EdgeDevice, EdgeDeviceStatus, EdgeDeviceType
+from .edge_module import EdgeModule, EdgeModuleType, WIZATA_MODULES
 from .dashboard import Dashboard, DashboardType
 from .mobile_asset import MobileAssetTile, MobileAssetTileType, MobileAssetConfig
 from .graylog_log import GraylogLog

{wizata_dsapi-2.0.0.dev8 → wizata_dsapi-2.0.0.dev9}/wizata_dsapi/edge_device.py

@@ -35,6 +35,8 @@ class EdgeDevice(ApiDto):
     :ivar EdgeDeviceStatus status: current status of the device.
     :ivar int status_code: numeric status code.
     :ivar EdgeDeviceType device_type: type of the edge device.
+    :ivar str connection_state: IoT Hub connection state ('Connected' or 'Disconnected'), enriched from IoT Hub.
+    :ivar str last_activity_time: last device activity timestamp from IoT Hub.
     """
 
     @classmethod
@@ -55,7 +57,9 @@ class EdgeDevice(ApiDto):
                  twin_id: uuid.UUID = None,
                  status: EdgeDeviceStatus = None,
                  status_code: int = None,
-                 device_type: EdgeDeviceType = None):
+                 device_type: EdgeDeviceType = None,
+                 connection_state: str = None,
+                 last_activity_time: str = None):
         if edge_device_id is None:
             self.edge_device_id = uuid.uuid4()
         else:
@@ -67,6 +71,8 @@ class EdgeDevice(ApiDto):
         self.status = status
         self.status_code = status_code
         self.device_type = device_type
+        self.connection_state = connection_state
+        self.last_activity_time = last_activity_time
 
     def api_id(self) -> str:
         return str(self.edge_device_id).upper()
@@ -97,6 +103,10 @@ class EdgeDevice(ApiDto):
             self.status_code = int(obj["statusCode"])
         if "deviceType" in obj and obj["deviceType"] is not None:
             self.device_type = EdgeDeviceType(obj["deviceType"])
+        if "connectionState" in obj and obj["connectionState"] is not None:
+            self.connection_state = obj["connectionState"]
+        if "lastActivityTime" in obj and obj["lastActivityTime"] is not None:
+            self.last_activity_time = obj["lastActivityTime"]
 
     def to_json(self, target: str = None):
         obj = {
@@ -116,4 +126,8 @@ class EdgeDevice(ApiDto):
             obj["statusCode"] = self.status_code
         if self.device_type is not None:
             obj["deviceType"] = self.device_type.value
+        if self.connection_state is not None:
+            obj["connectionState"] = self.connection_state
+        if self.last_activity_time is not None:
+            obj["lastActivityTime"] = self.last_activity_time
         return obj

wizata_dsapi-2.0.0.dev9/wizata_dsapi/edge_module.py

@@ -0,0 +1,115 @@
+from enum import Enum
+from .api_dto import Dto
+
+
+class EdgeModuleType(Enum):
+    """
+    Classification of an edge module.
+
+    WIZATA: modules developed and maintained by Wizata (configsync, message_client, pipeline_runner).
+    SYSTEM: IoT Edge runtime and third-party infrastructure (edgeAgent, edgeHub, influxdb, rabbitmq, loki, ...).
+    """
+    WIZATA = "wizata"
+    SYSTEM = "system"
+
+
+# Hardcoded registry of known Wizata modules.
+# Maps module_id to a human-readable description.
+WIZATA_MODULES = {
+    "configsync": "Configuration synchronization between cloud and edge",
+    "message_client": "Message client for edge-to-cloud telemetry",
+    "pipeline_runner": "Pipeline execution engine on edge",
+}
+
+
+def _extract_image_version(image: str) -> str:
+    """Extract the tag portion after ':' from a container image reference."""
+    if image and ":" in image:
+        return image.rsplit(":", 1)[1]
+    return None
+
+
+class EdgeModule(Dto):
+    """
+    Represents a module deployed on an IoT Edge device.
+
+    :ivar str module_id: module identifier (e.g. 'configsync', '$edgeAgent').
+    :ivar str runtime_status: runtime status from $edgeAgent reported twin (e.g. 'running', 'stopped').
+    :ivar int exit_code: last exit code of the module container.
+    :ivar str last_start_time: last container start time (UTC).
+    :ivar str last_exit_time: last container exit time (UTC).
+    :ivar str image: container image reference.
+    :ivar str image_version: tag extracted from image (e.g. 'p12-20260313.3'), only set for wizata modules.
+    :ivar str last_updated: when this module's status was last reported to IoT Hub (from $edgeAgent twin $metadata).
+    :ivar EdgeModuleType module_type: wizata or system classification.
+    """
+
+    def __init__(self,
+                 module_id: str = None,
+                 runtime_status: str = None,
+                 exit_code: int = None,
+                 last_start_time: str = None,
+                 last_exit_time: str = None,
+                 image: str = None,
+                 image_version: str = None,
+                 last_updated: str = None,
+                 module_type: EdgeModuleType = None):
+        self.module_id = module_id
+        self.runtime_status = runtime_status
+        self.exit_code = exit_code
+        self.last_start_time = last_start_time
+        self.last_exit_time = last_exit_time
+        self.image = image
+        self.last_updated = last_updated
+        if module_type is not None:
+            self.module_type = module_type
+        elif module_id is not None:
+            self.module_type = EdgeModuleType.WIZATA if module_id in WIZATA_MODULES else EdgeModuleType.SYSTEM
+        else:
+            self.module_type = None
+        # image_version: explicit value wins, otherwise extract for wizata modules only
+        if image_version is not None:
+            self.image_version = image_version
+        elif self.module_type == EdgeModuleType.WIZATA and image:
+            self.image_version = _extract_image_version(image)
+        else:
+            self.image_version = None
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "EdgeModule":
+        module_type = None
+        if "moduleType" in data and data["moduleType"] is not None:
+            module_type = EdgeModuleType(data["moduleType"])
+        return cls(
+            module_id=data.get("moduleId"),
+            runtime_status=data.get("runtimeStatus"),
+            exit_code=data.get("exitCode"),
+            last_start_time=data.get("lastStartTime"),
+            last_exit_time=data.get("lastExitTime"),
+            image=data.get("image"),
+            image_version=data.get("imageVersion"),
+            last_updated=data.get("lastUpdated"),
+            module_type=module_type,
+        )
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.module_id is not None:
+            obj["moduleId"] = self.module_id
+        if self.runtime_status is not None:
+            obj["runtimeStatus"] = self.runtime_status
+        if self.exit_code is not None:
+            obj["exitCode"] = self.exit_code
+        if self.last_start_time is not None:
+            obj["lastStartTime"] = self.last_start_time
+        if self.last_exit_time is not None:
+            obj["lastExitTime"] = self.last_exit_time
+        if self.image is not None:
+            obj["image"] = self.image
+        if self.image_version is not None:
+            obj["imageVersion"] = self.image_version
+        if self.last_updated is not None:
+            obj["lastUpdated"] = self.last_updated
+        if self.module_type is not None:
+            obj["moduleType"] = self.module_type.value
+        return obj

wizata_dsapi-2.0.0.dev9/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.0.0.dev9"

{wizata_dsapi-2.0.0.dev8 → wizata_dsapi-2.0.0.dev9/wizata_dsapi.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.0.0.dev8
+Version: 2.0.0.dev9
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.0.0.dev8 → wizata_dsapi-2.0.0.dev9}/wizata_dsapi.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ wizata_dsapi/datastore.py
 wizata_dsapi/ds_dataframe.py
 wizata_dsapi/dsapi_json_encoder.py
 wizata_dsapi/edge_device.py
+wizata_dsapi/edge_module.py
 wizata_dsapi/evaluation.py
 wizata_dsapi/execution.py
 wizata_dsapi/execution_log.py

wizata_dsapi-2.0.0.dev8/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.0.0.dev8"