old_contents: """ Support for WeMo switchces. """
  import logging
  
  from homeassistant.helpers import ToggleDevice
  from homeassistant.components.switch import (
      ATTR_TODAY_MWH, ATTR_CURRENT_POWER_MWH)
  
  
  # pylint: disable=unused-argument
  def setup_platform(hass, config, add_devices_callback, discovery_info=None):
      """ Find and return wemo switches. """
      try:
          # pylint: disable=no-name-in-module, import-error
          import homeassistant.external.pywemo.pywemo as pywemo
          import homeassistant.external.pywemo.pywemo.discovery as discovery
      except ImportError:
          logging.getLogger(__name__).exception((
              "Failed to import pywemo. "
              "Did you maybe not run `git submodule init` "
              "and `git submodule update`?"))
  
          return
  
      if discovery_info is not None:
          device = discovery.device_from_description(discovery_info)
  
          if device:
              add_devices_callback([device])
  
          return
  
      logging.getLogger(__name__).info("Scanning for WeMo devices")
      switches = pywemo.discover_devices()
  
      # Filter out the switches and wrap in WemoSwitch object
      add_devices_callback(
          [WemoSwitch(switch) for switch in switches
           if isinstance(switch, pywemo.Switch)])
  
  
  class WemoSwitch(ToggleDevice):
      """ represents a WeMo switch within home assistant. """
      def __init__(self, wemo):
          self.wemo = wemo
  
      @property
      def unique_id(self):
          """ Returns the id of this WeMo switch """
          return "{}.{}".format(self.__class__, self.wemo.serialnumber)
  
      @property
      def name(self):
          """ Returns the name of the switch if any. """
          return self.wemo.name
  
      @property
      def state_attributes(self):
          """ Returns optional state attributes. """
          if self.wemo.model.startswith('Belkin Insight'):
              cur_info = self.wemo.insight_params
  
              return {
                  ATTR_CURRENT_POWER_MWH: cur_info['currentpower'],
                  ATTR_TODAY_MWH: cur_info['todaymw']
              }
  
      @property
      def is_on(self):
          """ True if switch is on. """
          return self.wemo.get_state()
  
      def turn_on(self, **kwargs):
          """ Turns the switch on. """
          self.wemo.on()
  
      def turn_off(self):
          """ Turns the switch off. """
          self.wemo.off()
  
      def update(self):
          """ Update Wemo state. """
          self.wemo.get_state(True)
  
new_contents: """ Support for WeMo switchces. """
  import logging
  
  from homeassistant.helpers import ToggleDevice
  from homeassistant.components.switch import (
      ATTR_TODAY_MWH, ATTR_CURRENT_POWER_MWH)
  
  
  # pylint: disable=unused-argument
  def setup_platform(hass, config, add_devices_callback, discovery_info=None):
      """ Find and return wemo switches. """
      try:
          # pylint: disable=no-name-in-module, import-error
          import homeassistant.external.pywemo.pywemo as pywemo
          import homeassistant.external.pywemo.pywemo.discovery as discovery
      except ImportError:
          logging.getLogger(__name__).exception((
              "Failed to import pywemo. "
              "Did you maybe not run `git submodule init` "
              "and `git submodule update`?"))
  
          return
  
      if discovery_info is not None:
          device = discovery.device_from_description(discovery_info)
  
          if device:
              add_devices_callback([WemoSwitch(device)])
  
          return
  
      logging.getLogger(__name__).info("Scanning for WeMo devices")
      switches = pywemo.discover_devices()
  
      # Filter out the switches and wrap in WemoSwitch object
      add_devices_callback(
          [WemoSwitch(switch) for switch in switches
           if isinstance(switch, pywemo.Switch)])
  
  
  class WemoSwitch(ToggleDevice):
      """ represents a WeMo switch within home assistant. """
      def __init__(self, wemo):
          self.wemo = wemo
  
      @property
      def unique_id(self):
          """ Returns the id of this WeMo switch """
          return "{}.{}".format(self.__class__, self.wemo.serialnumber)
  
      @property
      def name(self):
          """ Returns the name of the switch if any. """
          return self.wemo.name
  
      @property
      def state_attributes(self):
          """ Returns optional state attributes. """
          if self.wemo.model.startswith('Belkin Insight'):
              cur_info = self.wemo.insight_params
  
              return {
                  ATTR_CURRENT_POWER_MWH: cur_info['currentpower'],
                  ATTR_TODAY_MWH: cur_info['todaymw']
              }
  
      @property
      def is_on(self):
          """ True if switch is on. """
          return self.wemo.get_state()
  
      def turn_on(self, **kwargs):
          """ Turns the switch on. """
          self.wemo.on()
  
      def turn_off(self):
          """ Turns the switch off. """
          self.wemo.off()
  
      def update(self):
          """ Update Wemo state. """
          self.wemo.get_state(True)
  
subject: Fix a WeMo discovery issue