BATCAVE/ffstatus/__init__.py

from copy import deepcopy

from .alfred import AlfredParser
from .batman import BatmanParser
from .dashing import DashingClient
from .graphite import GraphitePush

__all__ = [
	'AlfredParser', 'BatmanParser',
	'DashingClient', 'GraphitePush',
	'dict_merge', 'merge_alfred_batman', 'mac2id' ]

def mac2id(mac):
	return mac.lower().replace(':', '')

def dict_merge(a, b):
    '''recursively merges dict's. not just simple a['key'] = b['key'], if
    both a and bhave a key who's value is a dict then dict_merge is called
    on both values and the result stored in the returned dictionary.'''
    if not isinstance(b, dict):
        return b
    result = deepcopy(a)
    for k, v in b.iteritems():
        if k in result and isinstance(result[k], dict):
                result[k] = dict_merge(result[k], v)
        else:
            result[k] = deepcopy(v)
    return result

def merge_alfred_batman(alfreddata, batmandata):
	merged = {}
	for nodeid in alfreddata:
		nodeinfo = alfreddata[nodeid]
		candidates = set()
		candidates.add(nodeid)
		if 'mac' in nodeinfo: candidates.add(mac2id(nodeinfo['mac']))
		if 'macs' in nodeinfo:
			for mac in nodeinfo['macs']:
				candidates.add(mac2id(mac))

		for candidate in candidates:
			if candidate in batmandata:
				nodeinfo = dict_merge(nodeinfo, batmandata[candidate])

		merged[nodeid] = nodeinfo

	return merged