chapter 27 examples

examples/27_oop_inheritance/cisco_ssh_class.py

@@ -0,0 +1,31 @@
+from connect_ssh_class import ConnectSSH
+import time
+
+
+class CiscoSSH(ConnectSSH):
+    def __init__(self, ip, username, password, enable_password,
+                 disable_paging=True):
+        super().__init__(ip, username, password)
+        self._ssh.send('enable\n')
+        self._ssh.send(enable_password +'\n')
+        if disable_paging:
+            self._ssh.send('terminal length 0\n')
+        time.sleep(1)
+        self._ssh.recv(self._MAX_READ)
+
+    def config_mode(self):
+        self._ssh.send('conf t\n')
+        time.sleep(0.2)
+        result = self._ssh.recv(self._MAX_READ).decode('ascii')
+        return result
+
+    def exit_config_mode(self):
+        return super().send_command('end')
+
+    def send_config_commands(self, commands):
+        output = self.config_mode()
+        output += super().send_config_commands(commands)
+        output += self.exit_config_mode()
+        return output
+
+

examples/27_oop_inheritance/connect_ssh_class.py

@@ -0,0 +1,48 @@
+import paramiko
+import time
+
+
+class ConnectSSH:
+    def __init__(self, ip, username, password):
+        self.ip = ip
+        self.username = username
+        self.password = password
+        self._MAX_READ = 10000
+
+        client = paramiko.SSHClient()
+        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+        client.connect(
+            hostname=ip,
+            username=username,
+            password=password,
+            look_for_keys=False,
+            allow_agent=False)
+
+        self._ssh = client.invoke_shell()
+        time.sleep(1)
+        self._ssh.recv(self._MAX_READ)
+
+    def send_command(self, command):
+        self._ssh.send(command + '\n')
+        time.sleep(2)
+        result = self._ssh.recv(self._MAX_READ).decode('ascii')
+        return result
+
+    def send_config_commands(self, commands):
+        if isinstance(commands, str):
+            commands = [commands]
+        for command in commands:
+            self._ssh.send(command + '\n')
+            time.sleep(0.5)
+        result = self._ssh.recv(self._MAX_READ).decode('ascii')
+        return result
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self._ssh.close()
+
+    def close(self):
+        self._ssh.close()

examples/27_oop_inheritance/netmiko_inheritance.py

@@ -0,0 +1,15 @@
+from netmiko.cisco.cisco_ios import CiscoIosSSH
+
+
+class ErrorInCommand(Exception):
+    """Это исключение генерируется при ошибке в команде"""
+
+
+class MyNetmiko(CiscoIosSSH):
+    def send_command(self, command, **kwargs):
+        output = super().send_command(command, **kwargs)
+        if 'Invalid input' in output:
+            raise ErrorInCommand('Возникла ошибка')
+        return output
+
+