first commit

exectrace/core/models.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any, Dict, List
+
+from exectrace.utils.time_utils import utc_now_iso
+
+
+@dataclass
+class Action:
+    action_type: str
+    timestamp: str
+    payload: Dict[str, Any] = field(default_factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "action_type": self.action_type,
+            "timestamp": self.timestamp,
+            "payload": self.payload,
+        }
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "Action":
+        return cls(
+            action_type=data["action_type"],
+            timestamp=data["timestamp"],
+            payload=data.get("payload", {}),
+        )
+
+
+@dataclass
+class Workflow:
+    name: str
+    version: str = "1.0"
+    created_at: str = field(default_factory=utc_now_iso)
+    updated_at: str = field(default_factory=utc_now_iso)
+    actions: List[Action] = field(default_factory=list)
+
+    def add_action(self, action_type: str, payload: Dict[str, Any]) -> None:
+        self.actions.append(Action(action_type=action_type, timestamp=utc_now_iso(), payload=payload))
+        self.updated_at = utc_now_iso()
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "name": self.name,
+            "version": self.version,
+            "created_at": self.created_at,
+            "updated_at": self.updated_at,
+            "actions": [action.to_dict() for action in self.actions],
+        }
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "Workflow":
+        return cls(
+            name=data["name"],
+            version=data.get("version", "1.0"),
+            created_at=data.get("created_at", utc_now_iso()),
+            updated_at=data.get("updated_at", utc_now_iso()),
+            actions=[Action.from_dict(item) for item in data.get("actions", [])],
+        )