import functools from typing import Iterator def polling(func): """ Offers memoization for a set into get operation. If sync clear dirty parameters before fetching new value. Useful for loop getting if not running callbacks """ @functools.wraps(func) def wrapper(*args, **kwargs): get = func.__name__ == "get" mb_get = func.__name__ == "get_buttonstatus" remote, *remaining = args if get: param, *rem = remaining elif mb_get: id, mode, *rem = remaining param = f"mb_{id}_{mode}" if param in remote.cache: return remote.cache.pop(param) if remote.sync: remote.clear_dirty() return func(*args, **kwargs) return wrapper def script(func): """Convert dictionary to script""" def wrapper(*args): remote, script = args if isinstance(script, dict): params = "" for key, val in script.items(): obj, m2, *rem = key.split("-") index = int(m2) if m2.isnumeric() else int(*rem) params += ";".join( f"{obj}{f'.{m2}stream' if not m2.isnumeric() else ''}[{index}].{k}={int(v) if isinstance(v, bool) else v}" for k, v in val.items() ) params += ";" script = params return func(remote, script) return wrapper def comp(t0: tuple, t1: tuple) -> Iterator[bool]: """ Generator function, accepts two tuples. Evaluates equality of each member in both tuples. """ for a, b in zip(t0, t1): yield a == b