capy.wifi v0.5.4

def scan(timeout=1, _wanted=None):
    found, me = {}, _local_ip()
    with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
        end = time.monotonic() + timeout
        next_ask = 0
        while time.monotonic() < end:
            # On redemande 3 fois : en wifi, un message UDP peut se perdre
            if time.monotonic() >= next_ask:
                for target in ("255.255.255.255", "127.0.0.1"):  # Tout le wifi + cet appareil (même sans wifi)
                    try:
                        s.sendto(b"capy?", (target, DISCOVERY_PORT))
                    except OSError:
                        pass
                next_ask = time.monotonic() + timeout / 3
            s.settimeout(max(min(end, next_ask) - time.monotonic(), 0.001))
            try:
                data, (ip, _) = s.recvfrom(1024)
                info = jsonlib.loads(data)
                name, port = info["name"], int(info["port"])
            except socket.timeout:
                continue
            except (OSError, ValueError, KeyError, TypeError):
                continue
            # Serveur sur cet appareil -> 127.0.0.1 (20x plus rapide que passer par l'IP du wifi)
            found.setdefault(name, ("127.0.0.1" if ip == me else ip, port))
            _secure.setdefault(name, info.get("tls") is True)
            if name == _wanted:  # find() : on a ce qu'on cherche, inutile d'attendre
                break
    _found.update(found)
    return found

def find(name, timeout=1):
    if name not in _found:
        found = scan(timeout, _wanted=name)
        if name not in found:
            raise ConnectionError(f"capy.wifi : aucun serveur {name!r} sur le wifi (trouvés : {', '.join(found) or 'aucun'})")
    return _found[name]

def call(name, action, /, tls=True, **params) -> Any:
    for attempt in range(2):
        host, port = find(name)
        if _secure.get(name, True) != tls:
            state = "n'est pas chiffré : mets tls=False" if tls else "est chiffré : enlève tls=False"
            raise ConnectionError(f"capy.wifi : le serveur {name!r} {state} dans capy.wifi.call()")
        try:
            return capy._call(host, port, action, params, tls, label=f"wifi:{name}", who="capy.wifi")
        except ConnectionError as e:
            if attempt or "chiffr" in str(e):
                raise
            _found.pop(name, None)  # Serveur relancé (autre port, autre IP) -> on le recherche une fois
            _secure.pop(name, None)