mirror.MikroTik/tomaae.homeassistant-mikrotik_router
1
0
Fork 0
mirror of https://github.com/tomaae/homeassistant-mikrotik_router.git synced 2025-06-28 11:40:08 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

reformatted code using black

Browse source
This commit is contained in:
tomaae 2020-03-16 04:51:41 +01:00
parent cd4d7c4e81
commit e418c20252
8 changed files with 512 additions and 322 deletions
Download patch file Download diff file Expand all files Collapse all files

99
custom_components/mikrotik_router/helper.py
View file

@ -1,6 +1,7 @@
"""Helper functions for Mikrotik Router."""
import logging
_LOGGER = logging.getLogger(__name__)
@ -37,7 +38,17 @@ def from_entry_bool(entry, param, default=False, reverse=False) -> bool:
# ---------------------------
# parse_api
# ---------------------------
def parse_api(data=None, source=None, key=None, key_search=None, vals=None, val_proc=None, ensure_vals=None, only=None, skip=None) -> dict:
def parse_api(
data=None,
source=None,
key=None,
key_search=None,
vals=None,
val_proc=None,
ensure_vals=None,
only=None,
skip=None,
) -> dict:
"""Get data from API"""
if not source:
if not key and not key_search:
@ -122,7 +133,7 @@ def matches_only(entry, only) -> bool:
"""Return True if all variables are matched"""
ret = False
for val in only:
if val['key'] in entry and entry[val['key']] == val['value']:
if val["key"] in entry and entry[val["key"]] == val["value"]:
ret = True
else:
ret = False
@ -138,7 +149,7 @@ def can_skip(entry, skip) -> bool:
"""Return True if at least one variable matches"""
ret = False
for val in skip:
if val['name'] in entry and entry[val['name']] == val['value']:
if val["name"] in entry and entry[val["name"]] == val["value"]:
ret = True
break
@ -151,23 +162,25 @@ def can_skip(entry, skip) -> bool:
def fill_defaults(data, vals) -> dict:
"""Fill defaults if source is not present"""
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
_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 _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 _name not in data:
data[_name] = from_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
elif _type == "bool":
_default = val["default"] if "default" in val else False
_reverse = val["reverse"] if "reverse" in val else False
if _name not in data:
data[_name] = from_entry_bool([], _source, default=_default, reverse=_reverse)
data[_name] = from_entry_bool(
[], _source, default=_default, reverse=_reverse
)
return data
@ -178,28 +191,32 @@ def fill_defaults(data, vals) -> dict:
def fill_vals(data, entry, uid, vals) -> dict:
"""Fill all data"""
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
_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 _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
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)
data[uid][_name] = from_entry_bool(
entry, _source, default=_default, reverse=_reverse
)
else:
data[_name] = from_entry_bool(entry, _source, default=_default, reverse=_reverse)
data[_name] = from_entry_bool(
entry, _source, default=_default, reverse=_reverse
)
return data
@ -211,13 +228,13 @@ def fill_ensure_vals(data, uid, ensure_vals) -> dict:
"""Add required keys which are not available in data"""
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
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
if val["name"] not in data:
_default = val["default"] if "default" in val else ""
data[val["name"]] = _default
return data
@ -233,27 +250,27 @@ def fill_vals_proc(data, uid, vals_proc) -> dict:
_action = None
_value = None
for val in val_sub:
if 'name' in val:
_name = val['name']
if "name" in val:
_name = val["name"]
continue
if 'action' in val:
_action = val['action']
if "action" in val:
_action = val["action"]
continue
if not _name and not _action:
break
if _action == 'combine':
if 'key' in val:
tmp = _data[val['key']] if val['key'] in _data else 'unknown'
if _action == "combine":
if "key" in val:
tmp = _data[val["key"]] if val["key"] in _data else "unknown"
if not _value:
_value = tmp
else:
_value = "{}{}".format(_value, tmp)
if 'text' in val:
tmp = val['text']
if "text" in val:
tmp = val["text"]
if not _value:
_value = tmp
else: