Feature raritan sessions

cloudshell/raritan/cli/raritan_cli_configurator.py

@@ -17,13 +17,13 @@
 from cloudshell.cli.service.command_mode_helper import CommandModeHelper
 from cloudshell.cli.session.console_ssh import ConsoleSSHSession
 from cloudshell.cli.session.console_telnet import ConsoleTelnetSession
-from cloudshell.cli.session.ssh_session import SSHSession
-from cloudshell.cli.session.telnet_session import TelnetSession
 
 from cloudshell.raritan.cli.raritan_command_modes import (
     ConfigCommandMode,
     EnableCommandMode,
 )
+from cloudshell.raritan.cli.raritan_ssh_session import RaritanSSHSession
+from cloudshell.raritan.cli.raritan_telnet_session import RaritanTelnetSession
 
 if TYPE_CHECKING:
     from cloudshell.cli.service.cli import CLI
@@ -33,8 +33,8 @@
 @define
 class RaritanCliConfigurator(AbstractModeConfigurator):
     REGISTERED_SESSIONS: ClassVar[tuple[SessionFactory]] = (
-        CloudInfoAccessKeySessionFactory(SSHSession),
-        GenericSessionFactory(TelnetSession),
+        CloudInfoAccessKeySessionFactory(RaritanSSHSession),
+        GenericSessionFactory(RaritanTelnetSession),
         ConsoleSessionFactory(ConsoleSSHSession),
         ConsoleSessionFactory(
             ConsoleTelnetSession, session_kwargs={"start_with_new_line": False}

cloudshell/raritan/cli/raritan_ssh_session.py

@@ -0,0 +1,28 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from cloudshell.cli.session.ssh_session import SSHSession
+
+if TYPE_CHECKING:
+    from logging import Logger
+
+
+class RaritanSSHSession(SSHSession):
+    def _connect_actions(self, prompt: str, logger: Logger) -> None:
+        action_map = dict()
+        cli_action_key = r"Restricted Service Agreement"
+
+        def action(session: RaritanSSHSession, sess_logger: Logger) -> None:
+            session.send_line("y", sess_logger)
+            del action_map[cli_action_key]
+
+        action_map[cli_action_key] = action
+        self.hardware_expect(
+            None,
+            expected_string=prompt,
+            action_map=action_map,
+            timeout=self._timeout,
+            logger=logger,
+        )
+        self._on_session_start(logger)

cloudshell/raritan/cli/raritan_telnet_session.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from cloudshell.cli.session.telnet_session import TelnetSession
+
+if TYPE_CHECKING:
+    from logging import Logger
+    from cloudshell.cli.types import T_ACTION_MAP
+
+
+class RaritanTelnetSession(TelnetSession):
+    def _connect_action_map(self) -> T_ACTION_MAP:
+        am = super()._connect_action_map
+        cli_action_key = r"Restricted Service Agreement"
+
+        def action(session: RaritanTelnetSession, sess_logger: Logger) -> None:
+            session.send_line("y", sess_logger)
+            del am[cli_action_key]
+
+        am[cli_action_key] = action
+
+        return am

version.txt

@@ -1 +1 @@
-1.0.1
+1.0.2