wizata-dsapi 2.0.0.dev17__tar.gz → 2.0.0.dev19__tar.gz

{wizata_dsapi-2.0.0.dev17/wizata_dsapi.egg-info → wizata_dsapi-2.0.0.dev19}/PKG-INFO

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

{wizata_dsapi-2.0.0.dev17 → wizata_dsapi-2.0.0.dev19}/wizata_dsapi/__init__.py

@@ -51,7 +51,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, \
-    NotificationPreference, RecipientList, Recipient, RecipientType
+    MonitoredPipeline, NotificationPreference, RecipientList, Recipient, RecipientType
 from .edge_device import EdgeDevice, EdgeDeviceStatus, EdgeDeviceType
 from .edge_module import EdgeModule, EdgeModuleType, WIZATA_MODULES
 from .edge_config import EdgeConsumer, EdgeWriter, EdgeTrigger

{wizata_dsapi-2.0.0.dev17 → wizata_dsapi-2.0.0.dev19}/wizata_dsapi/notification.py

@@ -19,7 +19,7 @@ RecipientList — named list of internal users (follow preferences) and free ent
                 (explicit address + channel).
 """
 
-from datetime import datetime
+from datetime import datetime, timezone
 from enum import Enum
 from typing import Optional
 import uuid
@@ -38,19 +38,13 @@ class DeviceToken:
 
     Primary key is ``(token, tenant_id)`` — one row per device per tenant.
 
-    Subscription model:
-    - ``scopes``: which event types to receive.
-                  Defaults to ``[PIPELINE_USER]``.
-    - ``twins``:  hardware IDs of twins to filter on.
-                  Empty list means all twins.
+    Pure token registration — scopes and twins are on NotificationPreference.
     """
 
     def __init__(self):
         self.token: Optional[str] = None
         self.tenant_id: Optional[str] = None
         self.user_id: Optional[str] = None
-        self.scopes: list = [NotificationScope.PIPELINE_USER]
-        self.twins: list = []
         self.platform: Optional[str] = None
         self.created_at: Optional[datetime] = None
 
@@ -66,8 +60,6 @@ class DeviceToken:
             obj["tenantId"] = self.tenant_id
         if self.user_id is not None:
             obj["userId"] = self.user_id
-        obj["scopes"] = [s.value if isinstance(s, NotificationScope) else s for s in self.scopes]
-        obj["twins"] = list(self.twins)
         if self.platform is not None:
             obj["platform"] = self.platform
         if self.created_at is not None:
@@ -80,14 +72,6 @@ class DeviceToken:
         obj.token = data.get("token")
         obj.tenant_id = data.get("tenantId") or data.get("tenant_id")
         obj.user_id = data.get("userId") or data.get("user_id")
-
-        raw_scopes = data.get("scopes")
-        if raw_scopes is not None:
-            obj.scopes = [NotificationScope(s.lower()) for s in raw_scopes]
-        else:
-            obj.scopes = [NotificationScope.PIPELINE_USER]
-
-        obj.twins = data.get("twins") or []
         obj.platform = data.get("platform")
 
         created_raw = data.get("createdAt") or data.get("created_at")
@@ -108,7 +92,7 @@ class DeviceToken:
     def __repr__(self) -> str:
         return (
             f"DeviceToken(tenant={self.tenant_id!r}, user={self.user_id!r}, "
-            f"scopes={self.scopes}, twins={self.twins!r})"
+            f"platform={self.platform!r})"
         )
 
 
@@ -354,6 +338,54 @@ class MonitoredDatapoint:
         return f"MonitoredDatapoint(hardware_id={self.hardware_id!r})"
 
 
+class MonitoredPipeline:
+    """
+    Registry entry for a pipeline flagged for monitoring.
+
+    Stored in PostgreSQL ``"MonitoredPipeline"`` table.
+    PK is ``key`` (pipeline key string, one row per pipeline).
+
+    The pipeline_monitor job reads this table and creates independent
+    AlertRule/AlertActive entries per event type (not_triggered, failed).
+    """
+
+    def __init__(self, key: str = None):
+        self.key: Optional[str] = key
+        self.created_at: Optional[datetime] = None
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.key is not None:
+            obj["key"] = self.key
+        if self.created_at is not None:
+            obj["createdAt"] = int(self.created_at.timestamp() * 1000)
+        return obj
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "MonitoredPipeline":
+        obj = cls()
+        obj.key = data.get("key")
+        raw = data.get("createdAt") or data.get("created_at")
+        if raw is not None:
+            if isinstance(raw, (int, float)):
+                obj.created_at = datetime.fromtimestamp(raw / 1000, tz=timezone.utc)
+            elif isinstance(raw, datetime):
+                obj.created_at = raw
+        return obj
+
+    def api_id(self) -> str:
+        return self.key
+
+    def api_key(self) -> str:
+        return self.key
+
+    def set_id(self, value):
+        self.key = str(value)
+
+    def __repr__(self) -> str:
+        return f"MonitoredPipeline(key={self.key!r})"
+
+
 class NotificationPreference:
     """
     Per-user notification channel preferences stored in PostgreSQL
@@ -370,6 +402,8 @@ class NotificationPreference:
         self.email_enabled: bool = False
         self.sms_enabled: bool = False
         self.whatsapp_enabled: bool = False
+        self.scopes: list = [NotificationScope.PIPELINE_USER]
+        self.twins: list = []
         self.created_at: Optional[datetime] = None
         self.updated_at: Optional[datetime] = None
 
@@ -400,6 +434,8 @@ class NotificationPreference:
         obj["emailEnabled"] = self.email_enabled
         obj["smsEnabled"] = self.sms_enabled
         obj["whatsappEnabled"] = self.whatsapp_enabled
+        obj["scopes"] = [s.value if isinstance(s, NotificationScope) else s for s in self.scopes]
+        obj["twins"] = list(self.twins)
         if self.created_at is not None:
             obj["createdAt"] = self.created_at.isoformat()
         if self.updated_at is not None:
@@ -415,6 +451,13 @@ class NotificationPreference:
         obj.sms_enabled = data.get("smsEnabled", data.get("sms_enabled", False))
         obj.whatsapp_enabled = data.get("whatsappEnabled", data.get("whatsapp_enabled", False))
 
+        raw_scopes = data.get("scopes")
+        if raw_scopes is not None:
+            obj.scopes = [NotificationScope(s.lower()) for s in raw_scopes]
+        else:
+            obj.scopes = [NotificationScope.PIPELINE_USER]
+        obj.twins = data.get("twins") or []
+
         for key in ("createdAt", "created_at"):
             raw = data.get(key)
             if raw is not None:
@@ -452,7 +495,10 @@ class NotificationPreference:
             channels.append("sms")
         if self.whatsapp_enabled:
             channels.append("whatsapp")
-        return f"NotificationPreference(user={self.user_id!r}, channels={channels})"
+        return (
+            f"NotificationPreference(user={self.user_id!r}, channels={channels}, "
+            f"scopes={[s.value for s in self.scopes]}, twins={self.twins!r})"
+        )
 
 
 class RecipientType(str, Enum):

wizata_dsapi-2.0.0.dev19/wizata_dsapi/version.py

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

{wizata_dsapi-2.0.0.dev17 → wizata_dsapi-2.0.0.dev19/wizata_dsapi.egg-info}/PKG-INFO

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

wizata_dsapi-2.0.0.dev17/wizata_dsapi/version.py

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