root / ncclient / content.py @ ee4bb099
History | View | Annotate | Download (1.9 kB)
1 |
# Copyright 2009 Shikhar Bhushan
|
---|---|
2 |
#
|
3 |
# Licensed under the Apache License, Version 2.0 (the "License");
|
4 |
# you may not use this file except in compliance with the License.
|
5 |
# You may obtain a copy of the License at
|
6 |
#
|
7 |
# http://www.apache.org/licenses/LICENSE-2.0
|
8 |
#
|
9 |
# Unless required by applicable law or agreed to in writing, software
|
10 |
# distributed under the License is distributed on an "AS IS" BASIS,
|
11 |
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12 |
# See the License for the specific language governing permissions and
|
13 |
# limitations under the License.
|
14 |
|
15 |
import logging |
16 |
from xml.etree import cElementTree as ElementTree |
17 |
|
18 |
logger = logging.getLogger('ncclient.content')
|
19 |
|
20 |
BASE_NS = 'urn:ietf:params:xml:ns:netconf:base:1.0'
|
21 |
NOTIFICATION_NS = 'urn:ietf:params:xml:ns:netconf:notification:1.0'
|
22 |
|
23 |
def qualify(tag, ns=BASE_NS): |
24 |
return '{%s}%s' % (ns, tag) |
25 |
|
26 |
_ = qualify |
27 |
|
28 |
def make_hello(capabilities): |
29 |
return '<hello xmlns="%s">%s</hello>' % (BASE_NS, capabilities) |
30 |
|
31 |
def make_rpc(id, op): |
32 |
return '<rpc message-id="%s" xmlns="%s">%s</rpc>' % (id, BASE_NS, op) |
33 |
|
34 |
def parse_hello(raw): |
35 |
from capabilities import Capabilities |
36 |
id, capabilities = 0, Capabilities() |
37 |
root = ElementTree.fromstring(raw) |
38 |
if root.tag == _('hello'): |
39 |
for child in root.getchildren(): |
40 |
if child.tag == _('session-id'): |
41 |
id = int(child.text)
|
42 |
elif child.tag == _('capabilities'): |
43 |
for cap in child.getiterator(_('capability')): |
44 |
capabilities.add(cap.text) |
45 |
return id, capabilities |
46 |
|
47 |
def parse_message_root(raw): |
48 |
from cStringIO import StringIO |
49 |
fp = StringIO(raw) |
50 |
for event, element in ElementTree.iterparse(fp, events=('start',)): |
51 |
if element.tag == _('rpc'): |
52 |
return element.attrib['message-id'] |
53 |
elif element.tag == _('notification', NOTIFICATION_NS): |
54 |
return 'notification' |
55 |
else:
|
56 |
return None |