tomaae.homeassistant-mikrotik_router/custom_components/mikrotik_router/helper.py

"""Helper functions for Mikrotik Router."""

import logging
_LOGGER = logging.getLogger(__name__)


# ---------------------------
#   from_entry
# ---------------------------
def from_entry(entry, param, default=""):
    """Validate and return a value from a Mikrotik API dict"""
    if param not in entry:
        return default

    return entry[param]


# ---------------------------
#   from_entry_bool
# ---------------------------
def from_entry_bool(entry, param, default=False, reverse=False):
    """Validate and return a bool value from a Mikrotik API dict"""
    if param not in entry:
        return default

    if not reverse:
        ret = entry[param]
    else:
        if entry[param]:
            ret = False
        else:
            ret = True

    return ret


# ---------------------------
#   from_list
# ---------------------------
async def from_list(data=None, source=None, key=None, key_search=None, vals=[], ensure_vals=None, only=None, skip=None):
    if not source:
        return data

    keymap = generate_keymap(data, key_search)

    for entry in source:
        if only and not await matches_only(entry, only):
            continue

        if skip and await can_skip(entry, skip):
            continue

        if key or key_search:
            uid = await get_uid(entry, key, key_search, keymap)
            if not uid:
                continue

            if uid not in data:
                data[uid] = {}

        else:
            uid=None

        _LOGGER.debug("Processing entry {}, entry {}".format(source, entry))
        data = await fill_vals(data, uid, vals)
        if ensure_vals:
            data = await fill_ensure_vals(data, uid, ensure_vals)

    return data


# ---------------------------
#   get_uid
# ---------------------------
async def get_uid(entry, key, key_search, keymap):
    if not key_search:
        if key not in entry:
            return False

        if not entry[key]:
            return False

    else:
        if not keymap or key_search not in entry or entry[key_search] not in keymap:
            return False

        uid = keymap[entry[key_search]]

    return entry[key]


# ---------------------------
#   generate_keymap
# ---------------------------
async def generate_keymap(data, key_search):
    if not key_search:
        return None

    for uid in data:
        if key_search not in uid:
            continue

        keymap[data[uid]['name']] = data[uid]['default-name']

    return keymap


# ---------------------------
#   matches_only
# ---------------------------
async def matches_only(entry, only):
    can_continue = False
    for val in only:
        if val['name'] in entry and entry[val['name']] == val['value']:
            can_continue = True
        else:
            can_continue = False
            break

    return can_continue


# ---------------------------
#   can_skip
# ---------------------------
async def can_skip(entry, skip):
    can_skip = False
    for val in skip:
        if val['name'] in entry and entry[val['name']] == val['value']:
            can_skip = True
            break

    return can_skip


# ---------------------------
#   fill_vals
# ---------------------------
async def fill_vals(data, uid, vals):
    for val in vals:
        _name = val['name']
        _type = val['type'] if 'type' in val else 'str'
        _source = val['source'] if 'source' in val else _name

        if _type == 'str':
            _default = val['default'] if 'default' in val else ''
            if 'default_val' in val and val['default_val'] in val:
                _default = val[val['default_val']]

            if uid:
                data[uid][_name] = from_entry(entry, _source, default=_default)
            else:
                data[_name] = from_entry(entry, _source, default=_default)

        elif _type == 'bool':
            _default = val['default'] if 'default' in val else False
            _reverse = val['reverse'] if 'reverse' in val else False

            if uid:
                data[uid][_name] = from_entry_bool(entry, _source, default=_default, reverse=_reverse)
            else:
                data[_name] = from_entry_bool(entry, _source, default=_default, reverse=_reverse)

    return data


# ---------------------------
#   fill_ensure_vals
# ---------------------------
async def fill_ensure_vals(data, uid, ensure_vals):
    for val in ensure_vals:
        if uid:
            if val['name'] not in data[uid]:
                _default = val['default'] if 'default' in val else ''
                data[uid][val['name']] = _default
        else:
            if val['name'] not in data:
                _default = val['default'] if 'default' in val else ''
                data[val['name']] = _default

    return data