py_ssh.py

import paramiko


BUF_SIZE = 8192
TIMEOUT = None

class SSHCommand:
    def __init__(self, address, username, password):
        print("Connecting to server on ip", str(address) + ".")

        self._client = paramiko.client.SSHClient()
        self._client.set_missing_host_key_policy(
            paramiko.client.AutoAddPolicy())
        self._client.connect(address, username=username, password=password,
                            look_for_keys=False)

        self._transport = self._client.get_transport()

    def close(self):
        self._client.close()

    def send_cmd(self, cmd):
        chan = self._transport.open_session(timeout=TIMEOUT)
        chan.settimeout(TIMEOUT)
        chan.exec_command(cmd)

        data = self._recv(chan)
        status = chan.recv_exit_status()

        return status, data

    def _recv(self, chan):
        buf = []
        while True:
            data = chan.recv(BUF_SIZE)
            if data == b'':
                break

            data = data.decode("utf-8")
            data = data.replace('\r', '')
            buf.append(data)


        return "".join(buf)

client = SSHCommand("localhost", ...)

data = client.send_cmd("ls /")
print(data)

data = client.send_cmd("ls /XXX")
print(data)