aruba_device.py

import paramiko
import time


class controller:
    def __init__(self, ip, user, pw, en):
        self.ip = ip
        self.user = user
        self.pw = pw
        self.en = en
        self.ssh = self.connect()
        self.aps = set()

    # Create SSH on tcp/22 connection and set host key policy to "auto add" unknown host keys
    def connect(self):
        conn = paramiko.SSHClient()
        conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        conn.connect(self.ip, 22, self.user, self.pw)
        return conn

    # Sends a command to the controller.  Each time a command is sent, the shell
    # is invoked, then the method enters enable mode and disables paging.
    # After this the command is sent and output is retuned
    def send_cmd(self, cmd):
        output = ''
        shell = self.ssh.invoke_shell()
        shell.send('en\n'+self.en+'\n'+'no paging\n')
        time.sleep(0.5)
        while shell.recv_ready():
            shell.recv(1024)
        for c in cmd:
            shell.send(c+'\n')
            time.sleep(.5)
            while shell.recv_ready():
                output += shell.recv(4096)
                time.sleep(.5)
        return output