ncclient:NETCONF Python client

# Copyright 2009 Shikhar Bhushan

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#    http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

"TODO: docstring"

from xml.etree import cElementTree as ET

### Namespace-related ###

BASE_NS = 'urn:ietf:params:xml:ns:netconf:base:1.0'

# and this is BASE_NS according to cisco devices...

CISCO_BS = 'urn:ietf:params:netconf:base:1.0'

try:

    register_namespace = ET.register_namespace

except AttributeError:

    def register_namespace(prefix, uri):

        from xml.etree import ElementTree

        # cElementTree uses ElementTree's _namespace_map, so that's ok

        ElementTree._namespace_map[uri] = prefix

# we'd like BASE_NS to be prefixed as "netconf"

register_namespace('netconf', BASE_NS)

qualify = lambda tag, ns=BASE_NS: '{%s}%s' % (ns, tag)

# i would have written a def if lambdas weren't so much fun

multiqualify = lambda tag, nslist=(BASE_NS, CISCO_BS): [qualify(tag, ns)

                                                        for ns in nslist]

unqualify = lambda tag: tag[tag.rfind('}')+1:]

### Build XML using Python data structures ###

class XMLConverter:

    """Build an ElementTree.Element instance from an XML tree specification

    based on nested dictionaries. TODO: describe spec

    """

    def __init__(self, spec):

        "TODO: docstring"

        self._root = XMLConverter.build(spec)

    def to_string(self, encoding='utf-8'):

        "TODO: docstring"

        xml = ET.tostring(self._root, encoding)

        # some etree versions don't always include xml decl e.g. with utf-8

        # this is a problem with some devices

        if not xml.startswith('<?xml'):

            return ((u'<?xml version="1.0" encoding="%s"?>'

                     % encoding).encode(encoding) + xml)

        else:

            return xml

    @property

    def tree(self):

        "TODO: docstring"

        return self._root

    @staticmethod

    def build(spec):

        "TODO: docstring"

        if 'tag' in spec:

            ele = ET.Element(spec.get('tag'), spec.get('attributes', {}))

            ele.text = spec.get('text', '')

            children = spec.get('children', [])

            if isinstance(children, dict):

                children = [children]

            for child in children:

                ele.append(TreeBuilder.build(child))

            return ele

        elif 'xml' in spec:

            return ET.XML(spec['xml'])

        elif 'comment' in spec:

            return ET.Comment(spec.get('comment'))

        else:

            raise ValueError('Invalid tree spec')