wnm 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl

wnm/actions.py

@@ -0,0 +1,45 @@
+"""Action model for representing planned node operations.
+
+This module defines the action types and data structures used by the decision engine
+to represent planned operations on nodes.
+"""
+
+from dataclasses import dataclass
+from enum import Enum
+from typing import Optional
+
+
+class ActionType(Enum):
+    """Types of actions that can be performed on nodes."""
+
+    ADD_NODE = "add"
+    REMOVE_NODE = "remove"
+    UPGRADE_NODE = "upgrade"
+    START_NODE = "start"
+    STOP_NODE = "stop"
+    RESTART_NODE = "restart"
+    SURVEY_NODES = "survey"
+    RESURVEY_NODES = "resurvey"
+
+
+@dataclass
+class Action:
+    """Represents a planned action to be executed.
+
+    Attributes:
+        type: The type of action to perform
+        node_id: Optional node ID if action is node-specific
+        priority: Higher values indicate more urgent actions (default: 0)
+        reason: Human-readable explanation of why this action is needed
+    """
+
+    type: ActionType
+    node_id: Optional[int] = None
+    priority: int = 0
+    reason: str = ""
+
+    def __repr__(self) -> str:
+        """Return a string representation of the action."""
+        if self.node_id is not None:
+            return f"Action({self.type.value}, node={self.node_id}, reason={self.reason})"
+        return f"Action({self.type.value}, reason={self.reason})"

wnm/common.py

@@ -0,0 +1,21 @@
+# Primary node for want of one
+QUEEN = 1
+
+# Donation address, default to faucet vault, can be overridden in config
+DONATE = "0x00455d78f850b0358E8cea5be24d415E01E107CF"
+
+# Keep these as strings so they can be grepped in logs
+STOPPED = "STOPPED"  # 0 Node is not responding to it's metrics port
+RUNNING = "RUNNING"  # 1 Node is responding to it's metrics port
+UPGRADING = "UPGRADING"  # 2 Upgrade in progress
+DISABLED = "DISABLED"  # -1 Do not start
+RESTARTING = "RESTARTING"  # 3 re/starting a server intionally
+MIGRATING = "MIGRATING"  # 4 Moving volumes in progress
+REMOVING = "REMOVING"  # 5 Removing node in progress
+DEAD = "DEAD"  # -86 Broken node to cleanup
+
+# Magic numbers extracted from codebase
+MIN_NODES_THRESHOLD = 0  # Minimum nodes before considering actions
+PORT_MULTIPLIER = 1000  # Port calculation: PortStart * 1000 + node_id
+METRICS_PORT_BASE = 13000  # Metrics port calculation: 13000 + node_id
+DEFAULT_CRISIS_BYTES = 2 * 10**9  # Default crisis threshold in bytes (2GB)