""" Discovers Belkin Wemo devices. """

from . import SSDPDiscoverable


class Discoverable(SSDPDiscoverable):
    """ Adds support for discovering Belkin WeMo platform devices. """

    def info_from_entry(self, entry):
        """ Returns most important info from a uPnP entry. """
        device = entry.description['device']

        return (device['friendlyName'], device['modelName'],
                entry.values['location'])

    def get_entries(self):
        """ Returns all Belkin Wemo entries. """
        return self.find_by_device_description(
            {'manufacturer': 'Belkin International Inc.'})

""" Discovers Belkin Wemo devices. """

from . import SSDPDiscoverable


class Discoverable(SSDPDiscoverable):
    """ Adds support for discovering Belkin WeMo platform devices. """

    def info_from_entry(self, entry):
        """ Returns most important info from a uPnP entry. """
        device = entry.description['device']

        return (device['friendlyName'], device['modelName'],
                entry.values['location'], device['macAddress'])

    def get_entries(self):
        """ Returns all Belkin Wemo entries. """
        return self.find_by_device_description(
            {'manufacturer': 'Belkin International Inc.'})

Add MAC address to wemo discovery attributes
Add MAC address to wemo discovery attributes
