conson-xp 1.34.0__py3-none-any.whl → 1.35.0__py3-none-any.whl

xp/services/conbus/conbus_datapoint_service.py

@@ -5,48 +5,57 @@ This service handles datapoint query operations for modules through Conbus teleg
 
 import logging
 from datetime import datetime
-from typing import Callable, Optional
+from typing import Any, Optional
 
-from twisted.internet.posixbase import PosixReactorBase
+from psygnal import Signal
 
-from xp.models import ConbusClientConfig, ConbusDatapointResponse
+from xp.models import ConbusDatapointResponse
 from xp.models.protocol.conbus_protocol import TelegramReceivedEvent
 from xp.models.telegram.datapoint_type import DataPointType
 from xp.models.telegram.reply_telegram import ReplyTelegram
 from xp.models.telegram.system_function import SystemFunction
 from xp.models.telegram.telegram_type import TelegramType
-from xp.services.protocol import ConbusProtocol
+from xp.services.protocol.conbus_event_protocol import ConbusEventProtocol
 from xp.services.telegram.telegram_service import TelegramService
 
 
-class ConbusDatapointService(ConbusProtocol):
-    """
-    Service for querying datapoints from Conbus modules.
+class ConbusDatapointService:
+    """Service for querying datapoints from Conbus modules.
 
-    Uses ConbusProtocol to provide datapoint query functionality
+    Uses ConbusEventProtocol to provide datapoint query functionality
     for reading sensor data and module information.
+
+    Attributes:
+        conbus_protocol: Protocol instance for Conbus communication.
+        telegram_service: Service for parsing telegrams.
+        on_finish: Signal emitted when datapoint query completes (with response).
     """
 
+    on_finish: Signal = Signal(ConbusDatapointResponse)
+
     def __init__(
         self,
+        conbus_protocol: ConbusEventProtocol,
         telegram_service: TelegramService,
-        cli_config: ConbusClientConfig,
-        reactor: PosixReactorBase,
     ) -> None:
         """Initialize the Conbus datapoint service.
 
         Args:
+            conbus_protocol: Protocol instance for Conbus communication.
             telegram_service: Service for parsing telegrams.
-            cli_config: Configuration for Conbus client connection.
-            reactor: Twisted reactor for event loop.
         """
-        super().__init__(cli_config, reactor)
+        self.conbus_protocol = conbus_protocol
         self.telegram_service = telegram_service
+
+        # Connect protocol signals
+        self.conbus_protocol.on_connection_made.connect(self.connection_made)
+        self.conbus_protocol.on_telegram_sent.connect(self.telegram_sent)
+        self.conbus_protocol.on_telegram_received.connect(self.telegram_received)
+        self.conbus_protocol.on_timeout.connect(self.timeout)
+        self.conbus_protocol.on_failed.connect(self.failed)
+
         self.serial_number: str = ""
         self.datapoint_type: Optional[DataPointType] = None
-        self.datapoint_finished_callback: Optional[
-            Callable[[ConbusDatapointResponse], None]
-        ] = None
         self.service_response: ConbusDatapointResponse = ConbusDatapointResponse(
             success=False,
             serial_number=self.serial_number,
@@ -55,7 +64,7 @@ class ConbusDatapointService(ConbusProtocol):
         # Set up logging
         self.logger = logging.getLogger(__name__)
 
-    def connection_established(self) -> None:
+    def connection_made(self) -> None:
         """Handle connection established event."""
         self.logger.debug(
             f"Connection established, querying datapoint {self.datapoint_type}."
@@ -64,7 +73,7 @@ class ConbusDatapointService(ConbusProtocol):
             self.failed("Datapoint type not set")
             return
 
-        self.send_telegram(
+        self.conbus_protocol.send_telegram(
             telegram_type=TelegramType.SYSTEM,
             serial_number=self.serial_number,
             system_function=SystemFunction.READ_DATAPOINT,
@@ -128,9 +137,14 @@ class ConbusDatapointService(ConbusProtocol):
         self.service_response.datapoint_type = self.datapoint_type
         self.service_response.datapoint_telegram = datapoint_telegram
         self.service_response.data_value = datapoint_telegram.data_value
-        if self.datapoint_finished_callback:
-            self.datapoint_finished_callback(self.service_response)
-        self._stop_reactor()
+
+        # Emit finish signal
+        self.on_finish.emit(self.service_response)
+        self.stop_reactor()
+
+    def timeout(self) -> None:
+        """Handle timeout event."""
+        self.failed("Timeout")
 
     def failed(self, message: str) -> None:
         """Handle failed connection event.
@@ -143,14 +157,14 @@ class ConbusDatapointService(ConbusProtocol):
         self.service_response.timestamp = datetime.now()
         self.service_response.serial_number = self.serial_number
         self.service_response.error = message
-        if self.datapoint_finished_callback:
-            self.datapoint_finished_callback(self.service_response)
+
+        # Emit finish signal
+        self.on_finish.emit(self.service_response)
 
     def query_datapoint(
         self,
         serial_number: str,
         datapoint_type: DataPointType,
-        finish_callback: Callable[[ConbusDatapointResponse], None],
         timeout_seconds: Optional[float] = None,
     ) -> None:
         """Query a specific datapoint from a module.
@@ -158,13 +172,47 @@ class ConbusDatapointService(ConbusProtocol):
         Args:
             serial_number: 10-digit module serial number.
             datapoint_type: Type of datapoint to query.
-            finish_callback: Callback function to call when the datapoint is received.
             timeout_seconds: Timeout in seconds.
         """
         self.logger.info("Starting query_datapoint")
         if timeout_seconds:
-            self.timeout_seconds = timeout_seconds
-        self.datapoint_finished_callback = finish_callback
+            self.conbus_protocol.timeout_seconds = timeout_seconds
         self.serial_number = serial_number
         self.datapoint_type = datapoint_type
-        self.start_reactor()
+
+    def set_timeout(self, timeout_seconds: float) -> None:
+        """Set operation timeout.
+
+        Args:
+            timeout_seconds: Timeout in seconds.
+        """
+        self.conbus_protocol.timeout_seconds = timeout_seconds
+
+    def start_reactor(self) -> None:
+        """Start the reactor."""
+        self.conbus_protocol.start_reactor()
+
+    def stop_reactor(self) -> None:
+        """Stop the reactor."""
+        self.conbus_protocol.stop_reactor()
+
+    def __enter__(self) -> "ConbusDatapointService":
+        """Enter context manager - reset state for singleton reuse.
+
+        Returns:
+            Self for context manager protocol.
+        """
+        self.datapoint_response = ConbusDatapointResponse(success=False)
+        return self
+
+    def __exit__(
+        self, _exc_type: Optional[type], _exc_val: Optional[Exception], _exc_tb: Any
+    ) -> None:
+        """Exit context manager and disconnect signals."""
+        self.conbus_protocol.on_connection_made.disconnect(self.connection_made)
+        self.conbus_protocol.on_telegram_sent.disconnect(self.telegram_sent)
+        self.conbus_protocol.on_telegram_received.disconnect(self.telegram_received)
+        self.conbus_protocol.on_timeout.disconnect(self.timeout)
+        self.conbus_protocol.on_failed.disconnect(self.failed)
+        self.on_finish.disconnect()
+        self.stop_reactor()

xp/services/conbus/conbus_output_service.py

@@ -6,18 +6,17 @@ and processing ACK/NAK responses.
 
 import logging
 from datetime import datetime
-from typing import Callable, Optional
+from typing import Any, Optional
 
-from twisted.internet.posixbase import PosixReactorBase
+from psygnal import Signal
 
-from xp.models import ConbusClientConfig
 from xp.models.conbus.conbus_output import ConbusOutputResponse
 from xp.models.protocol.conbus_protocol import TelegramReceivedEvent
 from xp.models.telegram.action_type import ActionType
 from xp.models.telegram.output_telegram import OutputTelegram
 from xp.models.telegram.system_function import SystemFunction
 from xp.models.telegram.telegram_type import TelegramType
-from xp.services.protocol import ConbusProtocol
+from xp.services.protocol.conbus_event_protocol import ConbusEventProtocol
 from xp.services.telegram.telegram_output_service import (
     TelegramOutputService,
     XPOutputError,
@@ -30,33 +29,47 @@ class ConbusOutputError(Exception):
     pass
 
 
-class ConbusOutputService(ConbusProtocol):
+class ConbusOutputService:
     """
     Service for sending action telegrams to Conbus module outputs.
 
     Manages action telegram transmission (ON/OFF) and processes
     ACK/NAK responses from modules.
+
+    Attributes:
+        conbus_protocol: Protocol instance for Conbus communication.
+        on_finish: Signal emitted when operation finishes (with result).
     """
 
+    conbus_protocol: ConbusEventProtocol
+    on_finish: Signal = Signal(ConbusOutputResponse)
+
     def __init__(
         self,
+        conbus_protocol: ConbusEventProtocol,
         telegram_output_service: TelegramOutputService,
-        cli_config: ConbusClientConfig,
-        reactor: PosixReactorBase,
     ):
         """Initialize the Conbus output service.
 
         Args:
+            conbus_protocol: ConbusEventProtocol for communication.
             telegram_output_service: TelegramOutputService for telegram generation/parsing.
-            cli_config: Conbus client configuration.
-            reactor: Twisted reactor for async operations.
         """
-        super().__init__(cli_config, reactor)
+        self.conbus_protocol = conbus_protocol
         self.telegram_output_service = telegram_output_service
+
+        # Connect protocol signals
+        self.conbus_protocol.on_connection_made.connect(self.connection_made)
+        self.conbus_protocol.on_telegram_sent.connect(self.telegram_sent)
+        self.conbus_protocol.on_telegram_received.connect(self.telegram_received)
+        self.conbus_protocol.on_timeout.connect(self.timeout)
+        self.conbus_protocol.on_failed.connect(self.failed)
+
+        # Initialize state
         self.serial_number: str = ""
         self.output_number: int = 0
         self.action_type: ActionType = ActionType.ON_RELEASE
-        self.finish_callback: Optional[Callable[[ConbusOutputResponse], None]] = None
+        self.output_state: str = ""
         self.service_response: ConbusOutputResponse = ConbusOutputResponse(
             success=False,
             serial_number=self.serial_number,
@@ -68,7 +81,7 @@ class ConbusOutputService(ConbusProtocol):
         # Set up logging
         self.logger = logging.getLogger(__name__)
 
-    def connection_established(self) -> None:
+    def connection_made(self) -> None:
         """Handle connection established event."""
         self.logger.debug(
             f"Connection established, sending action {self.action_type} to output {self.output_number}."
@@ -84,7 +97,7 @@ class ConbusOutputService(ConbusProtocol):
 
         # Send F27D{output:02d}{action} telegram
         # F27 = ACTION, D = data with output number and action type
-        self.send_telegram(
+        self.conbus_protocol.send_telegram(
             telegram_type=TelegramType.SYSTEM,
             serial_number=self.serial_number,
             system_function=SystemFunction.ACTION,
@@ -148,8 +161,12 @@ class ConbusOutputService(ConbusProtocol):
         self.service_response.output_number = self.output_number
         self.service_response.action_type = self.action_type
         self.service_response.output_telegram = output_telegram
-        if self.finish_callback:
-            self.finish_callback(self.service_response)
+        self.on_finish.emit(self.service_response)
+
+    def timeout(self) -> None:
+        """Handle timeout event."""
+        self.logger.debug("Timeout occurred")
+        self.failed("Timeout")
 
     def failed(self, message: str) -> None:
         """Handle failed connection event.
@@ -164,15 +181,13 @@ class ConbusOutputService(ConbusProtocol):
         self.service_response.output_number = self.output_number
         self.service_response.action_type = self.action_type
         self.service_response.error = message
-        if self.finish_callback:
-            self.finish_callback(self.service_response)
+        self.on_finish.emit(self.service_response)
 
     def send_action(
         self,
         serial_number: str,
         output_number: int,
         action_type: ActionType,
-        finish_callback: Callable[[ConbusOutputResponse], None],
         timeout_seconds: Optional[float] = None,
     ) -> None:
         """Send an action telegram to a module output.
@@ -181,14 +196,59 @@ class ConbusOutputService(ConbusProtocol):
             serial_number: 10-digit module serial number.
             output_number: Output number (0-99).
             action_type: Action to perform (ON_RELEASE, OFF_PRESS, etc.).
-            finish_callback: Callback function to call when operation completes.
             timeout_seconds: Optional timeout in seconds.
         """
         self.logger.info("Starting send_action")
         if timeout_seconds:
-            self.timeout_seconds = timeout_seconds
+            self.conbus_protocol.timeout_seconds = timeout_seconds
         self.serial_number = serial_number
         self.output_number = output_number
         self.action_type = action_type
-        self.finish_callback = finish_callback
-        self.start_reactor()
+
+    def set_timeout(self, timeout_seconds: float) -> None:
+        """Set operation timeout.
+
+        Args:
+            timeout_seconds: Timeout in seconds.
+        """
+        self.conbus_protocol.timeout_seconds = timeout_seconds
+
+    def start_reactor(self) -> None:
+        """Start the reactor."""
+        self.conbus_protocol.start_reactor()
+
+    def stop_reactor(self) -> None:
+        """Stop the reactor."""
+        self.conbus_protocol.stop_reactor()
+
+    def __enter__(self) -> "ConbusOutputService":
+        """Enter context manager - reset state for singleton reuse.
+
+        Returns:
+            Self for context manager protocol.
+        """
+        # Reset state for singleton reuse
+        self.service_response = ConbusOutputResponse(
+            success=False,
+            serial_number="",
+            output_number=0,
+            action_type=ActionType.ON_RELEASE,
+            timestamp=datetime.now(),
+        )
+        self.output_state = ""
+        return self
+
+    def __exit__(
+        self, _exc_type: Optional[type], _exc_val: Optional[Exception], _exc_tb: Any
+    ) -> None:
+        """Exit context manager and disconnect signals."""
+        # Disconnect protocol signals
+        self.conbus_protocol.on_connection_made.disconnect(self.connection_made)
+        self.conbus_protocol.on_telegram_sent.disconnect(self.telegram_sent)
+        self.conbus_protocol.on_telegram_received.disconnect(self.telegram_received)
+        self.conbus_protocol.on_timeout.disconnect(self.timeout)
+        self.conbus_protocol.on_failed.disconnect(self.failed)
+        # Disconnect service signals
+        self.on_finish.disconnect()
+        # Stop reactor
+        self.stop_reactor()

xp/services/conbus/conbus_raw_service.py

@@ -5,49 +5,61 @@ This service handles sending raw telegram strings without prior validation.
 
 import logging
 from datetime import datetime
-from typing import Callable, Optional
+from typing import Any, Optional
 
-from twisted.internet.posixbase import PosixReactorBase
+from psygnal import Signal
 
-from xp.models import ConbusClientConfig
 from xp.models.conbus.conbus_raw import ConbusRawResponse
 from xp.models.protocol.conbus_protocol import TelegramReceivedEvent
-from xp.services.protocol import ConbusProtocol
+from xp.services.protocol.conbus_event_protocol import ConbusEventProtocol
 
 
-class ConbusRawService(ConbusProtocol):
+class ConbusRawService:
     """
     Service for sending raw telegram sequences to Conbus modules.
 
-    Uses ConbusProtocol to provide raw telegram functionality
+    Uses ConbusEventProtocol to provide raw telegram functionality
     for sending arbitrary telegram strings without validation.
+
+    Attributes:
+        conbus_protocol: Protocol instance for Conbus communication.
+        on_progress: Signal emitted when telegram is received (with frame).
+        on_finish: Signal emitted when operation finishes (with result).
     """
 
+    conbus_protocol: ConbusEventProtocol
+    on_progress: Signal = Signal(str)
+    on_finish: Signal = Signal(ConbusRawResponse)
+
     def __init__(
         self,
-        cli_config: ConbusClientConfig,
-        reactor: PosixReactorBase,
+        conbus_protocol: ConbusEventProtocol,
     ) -> None:
         """Initialize the Conbus raw service.
 
         Args:
-            cli_config: Configuration for Conbus client connection.
-            reactor: Twisted reactor for event loop.
+            conbus_protocol: ConbusEventProtocol instance.
         """
-        super().__init__(cli_config, reactor)
+        self.conbus_protocol = conbus_protocol
+
+        # Connect protocol signals
+        self.conbus_protocol.on_connection_made.connect(self.connection_made)
+        self.conbus_protocol.on_telegram_sent.connect(self.telegram_sent)
+        self.conbus_protocol.on_telegram_received.connect(self.telegram_received)
+        self.conbus_protocol.on_timeout.connect(self.timeout)
+        self.conbus_protocol.on_failed.connect(self.failed)
+
         self.raw_input: str = ""
-        self.progress_callback: Optional[Callable[[str], None]] = None
-        self.finish_callback: Optional[Callable[[ConbusRawResponse], None]] = None
         self.service_response: ConbusRawResponse = ConbusRawResponse(
             success=False,
         )
         # Set up logging
         self.logger = logging.getLogger(__name__)
 
-    def connection_established(self) -> None:
+    def connection_made(self) -> None:
         """Handle connection established event."""
         self.logger.debug(f"Connection established, sending {self.raw_input}")
-        self.sendFrame(self.raw_input.encode())
+        self.conbus_protocol.send_raw_telegram(self.raw_input)
 
     def telegram_sent(self, telegram_sent: str) -> None:
         """Handle telegram sent event.
@@ -71,19 +83,13 @@ class ConbusRawService(ConbusProtocol):
             self.service_response.received_telegrams = []
         self.service_response.received_telegrams.append(telegram_received.frame)
 
-        if self.progress_callback:
-            self.progress_callback(telegram_received.frame)
-
-    def timeout(self) -> bool:
-        """Handle timeout event.
+        self.on_progress.emit(telegram_received.frame)
 
-        Returns:
-            False to indicate connection should be closed.
-        """
-        self.logger.debug(f"Timeout: {self.timeout_seconds}s")
-        if self.finish_callback:
-            self.finish_callback(self.service_response)
-        return False
+    def timeout(self) -> None:
+        """Handle timeout event."""
+        timeout_seconds = self.conbus_protocol.timeout_seconds
+        self.logger.debug(f"Timeout: {timeout_seconds}s")
+        self.on_finish.emit(self.service_response)
 
     def failed(self, message: str) -> None:
         """Handle failed connection event.
@@ -95,28 +101,63 @@ class ConbusRawService(ConbusProtocol):
         self.service_response.success = False
         self.service_response.timestamp = datetime.now()
         self.service_response.error = message
-        if self.finish_callback:
-            self.finish_callback(self.service_response)
+        self.on_finish.emit(self.service_response)
 
     def send_raw_telegram(
         self,
         raw_input: str,
-        progress_callback: Callable[[str], None],
-        finish_callback: Callable[[ConbusRawResponse], None],
         timeout_seconds: Optional[float] = None,
     ) -> None:
         """Send a raw telegram string to the Conbus server.
 
         Args:
             raw_input: Raw telegram string to send.
-            progress_callback: Callback to handle progress updates.
-            finish_callback: Callback function to call when the operation is complete.
             timeout_seconds: Timeout in seconds.
         """
         self.logger.info("Starting send_raw_telegram")
         if timeout_seconds:
-            self.timeout_seconds = timeout_seconds
-        self.progress_callback = progress_callback
-        self.finish_callback = finish_callback
+            self.conbus_protocol.timeout_seconds = timeout_seconds
         self.raw_input = raw_input
-        self.start_reactor()
+
+    def set_timeout(self, timeout_seconds: float) -> None:
+        """Set operation timeout.
+
+        Args:
+            timeout_seconds: Timeout in seconds.
+        """
+        self.conbus_protocol.timeout_seconds = timeout_seconds
+
+    def start_reactor(self) -> None:
+        """Start the reactor."""
+        self.conbus_protocol.start_reactor()
+
+    def stop_reactor(self) -> None:
+        """Stop the reactor."""
+        self.conbus_protocol.stop_reactor()
+
+    def __enter__(self) -> "ConbusRawService":
+        """Enter context manager.
+
+        Returns:
+            Self for context manager protocol.
+        """
+        # Reset state for singleton reuse
+        self.service_response = ConbusRawResponse(success=False)
+        self.raw_input = ""
+        return self
+
+    def __exit__(
+        self, _exc_type: Optional[type], _exc_val: Optional[Exception], _exc_tb: Any
+    ) -> None:
+        """Exit context manager and disconnect signals."""
+        # Disconnect protocol signals
+        self.conbus_protocol.on_connection_made.disconnect(self.connection_made)
+        self.conbus_protocol.on_telegram_sent.disconnect(self.telegram_sent)
+        self.conbus_protocol.on_telegram_received.disconnect(self.telegram_received)
+        self.conbus_protocol.on_timeout.disconnect(self.timeout)
+        self.conbus_protocol.on_failed.disconnect(self.failed)
+        # Disconnect service signals
+        self.on_progress.disconnect()
+        self.on_finish.disconnect()
+        # Stop reactor
+        self.stop_reactor()