mdt/discoverer.py

from typing import cast
from zeroconf import ServiceBrowser, Zeroconf

class Discoverer:
    def __init__(self):
        self.zeroconf = Zeroconf()
        self.discoveries = {}
        browser = ServiceBrowser(self.zeroconf, "_google_mdt._tcp.local.", self)

    def add_service(self, zeroconf, type, name):
        info = self.zeroconf.get_service_info(type, name)
        if info:
            hostname = info.server.split('.')[0]
            address = socket.inet_ntoa(cast(bytes, info.address))
            self.discoveries[hostname] = address

    def remove_service(self, zeroconf, type, name):
        info = self.zeroconf.get_service_info(type, name)
        del(self.discoveries[info.server])