resource script sample for cloudshell-cli-handler

Author: katzy687

cloudshell-cli-samples/send_ssh_command/__main__.py

@@ -0,0 +1,37 @@
+from cloudshell.helpers.scripts.cloudshell_scripts_helpers import get_reservation_context_details, get_api_session, \
+    get_resource_context_details
+from cli import SSHHandler
+import os
+
+COMMAND_PARAM = "command"
+cli_command = os.environ.get(COMMAND_PARAM)
+if not cli_command:
+    raise ValueError(f"Did not receive cli command param {COMMAND_PARAM}")
+
+api = get_api_session()
+sb_details = get_reservation_context_details()
+sb_id = sb_details.id
+
+resource_details = get_resource_context_details()
+model = resource_details.model
+name = resource_details.name
+ip = resource_details.address
+attrs = resource_details.attributes
+normalized_attrs = {k.split(".")[-1]: v for k, v in attrs.items()}
+
+user = normalized_attrs["User"]
+if not user:
+    raise ValueError("Please populate User attribute for SSH session")
+encrypted_password = normalized_attrs["Password"]
+decrypted_password = api.DecryptPassword(encrypted_password).Value
+if not decrypted_password:
+    raise ValueError("Please populate Password attribute for SSH login")
+
+api.WriteMessageToReservationOutput(reservationId=sb_id,
+                                    message=f"Sending command to '{name}' at IP: {ip}")
+
+cli = SSHHandler(ip, user, decrypted_password)
+output = cli.send_command(cli_command)
+
+# printing to std_out will be the return value of resource scripts
+print(output)

cloudshell-cli-samples/send_ssh_command/cli.py

@@ -0,0 +1,15 @@
+from cloudshell.cli.service.cli import CLI
+from cloudshell.cli.session.ssh_session import SSHSession
+from cloudshell.cli.service.command_mode import CommandMode
+
+
+class SSHHandler:
+    def __init__(self, ip, user, password, prompt="#"):
+        self.cli = CLI()
+        self.mode = CommandMode(prompt)
+        self.session_types = [SSHSession(host=ip, username=user, password=password)]
+
+    def send_command(self, command):
+        with self.cli.get_session(self.session_types, self.mode) as cli_service:
+            output = cli_service.send_command(command)
+        return output

cloudshell-cli-samples/send_ssh_command/requirements.txt

@@ -0,0 +1,3 @@
+cloudshell-cli
+cloudshell-automation-api
+six