root / ncclient / transport / hello.py @ c35cebbf
History | View | Annotate | Download (2.5 kB)
1 | 68006767 | Shikhar Bhushan | # Copyright 2009 Shikhar Bhushan
|
---|---|---|---|
2 | 68006767 | Shikhar Bhushan | #
|
3 | 68006767 | Shikhar Bhushan | # Licensed under the Apache License, Version 2.0 (the "License");
|
4 | 68006767 | Shikhar Bhushan | # you may not use this file except in compliance with the License.
|
5 | 68006767 | Shikhar Bhushan | # You may obtain a copy of the License at
|
6 | 68006767 | Shikhar Bhushan | #
|
7 | 68006767 | Shikhar Bhushan | # http://www.apache.org/licenses/LICENSE-2.0
|
8 | 68006767 | Shikhar Bhushan | #
|
9 | 68006767 | Shikhar Bhushan | # Unless required by applicable law or agreed to in writing, software
|
10 | 68006767 | Shikhar Bhushan | # distributed under the License is distributed on an "AS IS" BASIS,
|
11 | 68006767 | Shikhar Bhushan | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12 | 68006767 | Shikhar Bhushan | # See the License for the specific language governing permissions and
|
13 | 68006767 | Shikhar Bhushan | # limitations under the License.
|
14 | 68006767 | Shikhar Bhushan | |
15 | 68006767 | Shikhar Bhushan | "All to do with NETCONF <hello> messages"
|
16 | 68006767 | Shikhar Bhushan | |
17 | 68006767 | Shikhar Bhushan | from xml.etree import cElementTree as ET |
18 | 68006767 | Shikhar Bhushan | |
19 | e91a5349 | Shikhar Bhushan | from ncclient.glue import Listener |
20 | e91a5349 | Shikhar Bhushan | from ncclient.content import TreeBuilder, BASE_NS |
21 | 68006767 | Shikhar Bhushan | from ncclient.content import qualify as _ |
22 | e91a5349 | Shikhar Bhushan | from ncclient.content import unqualify as __ |
23 | 68006767 | Shikhar Bhushan | |
24 | c35cebbf | Shikhar Bhushan | class HelloHandler(Listener): |
25 | 68006767 | Shikhar Bhushan | |
26 | 68006767 | Shikhar Bhushan | def __init__(self, init_cb, error_cb): |
27 | 68006767 | Shikhar Bhushan | self._init_cb, self._error_cb = init_cb, error_cb |
28 | 68006767 | Shikhar Bhushan | |
29 | 68006767 | Shikhar Bhushan | def __str__(self): |
30 | 68006767 | Shikhar Bhushan | return 'HelloListener' |
31 | 68006767 | Shikhar Bhushan | |
32 | 68006767 | Shikhar Bhushan | def callback(self, root, raw): |
33 | 68006767 | Shikhar Bhushan | if __(root[0]) == 'hello': |
34 | 68006767 | Shikhar Bhushan | try:
|
35 | 68006767 | Shikhar Bhushan | id, capabilities = parse(raw)
|
36 | 68006767 | Shikhar Bhushan | except Exception as e: |
37 | 68006767 | Shikhar Bhushan | self._error_cb(e)
|
38 | 68006767 | Shikhar Bhushan | else:
|
39 | 68006767 | Shikhar Bhushan | self._init_cb(id, capabilities) |
40 | 68006767 | Shikhar Bhushan | |
41 | 68006767 | Shikhar Bhushan | def errback(self, err): |
42 | 68006767 | Shikhar Bhushan | self._error_cb(err)
|
43 | c35cebbf | Shikhar Bhushan | |
44 | c35cebbf | Shikhar Bhushan | @staticmethod
|
45 | c35cebbf | Shikhar Bhushan | def build(capabilities, encoding='utf-8'): |
46 | c35cebbf | Shikhar Bhushan | "Given a list of capability URI's returns encoded <hello> message"
|
47 | c35cebbf | Shikhar Bhushan | spec = { |
48 | c35cebbf | Shikhar Bhushan | 'tag': _('hello', BASE_NS), |
49 | c35cebbf | Shikhar Bhushan | 'children': [{
|
50 | c35cebbf | Shikhar Bhushan | 'tag': 'capabilities', |
51 | c35cebbf | Shikhar Bhushan | 'children': # this is fun :-) |
52 | c35cebbf | Shikhar Bhushan | [{ 'tag': 'capability', 'text': uri} for uri in capabilities] |
53 | c35cebbf | Shikhar Bhushan | }] |
54 | c35cebbf | Shikhar Bhushan | } |
55 | c35cebbf | Shikhar Bhushan | return TreeBuilder(spec).to_string(encoding)
|
56 | c35cebbf | Shikhar Bhushan | |
57 | c35cebbf | Shikhar Bhushan | @staticmethod
|
58 | c35cebbf | Shikhar Bhushan | def parse(raw): |
59 | c35cebbf | Shikhar Bhushan | "Returns tuple of ('session-id', ['capability_uri', ...])"
|
60 | c35cebbf | Shikhar Bhushan | sid, capabilities = 0, []
|
61 | c35cebbf | Shikhar Bhushan | root = ET.fromstring(raw) |
62 | c35cebbf | Shikhar Bhushan | for child in root.getchildren(): |
63 | c35cebbf | Shikhar Bhushan | if __(child.tag) == 'session-id': |
64 | c35cebbf | Shikhar Bhushan | sid = child.text |
65 | c35cebbf | Shikhar Bhushan | elif __(child.tag) == 'capabilities': |
66 | c35cebbf | Shikhar Bhushan | for cap in child.getiterator(_('capability', BASE_NS)): |
67 | c35cebbf | Shikhar Bhushan | capabilities.append(cap.text) |
68 | c35cebbf | Shikhar Bhushan | # cisco doesn't namespace hello message
|
69 | c35cebbf | Shikhar Bhushan | for cap in child.getiterator('capability'): |
70 | c35cebbf | Shikhar Bhushan | capabilities.append(cap.text) |
71 | c35cebbf | Shikhar Bhushan | return sid, capabilities |