1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
Steps for adding a new vendor
=======
Create a new vendor directory underneath netmiko/netmiko:
```
$ cd netmiko/netmiko
$ mkdir arista
```
Make the directory a Python package:
```
$ cd arista
$ touch __init__.py
```
Create a new module for the vendor:
```
$ vi arista_ssh.py
from netmiko.cisco_base_connection import CiscoSSHConnection
class AristaSSH(CiscoSSHConnection):
pass
```
Inherit from the CiscoSSHConnection class. Note, the netmiko package will need to be in
your PYTHONPATH
Update \_\_init__.py to export the new class:
```
$ vi __init__.py
from netmiko.arista.arista_ssh import AristaSSH
__all__ = ['AristaSSH']
```
Update the dispatcher adding the new class:
```
$ cd netmiko/netmiko
$ vi ssh_dispatcher.py
from netmiko.cisco import CiscoIosSSH
from netmiko.cisco import CiscoAsaSSH
from netmiko.arista import AristaSSH # Add here
CLASS_MAPPER = {
'cisco_ios' : CiscoIosSSH,
'cisco_asa' : CiscoAsaSSH,
'arista_eos' : AristaSSH, # Add Here (device_type to class mapping)
}
```
Your class will need to support the following methods:
```
def __init__(self, ip, username, password, secret='', port=22, device_type='', verbose=True):
def establish_connection(self, sleep_time=3, verbose=True):
def disable_paging(self):
def find_prompt(self):
def clear_buffer(self):
def send_command(self, command_string, delay_factor=1, max_loops=30):
def strip_prompt(self, a_string):
def strip_command(self, command_string, output):
def normalize_linefeeds(self, a_string):
def enable(self):
def config_mode(self):
def exit_config_mode(self):
def send_config_set(self, config_commands=None):
def cleanup(self):
def disconnect(self):
```
As much as possible, you should re-use the inherited methods from CiscoSSHConnection
and BaseConnection (i.e. only re-write what you have to).
BaseConnection is intended to be generic (i.e. irrespective of the vendor)
CiscoSSHConnection is Cisco-IOS specific (because lots of vendors imitate Cisco IOS).
You should also write unit tests and verify the functionality of your new class.
See netmiko/tests/test_cisco_ios.py and netmiko/tests/test_cisco_ios_enable.py.
Note, you will also need to update the 'packages' section of 'setup.py' file if you are adding a
completely new package.
|
