| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125 |
- """
- Config parser for Tuya Local devices.
- """
- import logging
- from base64 import b64decode, b64encode
- from collections.abc import Sequence
- from datetime import datetime
- from fnmatch import fnmatch
- from numbers import Number
- from os import scandir
- from os.path import dirname, exists, join, splitext
- from homeassistant.util import slugify
- from homeassistant.util.yaml import load_yaml
- import custom_components.tuya_local.devices as config_dir
- _LOGGER = logging.getLogger(__name__)
- def _typematch(vtype, value):
- # Workaround annoying legacy of bool being a subclass of int in Python
- if vtype is int and isinstance(value, bool):
- return False
- # Allow integers to pass as floats.
- if vtype is float and isinstance(value, Number):
- return True
- if isinstance(value, vtype):
- return True
- # Allow values embedded in strings if they can be converted
- # But not for bool, as everything can be converted to bool
- elif isinstance(value, str) and vtype is not bool:
- try:
- vtype(value)
- return True
- except ValueError:
- return False
- return False
- def _scale_range(r, s):
- "Scale range r by factor s"
- return (r["min"] / s, r["max"] / s)
- _unsigned_fmts = {
- 1: "B",
- 2: "H",
- 3: "3s",
- 4: "I",
- }
- _signed_fmts = {
- 1: "b",
- 2: "h",
- 3: "3s",
- 4: "i",
- }
- def _bytes_to_fmt(bytes, signed=False):
- """Convert a byte count to an unpack format."""
- fmt = _signed_fmts if signed else _unsigned_fmts
- if bytes in fmt:
- return fmt[bytes]
- else:
- return f"{bytes}s"
- def _equal_or_in(value1, values2):
- """Return true if value1 is the same as values2, or appears in values2."""
- if not isinstance(values2, str) and isinstance(values2, Sequence):
- return value1 in values2
- else:
- return value1 == values2
- def _remove_duplicates(seq):
- """Remove dulicates from seq, maintaining order."""
- if not seq:
- return []
- seen = set()
- adder = seen.add
- return [x for x in seq if not (x in seen or adder(x))]
- def to_signed(val, bits):
- """Convert unsigned int to signed 2's complement of given bit length."""
- if val & (1 << (bits - 1)):
- return val - (1 << bits)
- return val
- class TuyaDeviceConfig:
- """Representation of a device config for Tuya Local devices."""
- def __init__(self, fname):
- """Initialize the device config.
- Args:
- fname (string): The filename of the yaml config to load."""
- _CONFIG_DIR = dirname(config_dir.__file__)
- self._fname = fname
- filename = join(_CONFIG_DIR, fname)
- self._config = load_yaml(filename)
- self._reported_deprecated_primary = False
- _LOGGER.debug("Loaded device config %s", fname)
- @property
- def name(self):
- """Return the friendly name for this device."""
- return self._config["name"]
- @property
- def config(self):
- """Return the config file associated with this device."""
- return self._fname
- @property
- def config_type(self):
- """Return the config type associated with this device."""
- return splitext(self._fname)[0]
- @property
- def legacy_type(self):
- """Return the legacy conf_type associated with this device."""
- return self._config.get("legacy_type", self.config_type)
- @property
- def primary_entity(self):
- """Return the primary type of entity for this device."""
- if "primary_entity" not in self._config:
- # primary entity is a deprecated fallback, so if it is
- # missing, we need to log a warning about the missing entities
- # list.
- _LOGGER.error(f"{self.config_type}.yaml does not define an entities list.")
- return TuyaEntityConfig(self, self._config["entities"][0])
- if not self._reported_deprecated_primary:
- _LOGGER.warning(
- f"{self.config_type}.yaml distinguishes between primary"
- " and secondary_entities. This is deprecated, please"
- " modify it to use a single list."
- )
- self._reported_deprecated_primary = True
- return TuyaEntityConfig(self, self._config["primary_entity"])
- def all_entities(self):
- """Iterate through all entities for this device."""
- entities = self._config.get("entities")
- if not entities:
- yield self.primary_entity
- entities = self._config.get("secondary_entities", {})
- for e in entities:
- yield TuyaEntityConfig(self, e)
- def matches(self, dps, product_ids):
- """Determine whether this config matches the provided dps map or
- product ids."""
- product_match = False
- if product_ids:
- for p in self._config.get("products", []):
- if p.get("id", "MISSING_ID!?!") in product_ids:
- product_match = True
- required_dps = self._get_required_dps()
- missing_dps = [dp for dp in required_dps if dp.id not in dps.keys()]
- if len(missing_dps) > 0:
- _LOGGER.debug(
- "Not match for %s, missing required DPs: %s",
- self.name,
- [{dp.id: dp.type.__name__} for dp in missing_dps],
- )
- incorrect_type_dps = [
- dp
- for dp in self._get_all_dps()
- if dp.id in dps.keys() and not _typematch(dp.type, dps[dp.id])
- ]
- if len(incorrect_type_dps) > 0:
- _LOGGER.debug(
- "Not match for %s, DPs have incorrect type: %s",
- self.name,
- [{dp.id: dp.type.__name__} for dp in incorrect_type_dps],
- )
- if product_match:
- _LOGGER.warning(
- "Product matches %s but dps mismatched",
- self.name,
- )
- return False
- return product_match or len(missing_dps) == 0
- def _get_all_dps(self):
- all_dps_list = []
- all_dps_list += [d for dev in self.all_entities() for d in dev.dps()]
- return all_dps_list
- def _get_required_dps(self):
- required_dps_list = [d for d in self._get_all_dps() if not d.optional]
- return required_dps_list
- def _entity_match_analyse(self, entity, keys, matched, dps, product_match):
- """
- Determine whether this entity can be a match for the dps
- Args:
- entity - the TuyaEntityConfig to check against
- keys - the unmatched keys for the device
- matched - the matched keys for the device
- dps - the dps values to be matched
- Side Effects:
- Moves items from keys to matched if they match dps
- Return Value:
- True if all dps in entity could be matched to dps, False otherwise
- """
- all_dp = keys + matched
- for d in entity.dps():
- if (d.id not in all_dp and not d.optional and not product_match) or (
- d.id in all_dp and not _typematch(d.type, dps[d.id])
- ):
- return False
- if d.id in keys:
- matched.append(d.id)
- keys.remove(d.id)
- return True
- def match_quality(self, dps, product_ids=None):
- """Determine the match quality for the provided dps map and product ids."""
- product_match = 0
- if product_ids:
- for p in self._config.get("products", []):
- if p.get("id", "MISSING_ID!?!") in product_ids:
- product_match = 101
- keys = list(dps.keys())
- matched = []
- if "updated_at" in keys:
- keys.remove("updated_at")
- total = len(keys)
- if total < 1:
- return product_match
- for e in self.all_entities():
- if not self._entity_match_analyse(e, keys, matched, dps, product_match > 0):
- return 0
- return product_match or round((total - len(keys)) * 100 / total)
- class TuyaEntityConfig:
- """Representation of an entity config for a supported entity."""
- def __init__(self, device, config):
- self._device = device
- self._config = config
- @property
- def name(self):
- """The friendly name for this entity."""
- return self._config.get("name")
- @property
- def translation_key(self):
- """The translation key for this entity."""
- return self._config.get("translation_key")
- @property
- def translation_only_key(self):
- """The translation key for this entity, not used for unique_id"""
- return self._config.get("translation_only_key")
- @property
- def translation_placeholders(self):
- """The translation placeholders for this entity."""
- return self._config.get("translation_placeholders", {})
- def unique_id(self, device_uid):
- """Return a suitable unique_id for this entity."""
- return f"{device_uid}-{slugify(self.config_id)}"
- @property
- def entity_category(self):
- return self._config.get("category")
- @property
- def deprecated(self):
- """Return whether this entity is deprecated."""
- return "deprecated" in self._config.keys()
- @property
- def deprecation_message(self):
- """Return a deprecation message for this entity"""
- replacement = self._config.get(
- "deprecated", "nothing, this warning has been raised in error"
- )
- return (
- f"The use of {self.config_id} for {self._device.name} is "
- f"deprecated and should be replaced by {replacement}."
- )
- @property
- def entity(self):
- """The entity type of this entity."""
- return self._config["entity"]
- @property
- def config_id(self):
- """The identifier for this entity in the config."""
- own_name = self._config.get("name")
- if own_name:
- return f"{self.entity}_{slugify(own_name)}"
- if self.translation_key:
- slug = f"{self.entity}_{self.translation_key}"
- for key, value in self.translation_placeholders.items():
- if key in slug:
- slug = slug.replace(key, slugify(value))
- else:
- slug = f"{slug}_{value}"
- return slug
- elif self.device_class:
- return f"{self.entity}_{self.device_class}"
- return self.entity
- @property
- def device_class(self):
- """The device class of this entity."""
- return self._config.get("class")
- def icon(self, device):
- """Return the icon for this entity, with state as given."""
- icon = self._config.get("icon", None)
- priority = self._config.get("icon_priority", 100)
- for d in self.dps():
- rule = d.icon_rule(device)
- if rule and rule["priority"] < priority:
- icon = rule["icon"]
- priority = rule["priority"]
- return icon
- @property
- def mode(self):
- """Return the mode (used by Number entities)."""
- return self._config.get("mode")
- def dps(self):
- """Iterate through the list of dps for this entity."""
- for d in self._config["dps"]:
- yield TuyaDpsConfig(self, d)
- def find_dps(self, name):
- """Find a dps with the specified name."""
- for d in self.dps():
- if d.name == name:
- return d
- return None
- def available(self, device):
- """Return whether this entity should be available, with state as given."""
- avail_dp = self.find_dps("available")
- if avail_dp and device.has_returned_state:
- return avail_dp.get_value(device)
- return device.has_returned_state
- def enabled_by_default(self, device):
- """Return whether this entity should be disabled by default."""
- hidden = self._config.get("hidden", False)
- if hidden == "unavailable":
- avail_dp = self.find_dps("available")
- if not avail_dp:
- _LOGGER.warning(
- "Entity %s / %s has hidden: unavailable but no available dp defined",
- self._device.config_type,
- self.name,
- )
- hidden = not self.available(device)
- return not hidden and not self.deprecated
- class TuyaDpsConfig:
- """Representation of a dps config."""
- def __init__(self, entity, config):
- self._entity = entity
- self._config = config
- self.stringify = False
- @property
- def id(self):
- return str(self._config["id"])
- @property
- def type(self):
- t = self._config["type"]
- types = {
- "boolean": bool,
- "integer": int,
- "string": str,
- "float": float,
- "bitfield": int,
- "json": str,
- "base64": str,
- "utf16b64": str,
- "hex": str,
- "unixtime": int,
- }
- return types.get(t)
- @property
- def rawtype(self):
- return self._config["type"]
- @property
- def name(self):
- return self._config["name"]
- @property
- def optional(self):
- return self._config.get("optional", False)
- @property
- def persist(self):
- return self._config.get("persist", True)
- @property
- def force(self):
- return self._config.get("force", False)
- @property
- def sensitive(self):
- return self._config.get("sensitive", False)
- @property
- def format(self):
- fmt = self._config.get("format")
- if fmt:
- unpack_fmt = ">"
- ranges = []
- names = []
- for f in fmt:
- name = f.get("name")
- b = f.get("bytes", 1)
- r = f.get("range")
- if r:
- mn = r.get("min")
- mx = r.get("max")
- else:
- mn = 0
- mx = 256**b - 1
- unpack_fmt = unpack_fmt + _bytes_to_fmt(b, mn < 0)
- ranges.append({"min": mn, "max": mx})
- names.append(name)
- _LOGGER.debug("format of %s found", unpack_fmt)
- return {"format": unpack_fmt, "ranges": ranges, "names": names}
- return None
- @property
- def mask(self):
- mask = self._config.get("mask")
- if mask:
- return int(mask, 16)
- @property
- def endianness(self):
- endianness = self._config.get("endianness", "big")
- return endianness
- def get_value(self, device):
- """Return the value of the dps from the given device."""
- mask = self.mask
- bytevalue = self.decoded_value(device)
- if mask and isinstance(bytevalue, bytes):
- value = int.from_bytes(bytevalue, self.endianness)
- scale = mask & (1 + ~mask)
- raw_result = (value & mask) // scale
- # Insert signed interpretation here
- if self._config.get("mask_signed", False):
- # Count how many bits are set in the mask
- bit_count = mask.bit_count()
- raw_result = to_signed(raw_result, bit_count)
- return self._map_from_dps(raw_result, device)
- else:
- return self._map_from_dps(device.get_property(self.id), device)
- def decoded_value(self, device):
- v = self._map_from_dps(device.get_property(self.id), device)
- if self.rawtype == "hex" and isinstance(v, str):
- try:
- return bytes.fromhex(v)
- except ValueError:
- _LOGGER.warning(
- "%s sent invalid hex '%s' for %s",
- device.name,
- v,
- self.name,
- )
- return None
- elif self.rawtype == "base64" and isinstance(v, str):
- try:
- return b64decode(v)
- except ValueError:
- _LOGGER.warning(
- "%s sent invalid base64 '%s' for %s",
- device.name,
- v,
- self.name,
- )
- return None
- else:
- return v
- def encode_value(self, v):
- if self.rawtype == "hex":
- return v.hex()
- elif self.rawtype == "base64":
- return b64encode(v).decode("utf-8")
- elif self.rawtype == "unixtime" and isinstance(v, datetime):
- return v.timestamp()
- else:
- return v
- def _match(self, matchdata, value):
- """Return true val1 matches val2"""
- if self.rawtype == "bitfield" and matchdata:
- try:
- return (int(value) & int(matchdata)) != 0
- except (TypeError, ValueError):
- return False
- else:
- return str(value) == str(matchdata)
- async def async_set_value(self, device, value):
- """Set the value of the dps in the given device to given value."""
- if self.readonly:
- raise TypeError(f"{self.name} is read only")
- if self.invalid_for(value, device):
- raise AttributeError(f"{self.name} cannot be set at this time")
- settings = self.get_values_to_set(device, value)
- await device.async_set_properties(settings)
- def mapping_available(self, mapping, device):
- """Determine if this mapping should be available."""
- if "available" in mapping:
- avail_dp = self._entity.find_dps(mapping.get("available"))
- if avail_dp:
- return avail_dp.get_value(device)
- return True
- def should_show_mapping(self, mapping, device):
- """Determine if this mapping should be shown in the list of values."""
- if "value" not in mapping or mapping.get("hidden", False):
- return False
- return self.mapping_available(mapping, device)
- def values(self, device):
- """Return the possible values a dps can take."""
- if "mapping" not in self._config.keys():
- return []
- val = []
- for m in self._config["mapping"]:
- if self.should_show_mapping(m, device):
- val.append(m["value"])
- # If there is mirroring without override, include mirrored values
- elif "value_mirror" in m:
- r_dps = self._entity.find_dps(m["value_mirror"])
- if r_dps:
- val = val + r_dps.values(device)
- for c in m.get("conditions", {}):
- if self.should_show_mapping(c, device):
- val.append(c["value"])
- elif "value_mirror" in c:
- r_dps = self._entity.find_dps(c["value_mirror"])
- if r_dps:
- val = val + r_dps.values(device)
- cond = self._active_condition(m, device)
- if cond and "mapping" in cond:
- c_val = []
- for m2 in cond["mapping"]:
- if self.should_show_mapping(m2, device):
- c_val.append(m2["value"])
- elif "value_mirror" in m:
- r_dps = self._entity.find_dps(m["value_mirror"])
- if r_dps:
- c_val = c_val + r_dps.values(device)
- # if given, the conditional mapping is an override
- if c_val:
- val = c_val
- break
- return _remove_duplicates(val)
- @property
- def default(self):
- """Return the default value for a dp."""
- if "mapping" not in self._config.keys():
- _LOGGER.debug(
- "No mapping for %s, unable to determine default value",
- self.name,
- )
- return None
- for m in self._config["mapping"]:
- if m.get("default", False):
- return m.get("value", m.get("dps_val", None))
- for c in m.get("conditions", {}):
- if c.get("default", False):
- return c.get("value", m.get("value", m.get("dps_val", None)))
- def range(self, device, scaled=True):
- """Return the range for this dps if configured."""
- scale = self.scale(device) if scaled else 1
- mapping = self._find_map_for_dps(device.get_property(self.id), device)
- r = self._config.get("range")
- if mapping:
- cond = self._active_condition(mapping, device)
- if cond:
- r = cond.get("range", r)
- if r and "min" in r and "max" in r:
- return _scale_range(r, scale)
- else:
- return None
- def scale(self, device):
- scale = 1
- mapping = self._find_map_for_dps(device.get_property(self.id), device)
- if mapping:
- scale = mapping.get("scale", 1)
- cond = self._active_condition(mapping, device)
- if cond:
- scale = cond.get("scale", scale)
- return scale
- def precision(self, device):
- if self.type is int:
- scale = self.scale(device)
- precision = 0
- while scale > 1.0:
- scale /= 10.0
- precision += 1
- return precision
- @property
- def suggested_display_precision(self):
- return self._config.get("precision")
- def step(self, device, scaled=True):
- step = 1
- scale = self.scale(device) if scaled else 1
- mapping = self._find_map_for_dps(device.get_property(self.id), device)
- if mapping:
- step = mapping.get("step", 1)
- cond = self._active_condition(mapping, device)
- if cond:
- step = cond.get("step", step)
- if step != 1 or scale != 1:
- _LOGGER.debug(
- "Step for %s is %s with scale %s",
- self.name,
- step,
- scale,
- )
- return step / scale if scaled else step
- @property
- def readonly(self):
- return self._config.get("readonly", False)
- def invalid_for(self, value, device):
- mapping = self._find_map_for_value(value, device)
- if mapping:
- cond = self._active_condition(mapping, device)
- if cond:
- return cond.get("invalid", False)
- return False
- @property
- def hidden(self):
- return self._config.get("hidden", False)
- @property
- def unit(self):
- return self._config.get("unit")
- @property
- def state_class(self):
- """The state class of this measurement."""
- return self._config.get("class")
- def _find_map_for_dps(self, value, device):
- default = None
- for m in self._config.get("mapping", {}):
- if not self.mapping_available(m, device) and "conditions" not in m:
- continue
- if "dps_val" not in m:
- default = m
- elif self._match(m["dps_val"], value):
- return m
- return default
- def _correct_type(self, result):
- """Convert value to the correct type for this dp."""
- if self.type is int:
- _LOGGER.debug("Rounding %s", self.name)
- result = int(round(result))
- elif self.type is bool:
- result = True if result else False
- elif self.type is float:
- result = float(result)
- elif self.type is str:
- result = str(result)
- if self.rawtype == "utf16b64":
- result = b64encode(result.encode("utf-16-be")).decode("utf-8")
- if self.stringify:
- result = str(result)
- return result
- def _map_from_dps(self, val, device):
- if val is not None and self.type is not str and isinstance(val, str):
- try:
- val = self.type(val)
- self.stringify = True
- except ValueError:
- self.stringify = False
- else:
- self.stringify = False
- # decode utf-16 base64 strings first, so normal strings can be matched
- if self.rawtype == "utf16b64" and isinstance(val, str):
- try:
- val = b64decode(val).decode("utf-16-be")
- except ValueError:
- _LOGGER.warning("Invalid utf16b64 %s", val)
- result = val
- scale = self.scale(device)
- replaced = False
- mapping = self._find_map_for_dps(val, device)
- if mapping:
- invert = mapping.get("invert", False)
- redirect = mapping.get("value_redirect")
- mirror = mapping.get("value_mirror")
- replaced = "value" in mapping
- result = mapping.get("value", result)
- target_range = mapping.get("target_range")
- cond = self._active_condition(mapping, device)
- if cond:
- if cond.get("invalid", False):
- return None
- replaced = replaced or "value" in cond
- result = cond.get("value", result)
- redirect = cond.get("value_redirect", redirect)
- mirror = cond.get("value_mirror", mirror)
- target_range = cond.get("target_range", target_range)
- for m in cond.get("mapping", {}):
- if str(m.get("dps_val")) == str(result):
- replaced = "value" in m
- result = m.get("value", result)
- if redirect:
- _LOGGER.debug("Redirecting %s to %s", self.name, redirect)
- r_dps = self._entity.find_dps(redirect)
- if r_dps:
- return r_dps.get_value(device)
- if mirror:
- r_dps = self._entity.find_dps(mirror)
- if r_dps:
- return r_dps.get_value(device)
- if invert and isinstance(result, Number):
- r = self._config.get("range")
- if r and "min" in r and "max" in r:
- result = -1 * result + r["min"] + r["max"]
- replaced = True
- if target_range and isinstance(result, Number):
- r = self._config.get("range")
- if r and "max" in r and "max" in target_range:
- from_min = r.get("min", 0)
- from_max = r["max"]
- to_min = target_range.get("min", 0)
- to_max = target_range["max"]
- result = to_min + (
- (result - from_min) * (to_max - to_min) / (from_max - from_min)
- )
- replaced = True
- if scale != 1 and isinstance(result, Number):
- result = result / scale
- replaced = True
- if self.rawtype == "unixtime" and isinstance(result, int):
- try:
- result = datetime.fromtimestamp(result)
- replaced = True
- except Exception:
- _LOGGER.warning("Invalid timestamp %d", result)
- if replaced:
- _LOGGER.debug(
- "%s: Mapped dps %s value from %s to %s",
- self._entity._device.name,
- self.id,
- val,
- result,
- )
- return result
- def _find_map_for_value(self, value, device):
- default = None
- nearest = None
- distance = float("inf")
- for m in self._config.get("mapping", {}):
- # no reverse mapping of hidden values
- ignore = m.get("hidden", False) or not self.mapping_available(m, device)
- if "dps_val" not in m and not ignore:
- default = m
- # The following avoids further matching on the above case
- # and in the null mapping case, which is intended to be
- # a one-way map to prevent the entity showing as unavailable
- # when no value is being reported by the device.
- if m.get("dps_val") is None:
- ignore = True
- if "value" in m and str(m["value"]) == str(value) and not ignore:
- return m
- if (
- "value" in m
- and isinstance(m["value"], Number)
- and isinstance(value, Number)
- and not ignore
- ):
- d = abs(m["value"] - value)
- if d < distance:
- distance = d
- nearest = m
- if "value" not in m and "value_mirror" in m and not ignore:
- r_dps = self._entity.find_dps(m["value_mirror"])
- if r_dps and str(r_dps.get_value(device)) == str(value):
- return m
- for c in m.get("conditions", {}):
- if c.get("hidden", False) or not self.mapping_available(c, device):
- continue
- if "value" in c and str(c["value"]) == str(value):
- c_dp = self._entity.find_dps(m.get("constraint", self.name))
- # only consider the condition a match if we can change
- # the dp to match, or it already matches
- if (c_dp and c_dp.id != self.id and not c_dp.readonly) or (
- _equal_or_in(
- device.get_property(c_dp.id),
- c.get("dps_val"),
- )
- ):
- return m
- if "value" not in c and "value_mirror" in c:
- r_dps = self._entity.find_dps(c["value_mirror"])
- if r_dps and str(r_dps.get_value(device)) == str(value):
- return m
- if nearest:
- return nearest
- return default
- def _active_condition(self, mapping, device, value=None):
- constraint = mapping.get("constraint", self.name)
- conditions = mapping.get("conditions")
- c_match = None
- if constraint and conditions:
- c_dps = self._entity.find_dps(constraint)
- # base64 and hex have to be decoded
- c_val = (
- None
- if c_dps is None
- else (
- c_dps.get_value(device)
- if c_dps.rawtype == "base64" or c_dps.rawtype == "hex"
- else device.get_property(c_dps.id)
- )
- )
- for cond in conditions:
- if not self.mapping_available(cond, device):
- continue
- if c_val is not None and (_equal_or_in(c_val, cond.get("dps_val"))):
- c_match = cond
- # Case where matching None, need extra checks to ensure we
- # are not just defaulting and it is really a match
- elif (
- c_val is None
- and c_dps is not None
- and "dps_val" in cond
- and cond.get("dps_val") is None
- ):
- c_match = cond
- # when changing, another condition may become active
- # return that if it exists over a current condition
- if value is not None and value == cond.get("value"):
- return cond
- return c_match
- def get_values_to_set(self, device, value):
- """Return the dps values that would be set when setting to value"""
- result = value
- dps_map = {}
- if self.readonly:
- return dps_map
- mapping = self._find_map_for_value(value, device)
- scale = self.scale(device)
- mask = self.mask
- if mapping:
- replaced = False
- redirect = mapping.get("value_redirect")
- invert = mapping.get("invert", False)
- target_range = mapping.get("target_range")
- step = mapping.get("step")
- if not isinstance(step, Number):
- step = None
- if "dps_val" in mapping:
- result = mapping["dps_val"]
- replaced = True
- # Conditions may have side effect of setting another value.
- cond = self._active_condition(mapping, device, value)
- if cond:
- cval = cond.get("value")
- if cval is None:
- r_dps = cond.get("value_mirror")
- if r_dps:
- mirror = self._entity.find_dps(r_dps)
- if mirror:
- cval = mirror.get_value(device)
- if cval == value:
- c_dps = self._entity.find_dps(mapping.get("constraint", self.name))
- cond_dpsval = cond.get("dps_val")
- single_match = isinstance(cond_dpsval, str) or (
- not isinstance(cond_dpsval, Sequence)
- )
- if c_dps and c_dps.id != self.id and single_match:
- c_val = c_dps._map_from_dps(
- cond.get("dps_val", device.get_property(c_dps.id)),
- device,
- )
- dps_map.update(c_dps.get_values_to_set(device, c_val))
- # Allow simple conditional mapping overrides
- for m in cond.get("mapping", {}):
- if m.get("value") == value and not m.get("hidden", False):
- result = m.get("dps_val", result)
- step = cond.get("step", step)
- redirect = cond.get("value_redirect", redirect)
- target_range = cond.get("target_range", target_range)
- if redirect:
- _LOGGER.debug("Redirecting %s to %s", self.name, redirect)
- r_dps = self._entity.find_dps(redirect)
- if r_dps:
- return r_dps.get_values_to_set(device, value)
- if scale != 1 and isinstance(result, Number):
- _LOGGER.debug("Scaling %s by %s", result, scale)
- result = result * scale
- remap = self._find_map_for_value(result, device)
- if (
- remap
- and "dps_val" in remap
- and "dps_val" not in mapping
- and not remap.get("hidden", False)
- ):
- result = remap["dps_val"]
- replaced = True
- if target_range and isinstance(result, Number):
- r = self._config.get("range")
- if r and "max" in r and "max" in target_range:
- from_min = target_range.get("min", 0)
- from_max = target_range["max"]
- to_min = r.get("min", 0)
- to_max = r["max"]
- result = to_min + (
- (result - from_min) * (to_max - to_min) / (from_max - from_min)
- )
- replaced = True
- if invert:
- r = self._config.get("range")
- if r and "min" in r and "max" in r:
- result = -1 * result + r["min"] + r["max"]
- replaced = True
- if step and isinstance(result, Number):
- _LOGGER.debug("Stepping %s to %s", result, step)
- result = step * round(float(result) / step)
- remap = self._find_map_for_value(result, device)
- if (
- remap
- and "dps_val" in remap
- and "dps_val" not in mapping
- and not remap.get("hidden", False)
- ):
- result = remap["dps_val"]
- replaced = True
- if replaced:
- _LOGGER.debug(
- "%s: Mapped dps %s to %s from %s",
- self._entity._device.name,
- self.id,
- result,
- value,
- )
- r = self.range(device, scaled=False)
- if r and isinstance(result, Number):
- mn = r[0]
- mx = r[1]
- if round(result) < mn or round(result) > mx:
- # Output scaled values in the error message
- r = self.range(device, scaled=True)
- mn = r[0]
- mx = r[1]
- raise ValueError(f"{self.name} ({value}) must be between {mn} and {mx}")
- if mask and isinstance(result, bool):
- result = int(result)
- if mask and isinstance(result, Number):
- # mask is in hex, 2 digits/characters per byte
- hex_mask = self._config.get("mask")
- length = int(len(hex_mask) / 2)
- # Convert to int
- endianness = self.endianness
- mask_scale = mask & (1 + ~mask)
- current_value = int.from_bytes(self.decoded_value(device), endianness)
- result = (current_value & ~mask) | (mask & int(result * mask_scale))
- result = self.encode_value(result.to_bytes(length, endianness))
- dps_map[self.id] = self._correct_type(result)
- return dps_map
- def icon_rule(self, device):
- mapping = self._find_map_for_dps(device.get_property(self.id), device)
- icon = None
- priority = 100
- if mapping:
- icon = mapping.get("icon", icon)
- priority = mapping.get("icon_priority", 10 if icon else 100)
- cond = self._active_condition(mapping, device)
- if cond and cond.get("icon_priority", 10) < priority:
- icon = cond.get("icon", icon)
- priority = cond.get("icon_priority", 10 if icon else 100)
- return {"priority": priority, "icon": icon}
- def available_configs():
- """List the available config files."""
- _CONFIG_DIR = dirname(config_dir.__file__)
- for direntry in scandir(_CONFIG_DIR):
- if direntry.is_file() and fnmatch(direntry.name, "*.yaml"):
- yield direntry.name
- def possible_matches(dps, product_ids=None):
- """Return possible matching configs for a given set of
- dps values and product_ids."""
- for cfg in available_configs():
- parsed = TuyaDeviceConfig(cfg)
- try:
- if parsed.matches(dps, product_ids):
- yield parsed
- except TypeError:
- _LOGGER.error("Parse error in %s", cfg)
- def get_config(conf_type):
- """
- Return a config to use with config_type.
- """
- _CONFIG_DIR = dirname(config_dir.__file__)
- fname = conf_type + ".yaml"
- fpath = join(_CONFIG_DIR, fname)
- if exists(fpath):
- return TuyaDeviceConfig(fname)
- else:
- return config_for_legacy_use(conf_type)
- def config_for_legacy_use(conf_type):
- """
- Return a config to use with config_type for legacy transition.
- Note: as there are two variants for Kogan Socket, this is not guaranteed
- to be the correct config for the device, so only use it for looking up
- the legacy class during the transition period.
- """
- for cfg in available_configs():
- parsed = TuyaDeviceConfig(cfg)
- if parsed.legacy_type == conf_type:
- return parsed
- return None
|