old_contents: """ Support for WeMo switchces. """
  import logging
  
  from homeassistant.helpers import ToggleDevice
  from homeassistant.const import ATTR_FRIENDLY_NAME, CONF_HOSTS
  from homeassistant.components.switch import (
      ATTR_TODAY_MWH, ATTR_CURRENT_POWER_MWH)
  
  
  # pylint: disable=unused-argument
  def get_devices(hass, config):
      """ Find and return WeMo switches. """
  
      try:
          # Pylint does not play nice if not every folders has an __init__.py
          # pylint: disable=no-name-in-module, import-error
          import homeassistant.external.pywemo.pywemo as pywemo
      except ImportError:
          logging.getLogger(__name__).exception((
              "Failed to import pywemo. "
              "Did you maybe not run `git submodule init` "
              "and `git submodule update`?"))
  
          return []
  
      if CONF_HOSTS in config:
          switches = (pywemo.device_from_host(host) for host
                      in config[CONF_HOSTS].split(","))
  
      else:
          logging.getLogger(__name__).info("Scanning for WeMo devices")
          switches = pywemo.discover_devices()
  
      # Filter out the switches and wrap in WemoSwitch object
      return [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
  
      def get_name(self):
          """ Returns the name of the switch if any. """
          return self.wemo.name
  
      def turn_on(self, **kwargs):
          """ Turns the switch on. """
          self.wemo.on()
  
      def turn_off(self):
          """ Turns the switch off. """
          self.wemo.off()
  
      def is_on(self):
          """ True if switch is on. """
          return self.wemo.get_state(True)
  
      def get_state_attributes(self):
          """ Returns optional state attributes. """
          if self.wemo.model.startswith('Belkin Insight'):
              cur_info = self.wemo.insight_params
  
              return {
                  ATTR_FRIENDLY_NAME: self.wemo.name,
                  ATTR_CURRENT_POWER_MWH: cur_info['currentpower'],
                  ATTR_TODAY_MWH: cur_info['todaymw']
              }
          else:
              return {ATTR_FRIENDLY_NAME: self.wemo.name}
  
new_contents: """ Support for WeMo switchces. """
  import logging
  
  from homeassistant.helpers import ToggleDevice
  from homeassistant.const import ATTR_FRIENDLY_NAME
  from homeassistant.components.switch import (
      ATTR_TODAY_MWH, ATTR_CURRENT_POWER_MWH)
  
  
  # pylint: disable=unused-argument
  def get_devices(hass, config):
      """ Find and return WeMo switches. """
  
      try:
          # Pylint does not play nice if not every folders has an __init__.py
          # pylint: disable=no-name-in-module, import-error
          import homeassistant.external.pywemo.pywemo as pywemo
      except ImportError:
          logging.getLogger(__name__).exception((
              "Failed to import pywemo. "
              "Did you maybe not run `git submodule init` "
              "and `git submodule update`?"))
  
          return []
  
      logging.getLogger(__name__).info("Scanning for WeMo devices")
      switches = pywemo.discover_devices()
  
      # Filter out the switches and wrap in WemoSwitch object
      return [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
  
      def get_name(self):
          """ Returns the name of the switch if any. """
          return self.wemo.name
  
      def turn_on(self, **kwargs):
          """ Turns the switch on. """
          self.wemo.on()
  
      def turn_off(self):
          """ Turns the switch off. """
          self.wemo.off()
  
      def is_on(self):
          """ True if switch is on. """
          return self.wemo.get_state(True)
  
      def get_state_attributes(self):
          """ Returns optional state attributes. """
          if self.wemo.model.startswith('Belkin Insight'):
              cur_info = self.wemo.insight_params
  
              return {
                  ATTR_FRIENDLY_NAME: self.wemo.name,
                  ATTR_CURRENT_POWER_MWH: cur_info['currentpower'],
                  ATTR_TODAY_MWH: cur_info['todaymw']
              }
          else:
              return {ATTR_FRIENDLY_NAME: self.wemo.name}
  
subject: Remove unsupported specifying hosts for wemo