htcli 1.1.0__py3-none-any.whl

src/htcli/models/responses/config.py

@@ -0,0 +1,58 @@
+"""
+Config-related response models.
+"""
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+
+class ConfigInitResponse(BaseModel):
+    """Response model for config initialization."""
+
+    config_path: str = Field(..., description="Configuration file path")
+    created: bool = Field(..., description="Whether config was created")
+    message: str = Field(..., description="Initialization message")
+
+
+class ConfigShowResponse(BaseModel):
+    """Response model for config show."""
+
+    config_data: dict[str, Any] = Field(..., description="Config data")
+    config_path: str = Field(..., description="Config file path")
+    exists: bool = Field(..., description="Whether config file exists")
+
+
+class ConfigValidateResponse(BaseModel):
+    """Response model for config validation."""
+
+    config_path: str = Field(..., description="Configuration file path")
+    valid: bool = Field(..., description="Configuration validation status")
+    errors: list[str] = Field(default=[], description="Validation errors")
+    message: str = Field(..., description="Validation result message")
+
+
+class ConfigSetResponse(BaseModel):
+    """Response model for config set."""
+
+    config_path: str = Field(..., description="Configuration file path")
+    key: str = Field(..., description="Configuration key")
+    old_value: Any = Field(..., description="Previous value")
+    new_value: Any = Field(..., description="New value")
+    message: str = Field(..., description="Result message")
+
+
+class ConfigGetResponse(BaseModel):
+    """Response model for config get."""
+
+    key: str = Field(..., description="Configuration key")
+    value: Any = Field(..., description="Configuration value")
+    value_type: str = Field(..., description="Value type")
+
+
+class ConfigPathResponse(BaseModel):
+    """Response model for config path."""
+
+    config_path: str = Field(..., description="Configuration file path")
+    exists: bool = Field(..., description="Whether config file exists")
+    absolute_path: str = Field(..., description="Absolute path to config file")

src/htcli/models/responses/identity.py

@@ -0,0 +1,102 @@
+"""
+Identity-related response models.
+"""
+
+from typing import Optional
+
+from pydantic import BaseModel, Field, field_validator
+from scalecodec.base import RuntimeConfiguration, ScaleBytes
+from substrateinterface.base import load_type_registry_preset
+
+from .base import BaseResponse
+
+
+class IdentityInfo(BaseModel):
+    """Identity information structure with validation matching Rust ColdkeyIdentityData."""
+
+    coldkey: str = Field(..., description="Coldkey account ID")
+    hotkey: str = Field(..., description="Hotkey account ID")
+    name: bytes = Field(..., description="Identity name")
+    url: bytes = Field(default=b"", description="Identity URL")
+    image: bytes = Field(default=b"", description="Identity image URL")
+    discord: bytes = Field(default=b"", description="Discord handle")
+    x: bytes = Field(default=b"", description="X (Twitter) handle")
+    telegram: bytes = Field(default=b"", description="Telegram handle")
+    github: bytes = Field(default=b"", description="GitHub URL")
+    hugging_face: bytes = Field(default=b"", description="Hugging Face URL")
+    description: bytes = Field(default=b"", description="Identity description")
+    misc: bytes = Field(default=b"", description="Miscellaneous information")
+
+    @field_validator(
+        "name",
+        "url",
+        "image",
+        "discord",
+        "x",
+        "telegram",
+        "github",
+        "hugging_face",
+        "description",
+        "misc",
+    )
+    def validate_bytes_not_empty(cls, v):
+        """Ensure byte fields are not None."""
+        if v is None:
+            return b""
+        return v
+
+    @classmethod
+    def from_scale_bytes(
+        cls, data: bytes, coldkey: str = "", hotkey: str = ""
+    ) -> "IdentityInfo":
+        """Decode from SCALE bytes using scalecodec."""
+        if not ScaleBytes:
+            raise ImportError("scalecodec is required for decoding")
+
+        # Load type registry
+        type_registry = load_type_registry_preset("substrate-node-template")
+
+        # Create runtime configuration
+        runtime_config = RuntimeConfiguration()
+        runtime_config.update_type_registry(type_registry)
+
+        # Decode the data (ColdkeyIdentityData structure)
+        obj = runtime_config.create_scale_object(
+            "ColdkeyIdentityData", ScaleBytes(data)
+        )
+        obj.decode()
+
+        return cls(
+            coldkey=coldkey,
+            hotkey=hotkey,
+            name=obj["name"],
+            url=obj["url"],
+            image=obj["image"],
+            discord=obj["discord"],
+            x=obj["x"],
+            telegram=obj["telegram"],
+            github=obj["github"],
+            hugging_face=obj["hugging_face"],
+            description=obj["description"],
+            misc=obj["misc"],
+        )
+
+
+class IdentityRegisterResponse(BaseResponse):
+    """Response model for identity registration."""
+
+    identity_name: Optional[str] = Field(None, description="Registered identity name")
+    coldkey: Optional[str] = Field(None, description="Coldkey account ID")
+
+
+class IdentityRemoveResponse(BaseResponse):
+    """Response model for identity removal."""
+
+    identity_name: Optional[str] = Field(None, description="Removed identity name")
+    coldkey: Optional[str] = Field(None, description="Coldkey account ID")
+
+
+class IdentityAcceptResponse(BaseResponse):
+    """Response model for identity acceptance."""
+
+    identity: Optional[str] = Field(None, description="Accepted identity data")

src/htcli/models/responses/overwatch.py

@@ -0,0 +1,51 @@
+from typing import Any, Optional
+
+from pydantic import BaseModel, Field, field_validator
+from scalecodec import ScaleBytes
+from scalecodec.base import RuntimeConfiguration
+from scalecodec.type_registry import load_type_registry_preset
+
+
+class OverwatchNodeInfo(BaseModel):
+    """Overwatch node information structure matching Rust OverwatchNodeInfo with validation."""
+
+    overwatch_node_id: int = Field(..., description="Overwatch node ID", ge=0)
+    coldkey: str = Field(..., description="Coldkey account ID")
+    hotkey: str = Field(..., description="Hotkey account ID")
+    peer_ids: dict[int, bytes] = Field(default_factory=dict, description="Peer IDs by subnet")
+    penalties: int = Field(default=0, description="Penalties", ge=0)
+    stake_balance: int = Field(default=0, description="Stake balance in wei", ge=0)
+    reputation: Optional[dict[str, Any]] = Field(None, description="Reputation data")
+
+    @field_validator("peer_ids")
+    def validate_peer_ids(cls, v):
+        """Ensure peer IDs are valid."""
+        if v is None:
+            return {}
+        return v
+
+    @classmethod
+    def from_scale_bytes(cls, data: bytes) -> "OverwatchNodeInfo":
+        """Decode from SCALE bytes using scalecodec."""
+        if not ScaleBytes:
+            raise ImportError("scalecodec is required for decoding")
+
+        # Load type registry
+        type_registry = load_type_registry_preset("substrate-node-template")
+
+        # Create runtime configuration
+        runtime_config = RuntimeConfiguration()
+        runtime_config.update_type_registry(type_registry)
+
+        # Decode the data
+        obj = runtime_config.create_scale_object("OverwatchNodeInfo", ScaleBytes(data))
+        obj.decode()
+
+        return cls(
+            overwatch_node_id=obj["overwatch_node_id"],
+            coldkey=obj["coldkey"],
+            hotkey=obj["hotkey"],
+            peer_ids=list(obj["peer_ids"]),
+            penalties=obj["penalties"],
+            reputation=obj["reputation"],
+        )