foodforthought-cli 0.2.4__py3-none-any.whl → 0.2.8__py3-none-any.whl

ate/recording/wrapper.py

@@ -0,0 +1,95 @@
+"""
+Recording wrapper for automatic telemetry capture.
+
+Provides a simpler API than RecordingSession for cases where you
+want to wrap a driver and record everything automatically.
+"""
+
+from typing import Optional
+from ate.interfaces import RobotInterface
+from .session import RecordingSession
+
+
+class RecordingWrapper:
+    """
+    Wrapper that automatically records all driver interactions.
+
+    Unlike RecordingSession (context manager), RecordingWrapper creates
+    a persistent recording that captures everything until explicitly stopped.
+
+    Usage:
+        # Wrap a driver
+        raw_driver = MechDogDriver(port="/dev/...")
+        driver = RecordingWrapper(raw_driver, name="session_01")
+
+        # Use driver normally - all calls are recorded
+        driver.connect()
+        driver.stand()
+        driver.walk(Vector3.forward(), speed=0.3)
+        driver.stop()
+        driver.disconnect()
+
+        # Save the recording
+        driver.save_recording("session_01.demonstration")
+    """
+
+    def __init__(
+        self,
+        driver: RobotInterface,
+        name: str = "recording",
+        description: Optional[str] = None,
+        auto_start: bool = True,
+    ):
+        """
+        Initialize recording wrapper.
+
+        Args:
+            driver: Robot driver to wrap
+            name: Recording name
+            description: Optional description
+            auto_start: Start recording immediately
+        """
+        self._driver = driver
+        self._session = RecordingSession(
+            driver,
+            name=name,
+            description=description,
+        )
+
+        if auto_start:
+            self._session.start()
+
+    def __getattr__(self, name: str):
+        """Forward attribute access to wrapped driver."""
+        return getattr(self._driver, name)
+
+    def start_recording(self) -> None:
+        """Start recording (if not already)."""
+        self._session.start()
+
+    def stop_recording(self) -> None:
+        """Stop recording."""
+        self._session.stop()
+
+    def save_recording(self, path: str) -> None:
+        """Save recording to file."""
+        self._session.save(path)
+
+    def get_recording_summary(self) -> str:
+        """Get recording summary."""
+        return self._session.summary()
+
+    @property
+    def is_recording(self) -> bool:
+        """Check if currently recording."""
+        return self._session.is_recording
+
+    @property
+    def recording_duration(self) -> Optional[float]:
+        """Get recording duration in seconds."""
+        return self._session.duration
+
+    @property
+    def call_count(self) -> int:
+        """Get number of recorded calls."""
+        return len(self._session.calls)

ate/robot/__init__.py

@@ -0,0 +1,79 @@
+"""
+Robot management system for ATE.
+
+Provides:
+- Auto-discovery of robots on network and USB
+- Robot profiles for easy configuration
+- Capability introspection
+- Interactive setup wizard
+
+Example usage:
+    from ate.robot import RobotManager, discover_robots
+
+    # Auto-discover robots
+    found = discover_robots()
+    # > [DiscoveredRobot(type='mechdog', port='/dev/cu.usbserial-10', camera_ip='192.168.1.100')]
+
+    # Load saved profile
+    manager = RobotManager()
+    dog = manager.get_robot("my_mechdog")
+
+    # Introspect capabilities
+    info = manager.get_capabilities(dog)
+    # > {'locomotion': ['walk', 'turn', 'stand'], 'camera': ['get_image']}
+"""
+
+from .discovery import (
+    DiscoveredRobot,
+    discover_robots,
+    discover_serial_robots,
+    discover_network_cameras,
+)
+
+from .profiles import (
+    RobotProfile,
+    load_profile,
+    save_profile,
+    list_profiles,
+    delete_profile,
+)
+
+from .registry import (
+    RobotRegistry,
+    KNOWN_ROBOTS,
+    get_robot_info,
+)
+
+from .introspection import (
+    get_capabilities,
+    get_methods,
+    test_capability,
+)
+
+from .manager import (
+    RobotManager,
+)
+
+__all__ = [
+    # Discovery
+    "DiscoveredRobot",
+    "discover_robots",
+    "discover_serial_robots",
+    "discover_network_cameras",
+    # Profiles
+    "RobotProfile",
+    "load_profile",
+    "save_profile",
+    "list_profiles",
+    "delete_profile",
+    # Registry
+    "RobotRegistry",
+    "KNOWN_ROBOTS",
+    "get_robot_info",
+    # Introspection
+    "get_capabilities",
+    "get_methods",
+    "test_capability",
+    # Manager
+    "RobotManager",
+]