heppy.Doc

Complexity

Total Complexity

15

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

8 Methods

Rating	Name	Duplication	Size	Complexity
A	get()	0	2	1
A	build_module()	0	4	1
B	get_module()	0	11	5
A	addto()	0	5	3
A	set()	0	2	1
A	mget()	0	3	2
A	has()	0	2	1
A	addpair()	0	2	1

from importlib import import_module

class Doc:
    nsmap = {
        'builtin':      'builtin',
        'epp':          'urn:ietf:params:xml:ns:epp-1.0',
        'host':         'urn:ietf:params:xml:ns:host-1.0',
        'domain':       'urn:ietf:params:xml:ns:domain-1.0',
        'contact':      'urn:ietf:params:xml:ns:contact-1.0',
        'secDNS':       'urn:ietf:params:xml:ns:secDNS-1.1',
        'fee':          'urn:ietf:params:xml:ns:fee-0.7',
        'rgp':          'urn:ietf:params:xml:ns:rgp-1.0',
        'oxrs':         'urn:afilias:params:xml:ns:oxrs-1.1',
        'namestoreExt': 'http://www.verisign-grs.com/epp/namestoreExt-1.1',
    }

    okcodes = {
        '1000': 'completed',
        '1001': 'pending',
        '1300': 'no messages',
        '1301': 'ack to dequeue',
        '1500': 'ending session',
    }

    modules = {}

    def get_module(self, ns):
        if ns in self.nsmap:
            ns = self.nsmap[ns]
        if self.modules == {}:
            for name,nsi in self.nsmap.iteritems():
                self.modules[nsi] = name
        module = self.modules.get(ns)
        if isinstance(module, basestring):
            module = self.build_module(ns, module)
            self.modules[ns] = module
        return module

    def build_module(self, ns, name):
        lib = import_module('heppy.modules.' + name)
        type = getattr(lib, name)
        return type(ns)

    def get(self, name, default = None):
        return self.data.get(name, default)

    def has(self, name):
        return name in self.data

    @staticmethod
    def mget(data, map):
        return {k:data.get(v or k) for k,v in map.iteritems()}

    def set(self, name, value):
        self.data[name] = value

    def addto(self, name, values):
        if not name in self.data:
            self.data[name] = {}
        for k,v in values.iteritems():
            self.data[name][k] = v

    def addpair(self, name, value):
        self.addto(name, {value: value})