root / ncclient / manager.py @ 4bc8021f
History | View | Annotate | Download (6.7 kB)
1 | 94803aaf | Shikhar Bhushan | # Copyright 2009 Shikhar Bhushan
|
---|---|---|---|
2 | 94803aaf | Shikhar Bhushan | #
|
3 | 94803aaf | Shikhar Bhushan | # Licensed under the Apache License, Version 2.0 (the "License");
|
4 | 94803aaf | Shikhar Bhushan | # you may not use this file except in compliance with the License.
|
5 | 94803aaf | Shikhar Bhushan | # You may obtain a copy of the License at
|
6 | 94803aaf | Shikhar Bhushan | #
|
7 | 94803aaf | Shikhar Bhushan | # http://www.apache.org/licenses/LICENSE-2.0
|
8 | 94803aaf | Shikhar Bhushan | #
|
9 | 94803aaf | Shikhar Bhushan | # Unless required by applicable law or agreed to in writing, software
|
10 | 94803aaf | Shikhar Bhushan | # distributed under the License is distributed on an "AS IS" BASIS,
|
11 | 94803aaf | Shikhar Bhushan | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12 | 94803aaf | Shikhar Bhushan | # See the License for the specific language governing permissions and
|
13 | 94803aaf | Shikhar Bhushan | # limitations under the License.
|
14 | 94803aaf | Shikhar Bhushan | |
15 | 4bc8021f | Shikhar Bhushan | "This module is a thin layer of abstraction around the library. It exposes all core functionality."
|
16 | 216bb34c | Shikhar Bhushan | |
17 | 94803aaf | Shikhar Bhushan | import capabilities |
18 | 0b7d3b31 | Shikhar Bhushan | import operations |
19 | 94803aaf | Shikhar Bhushan | import transport |
20 | 94803aaf | Shikhar Bhushan | |
21 | 454339d6 | Shikhar Bhushan | import logging |
22 | 454339d6 | Shikhar Bhushan | logger = logging.getLogger('ncclient.manager')
|
23 | 454339d6 | Shikhar Bhushan | |
24 | dd225c7a | Shikhar Bhushan | |
25 | 68ac4439 | Shikhar Bhushan | CAPABILITIES = capabilities.Capabilities([ |
26 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:base:1.0",
|
27 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:writable-running:1.0",
|
28 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:candidate:1.0",
|
29 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:confirmed-commit:1.0",
|
30 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:rollback-on-error:1.0",
|
31 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:startup:1.0",
|
32 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:url:1.0?scheme=http,ftp,file,https,sftp",
|
33 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:validate:1.0",
|
34 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:xpath:1.0",
|
35 | 68ac4439 | Shikhar Bhushan | "urn:liberouter:params:netconf:capability:power-control:1.0"
|
36 | 68ac4439 | Shikhar Bhushan | "urn:ietf:params:netconf:capability:interleave:1.0"
|
37 | 68ac4439 | Shikhar Bhushan | #'urn:ietf:params:netconf:capability:notification:1.0', # TODO
|
38 | 68ac4439 | Shikhar Bhushan | ]) |
39 | 4bc8021f | Shikhar Bhushan | """`~ncclient.capabilities.Capabilities` object representing the client's capabilities. This is used
|
40 | 4bc8021f | Shikhar Bhushan | during the initial capability exchange. Modify this if you need to announce some capability not
|
41 | 4bc8021f | Shikhar Bhushan | already included.
|
42 | 4bc8021f | Shikhar Bhushan | """
|
43 | 2f8bc438 | Shikhar Bhushan | |
44 | b660fcda | Shikhar Bhushan | OPERATIONS = { |
45 | b660fcda | Shikhar Bhushan | "get": operations.Get,
|
46 | b660fcda | Shikhar Bhushan | "get_config": operations.GetConfig,
|
47 | b660fcda | Shikhar Bhushan | "edit_config": operations.EditConfig,
|
48 | b660fcda | Shikhar Bhushan | "copy_config": operations.CopyConfig,
|
49 | b660fcda | Shikhar Bhushan | "validate": operations.Validate,
|
50 | b660fcda | Shikhar Bhushan | "commit": operations.Commit,
|
51 | b660fcda | Shikhar Bhushan | "discard_changes": operations.DiscardChanges,
|
52 | b660fcda | Shikhar Bhushan | "delete_config": operations.DeleteConfig,
|
53 | b660fcda | Shikhar Bhushan | "lock": operations.Lock,
|
54 | b660fcda | Shikhar Bhushan | "unlock": operations.Unlock,
|
55 | b660fcda | Shikhar Bhushan | "close_session": operations.CloseSession,
|
56 | b660fcda | Shikhar Bhushan | "kill_session": operations.KillSession,
|
57 | b660fcda | Shikhar Bhushan | "poweroff_machine": operations.PoweroffMachine,
|
58 | b660fcda | Shikhar Bhushan | "reboot_machine": operations.RebootMachine
|
59 | b660fcda | Shikhar Bhushan | } |
60 | 4bc8021f | Shikhar Bhushan | """Dictionary of method names and corresponding `~ncclient.operations.RPC` subclasses. `Manager`
|
61 | 4bc8021f | Shikhar Bhushan | uses this to lookup operations, e.g. "get_config" is mapped to `~ncclient.operations.GetConfig`. It
|
62 | 4bc8021f | Shikhar Bhushan | is thus possible to add additional operations to the `Manager` API.
|
63 | 4bc8021f | Shikhar Bhushan | """
|
64 | 4f650d54 | Shikhar Bhushan | |
65 | 68ac4439 | Shikhar Bhushan | def connect_ssh(*args, **kwds): |
66 | 4bc8021f | Shikhar Bhushan | """Initializes a NETCONF session over SSH, and creates a connected `Manager` instance. *host*
|
67 | 4bc8021f | Shikhar Bhushan | must be specified, all the other arguments are optional and depend on the kind of host key
|
68 | 4bc8021f | Shikhar Bhushan | verification and user authentication you want to complete.
|
69 | 4bc8021f | Shikhar Bhushan |
|
70 | 4bc8021f | Shikhar Bhushan | For the purpose of host key verification, on -NIX systems a user's :file:`~/.ssh/known_hosts`
|
71 | 4bc8021f | Shikhar Bhushan | file is automatically considered. The *unknown_host_cb* argument specifies a callback that will
|
72 | 4bc8021f | Shikhar Bhushan | be invoked when the server's host key cannot be verified. See
|
73 | 4bc8021f | Shikhar Bhushan | :func:`~ncclient.transport.ssh.default_unknown_host_cb` for function signature.
|
74 | 4bc8021f | Shikhar Bhushan |
|
75 | 4bc8021f | Shikhar Bhushan | First, ``publickey`` authentication is attempted. If a specific *key_filename* is specified, it
|
76 | 4bc8021f | Shikhar Bhushan | will be loaded and authentication attempted using it. If *allow_agent* is :const:`True` and an
|
77 | 4bc8021f | Shikhar Bhushan | SSH agent is running, the keys provided by the agent will be tried. If *look_for_keys* is
|
78 | 4bc8021f | Shikhar Bhushan | :const:`True`, keys in the :file:`~/.ssh/id_rsa` and :file:`~.ssh/id_dsa` will be tried. In case
|
79 | 4bc8021f | Shikhar Bhushan | an encrypted key file is encountered, the *password* argument will be used as a decryption
|
80 | 4bc8021f | Shikhar Bhushan | passphrase.
|
81 | 4bc8021f | Shikhar Bhushan |
|
82 | 4bc8021f | Shikhar Bhushan | If ``publickey`` authentication fails and the *password* argument has been supplied,
|
83 | 4bc8021f | Shikhar Bhushan | ``password`` / ``keyboard-interactive`` SSH authentication will be attempted.
|
84 | 4bc8021f | Shikhar Bhushan |
|
85 | 4bc8021f | Shikhar Bhushan | :param host: hostname or address on which to connect
|
86 | 4bc8021f | Shikhar Bhushan | :type host: `string`
|
87 | 4bc8021f | Shikhar Bhushan |
|
88 | 4bc8021f | Shikhar Bhushan | :param port: port on which to connect
|
89 | 4bc8021f | Shikhar Bhushan | :type port: `int`
|
90 | 4bc8021f | Shikhar Bhushan |
|
91 | 4bc8021f | Shikhar Bhushan | :param timeout: timeout for socket connect
|
92 | 4bc8021f | Shikhar Bhushan | :type timeout: `int`
|
93 | 4bc8021f | Shikhar Bhushan |
|
94 | 4bc8021f | Shikhar Bhushan | :param unknown_host_cb: optional; callback that is invoked when host key verification fails
|
95 | 4bc8021f | Shikhar Bhushan | :type unknown_host_cb: `function`
|
96 | 4bc8021f | Shikhar Bhushan |
|
97 | 4bc8021f | Shikhar Bhushan | :param username: username to authenticate with, if not specified the username of the logged-in user is used
|
98 | 4bc8021f | Shikhar Bhushan | :type username: `string`
|
99 | 4bc8021f | Shikhar Bhushan |
|
100 | 4bc8021f | Shikhar Bhushan | :param password: password for ``password`` authentication or passphrase for decrypting private key files
|
101 | 4bc8021f | Shikhar Bhushan | :type password: `string`
|
102 | 4bc8021f | Shikhar Bhushan |
|
103 | 4bc8021f | Shikhar Bhushan | :param key_filename: location of a private key file on the file system
|
104 | 4bc8021f | Shikhar Bhushan | :type key_filename: `string`
|
105 | 4bc8021f | Shikhar Bhushan |
|
106 | 4bc8021f | Shikhar Bhushan | :param allow_agent: whether to try connecting to SSH agent for keys
|
107 | 4bc8021f | Shikhar Bhushan | :type allow_agent: `bool`
|
108 | 4bc8021f | Shikhar Bhushan |
|
109 | 4bc8021f | Shikhar Bhushan | :param look_for_keys: whether to look in usual locations for keys
|
110 | 4bc8021f | Shikhar Bhushan | :type look_for_keys: `bool`
|
111 | 4bc8021f | Shikhar Bhushan |
|
112 | 4bc8021f | Shikhar Bhushan | :raises: :exc:`~ncclient.transport.SSHUnknownHostError`
|
113 | 4bc8021f | Shikhar Bhushan | :raises: :exc:`~ncclient.transport.AuthenticationError`
|
114 | 4bc8021f | Shikhar Bhushan |
|
115 | 4bc8021f | Shikhar Bhushan | :rtype: `Manager`
|
116 | 4bc8021f | Shikhar Bhushan | """
|
117 | bb19e320 | Shikhar Bhushan | session = transport.SSHSession(CAPABILITIES) |
118 | 68ac4439 | Shikhar Bhushan | session.load_known_hosts() |
119 | 68ac4439 | Shikhar Bhushan | session.connect(*args, **kwds) |
120 | 68ac4439 | Shikhar Bhushan | return Manager(session)
|
121 | 68ac4439 | Shikhar Bhushan | |
122 | 68ac4439 | Shikhar Bhushan | connect = connect_ssh |
123 | 4bc8021f | Shikhar Bhushan | "Same as :func:`connect_ssh`, since SSH is the default (and currently, the only) transport."
|
124 | 216bb34c | Shikhar Bhushan | |
125 | 68ac4439 | Shikhar Bhushan | class Manager(object): |
126 | 4f650d54 | Shikhar Bhushan | |
127 | 4f650d54 | Shikhar Bhushan | def __init__(self, session): |
128 | 179b00d4 | Shikhar Bhushan | self._session = session
|
129 | c8381a66 | Shikhar Bhushan | self._async_mode = False |
130 | c8381a66 | Shikhar Bhushan | self._timeout = None |
131 | 6c70b245 | Shikhar Bhushan | self._raise_mode = 'all' |
132 | 216bb34c | Shikhar Bhushan | |
133 | 454339d6 | Shikhar Bhushan | def __enter__(self): |
134 | 454339d6 | Shikhar Bhushan | return self |
135 | 454339d6 | Shikhar Bhushan | |
136 | c8381a66 | Shikhar Bhushan | def __exit__(self, *argss): |
137 | 68ac4439 | Shikhar Bhushan | self.close_session()
|
138 | 454339d6 | Shikhar Bhushan | return False |
139 | d6688264 | Shikhar Bhushan | |
140 | dd225c7a | Shikhar Bhushan | def __getattr__(self, name): |
141 | b660fcda | Shikhar Bhushan | op = OPERATIONS.get(name, None)
|
142 | 0304f041 | Shikhar Bhushan | if op is None: |
143 | dd225c7a | Shikhar Bhushan | raise AttributeError |
144 | c8381a66 | Shikhar Bhushan | else:
|
145 | eff6bf64 | Shikhar Bhushan | return op(self._session, |
146 | c8381a66 | Shikhar Bhushan | async=self._async_mode,
|
147 | 0304f041 | Shikhar Bhushan | timeout=self._timeout,
|
148 | 6c70b245 | Shikhar Bhushan | raise_mode=self._raise_mode).request
|
149 | 0304f041 | Shikhar Bhushan | |
150 | 216bb34c | Shikhar Bhushan | def locked(self, target): |
151 | 216bb34c | Shikhar Bhushan | return operations.LockContext(self._session, target) |
152 | 68ac4439 | Shikhar Bhushan | |
153 | cc9af1c3 | Shikhar Bhushan | @property
|
154 | a7cb58ce | Shikhar Bhushan | def client_capabilities(self): |
155 | cc9af1c3 | Shikhar Bhushan | return self._session._client_capabilities |
156 | 4f650d54 | Shikhar Bhushan | |
157 | 4f650d54 | Shikhar Bhushan | @property
|
158 | 4f650d54 | Shikhar Bhushan | def server_capabilities(self): |
159 | 4f650d54 | Shikhar Bhushan | return self._session._server_capabilities |
160 | a7cb58ce | Shikhar Bhushan | |
161 | a7cb58ce | Shikhar Bhushan | @property
|
162 | a7cb58ce | Shikhar Bhushan | def session_id(self): |
163 | a7cb58ce | Shikhar Bhushan | return self._session.id |
164 | 216bb34c | Shikhar Bhushan | |
165 | 216bb34c | Shikhar Bhushan | @property
|
166 | 216bb34c | Shikhar Bhushan | def connected(self): |
167 | 216bb34c | Shikhar Bhushan | return self._session.connected |
168 | c8381a66 | Shikhar Bhushan | |
169 | 68ac4439 | Shikhar Bhushan | def set_async_mode(self, mode): |
170 | 68ac4439 | Shikhar Bhushan | self._async_mode = mode
|
171 | c8381a66 | Shikhar Bhushan | |
172 | 6c70b245 | Shikhar Bhushan | def set_raise_mode(self, mode): |
173 | 68ac4439 | Shikhar Bhushan | assert(choice in ("all", "errors", "none")) |
174 | 6c70b245 | Shikhar Bhushan | self._raise_mode = mode
|
175 | c8381a66 | Shikhar Bhushan | |
176 | c8381a66 | Shikhar Bhushan | async_mode = property(fget=lambda self: self._async_mode, fset=set_async_mode) |
177 | c8381a66 | Shikhar Bhushan | |
178 | 68ac4439 | Shikhar Bhushan | raise_mode = property(fget=lambda self: self._raise_mode, fset=set_raise_mode) |