Move HassIntent handler code into helpers/intent

homeassistant/components/__init__.py

@@ -14,7 +14,11 @@
 import homeassistant.core as ha
 import homeassistant.config as conf_util
 from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.service import extract_entity_ids
+from homeassistant.helpers import intent
+from homeassistant.helpers.intent import (INTENT_TURN_ON,
+                                          INTENT_TURN_OFF, INTENT_TOGGLE)
 from homeassistant.const import (
     ATTR_ENTITY_ID, SERVICE_TURN_ON, SERVICE_TURN_OFF, SERVICE_TOGGLE,
     SERVICE_HOMEASSISTANT_STOP, SERVICE_HOMEASSISTANT_RESTART,
@@ -25,7 +29,6 @@
 SERVICE_RELOAD_CORE_CONFIG = 'reload_core_config'
 SERVICE_CHECK_CONFIG = 'check_config'
 
-
 def is_on(hass, entity_id=None):
     """Load up the module to call the is_on method.
 
@@ -154,6 +157,10 @@ def async_handle_turn_service(service):
         ha.DOMAIN, SERVICE_TURN_ON, async_handle_turn_service)
     hass.services.async_register(
         ha.DOMAIN, SERVICE_TOGGLE, async_handle_turn_service)
+    hass.helpers.intent.async_register(TurnOnIntent())
+    hass.helpers.intent.async_register(TurnOffIntent())
+    hass.helpers.intent.async_register(ToggleIntent())
+
 
     @asyncio.coroutine
     def async_handle_core_service(call):
@@ -200,3 +207,39 @@ def async_handle_reload_config(call):
         ha.DOMAIN, SERVICE_RELOAD_CORE_CONFIG, async_handle_reload_config)
 
     return True
+
+
+class TurnOnIntent(intent.ServiceIntentHandler):
+    """Handle component turn intents."""
+
+    intent_type = INTENT_TURN_ON
+    slot_schema = {
+        'name': cv.string,
+    }
+    domain = ha.DOMAIN
+    service = SERVICE_TURN_ON
+    response = "Turned on {}"
+
+
+class TurnOffIntent(intent.ServiceIntentHandler):
+    """Handle component turn intents."""
+
+    intent_type = INTENT_TURN_OFF
+    slot_schema = {
+        'name': cv.string,
+    }
+    domain = ha.DOMAIN
+    service = SERVICE_TURN_OFF
+    response = "Turned off {}"
+
+
+class ToggleIntent(intent.ServiceIntentHandler):
+    """Handle component turn intents."""
+
+    intent_type = INTENT_TOGGLE
+    slot_schema = {
+        'name': cv.string,
+    }
+    domain = ha.DOMAIN
+    service = SERVICE_TOGGLE
+    response = "Toggled {}"

homeassistant/components/conversation.py

@@ -7,30 +7,24 @@
 import asyncio
 import logging
 import re
-import warnings
 
 import voluptuous as vol
 
 from homeassistant import core
 from homeassistant.components import http
-from homeassistant.const import (
-    ATTR_ENTITY_ID, SERVICE_TURN_OFF, SERVICE_TURN_ON)
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers import intent
+from homeassistant.helpers.intent import (INTENT_TURN_ON, INTENT_TURN_OFF,
+                                          INTENT_TOGGLE)
 from homeassistant.loader import bind_hass
 
-REQUIREMENTS = ['fuzzywuzzy==0.16.0']
-
 _LOGGER = logging.getLogger(__name__)
 
 ATTR_TEXT = 'text'
 
 DEPENDENCIES = ['http']
 DOMAIN = 'conversation'
 
-INTENT_TURN_OFF = 'HassTurnOff'
-INTENT_TURN_ON = 'HassTurnOn'
-
 REGEX_TURN_COMMAND = re.compile(r'turn (?P<name>(?: |\w)+) (?P<command>\w+)')
 REGEX_TYPE = type(re.compile(''))
 
@@ -50,7 +44,7 @@
 @core.callback
 @bind_hass
 def async_register(hass, intent_type, utterances):
-    """Register an intent.
+    """Register utterances and any custom intents.
 
     Registrations don't require conversations to be loaded. They will become
     active once the conversation component is loaded.
@@ -75,8 +69,6 @@ def async_register(hass, intent_type, utterances):
 @asyncio.coroutine
 def async_setup(hass, config):
     """Register the process service."""
-    warnings.filterwarnings('ignore', module='fuzzywuzzy')
-
     config = config.get(DOMAIN, {})
     intents = hass.data.get(DOMAIN)
 
@@ -102,12 +94,12 @@ def process(service):
 
     hass.http.register_view(ConversationProcessView)
 
-    hass.helpers.intent.async_register(TurnOnIntent())
-    hass.helpers.intent.async_register(TurnOffIntent())
     async_register(hass, INTENT_TURN_ON,
                    ['Turn {name} on', 'Turn on {name}'])
-    async_register(hass, INTENT_TURN_OFF, [
-        'Turn {name} off', 'Turn off {name}'])
+    async_register(hass, INTENT_TURN_OFF,
+                   ['Turn {name} off', 'Turn off {name}'])
+    async_register(hass, INTENT_TOGGLE,
+                   ['Toggle {name}', '{name} toggle'])
 
     return True
 
@@ -151,79 +143,6 @@ def _process(hass, text):
             return response
 
 
-@core.callback
-def _match_entity(hass, name):
-    """Match a name to an entity."""
-    from fuzzywuzzy import process as fuzzyExtract
-    entities = {state.entity_id: state.name for state
-                in hass.states.async_all()}
-    entity_id = fuzzyExtract.extractOne(
-        name, entities, score_cutoff=65)[2]
-    return hass.states.get(entity_id) if entity_id else None
-
-
-class TurnOnIntent(intent.IntentHandler):
-    """Handle turning item on intents."""
-
-    intent_type = INTENT_TURN_ON
-    slot_schema = {
-        'name': cv.string,
-    }
-
-    @asyncio.coroutine
-    def async_handle(self, intent_obj):
-        """Handle turn on intent."""
-        hass = intent_obj.hass
-        slots = self.async_validate_slots(intent_obj.slots)
-        name = slots['name']['value']
-        entity = _match_entity(hass, name)
-
-        if not entity:
-            _LOGGER.error("Could not find entity id for %s", name)
-            return None
-
-        yield from hass.services.async_call(
-            core.DOMAIN, SERVICE_TURN_ON, {
-                ATTR_ENTITY_ID: entity.entity_id,
-            }, blocking=True)
-
-        response = intent_obj.create_response()
-        response.async_set_speech(
-            'Turned on {}'.format(entity.name))
-        return response
-
-
-class TurnOffIntent(intent.IntentHandler):
-    """Handle turning item off intents."""
-
-    intent_type = INTENT_TURN_OFF
-    slot_schema = {
-        'name': cv.string,
-    }
-
-    @asyncio.coroutine
-    def async_handle(self, intent_obj):
-        """Handle turn off intent."""
-        hass = intent_obj.hass
-        slots = self.async_validate_slots(intent_obj.slots)
-        name = slots['name']['value']
-        entity = _match_entity(hass, name)
-
-        if not entity:
-            _LOGGER.error("Could not find entity id for %s", name)
-            return None
-
-        yield from hass.services.async_call(
-            core.DOMAIN, SERVICE_TURN_OFF, {
-                ATTR_ENTITY_ID: entity.entity_id,
-            }, blocking=True)
-
-        response = intent_obj.create_response()
-        response.async_set_speech(
-            'Turned off {}'.format(entity.name))
-        return response
-
-
 class ConversationProcessView(http.HomeAssistantView):
     """View to retrieve shopping list content."""
 

homeassistant/helpers/intent.py

@@ -1,20 +1,27 @@
 """Module to coordinate user intentions."""
 import asyncio
 import logging
+import re
 
 import voluptuous as vol
 
 from homeassistant.core import callback
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.loader import bind_hass
+from homeassistant.const import ATTR_ENTITY_ID
 
-
-DATA_KEY = 'intent'
 _LOGGER = logging.getLogger(__name__)
 
+# #### INTENTS ####
+INTENT_TURN_OFF = 'HassTurnOff'
+INTENT_TURN_ON = 'HassTurnOn'
+INTENT_TOGGLE = 'HassToggle'
+
 SLOT_SCHEMA = vol.Schema({
 }, extra=vol.ALLOW_EXTRA)
 
+DATA_KEY = 'intent'
+
 SPEECH_TYPE_PLAIN = 'plain'
 SPEECH_TYPE_SSML = 'ssml'
 
@@ -87,7 +94,7 @@ class IntentHandler:
     intent_type = None
     slot_schema = None
     _slot_schema = None
-    platforms = None
+    platforms = []
 
     @callback
     def async_can_handle(self, intent_obj):
@@ -117,6 +124,60 @@ def __repr__(self):
         return '<{} - {}>'.format(self.__class__.__name__, self.intent_type)
 
 
+def fuzzyfinder(name, entities):
+    """Semi fuzzy matching function."""
+    matches = []
+    pattern = '.*?'.join(name)
+    regex = re.compile(pattern)
+    for entity in entities:
+        match = regex.search(entity)
+        if match:
+            matches.append((len(match.group()), match.start(), entity))
+    return [x for _, _, x in sorted(matches)]
+
+
+class ServiceIntentHandler(IntentHandler):
+    """Intent handler registration."""
+
+    domain = None
+    service = None
+    response = ''
+
+    @asyncio.coroutine
+    def async_handle(self, intent_obj):
+        """Handle the hass intent."""
+        hass = intent_obj.hass
+        slots = self.async_validate_slots(intent_obj.slots)
+        response = intent_obj.create_response()
+
+        name = slots['name']['value']
+        entities = {state.entity_id: state.name for state
+                    in hass.states.async_all()}
+        entity_name = name.replace(' ', '_').lower()
+        entity_name = entity_name.replace('the_', '')
+
+        matches = fuzzyfinder(entity_name, entities)
+        entity_id = matches[0] if matches else None
+        _LOGGER.debug("%s matched entity: %s", name, entity_id)
+
+        response = intent_obj.create_response()
+        if not entity_id:
+            response.async_set_speech(
+                "Could not find entity id matching {}".format(name))
+            _LOGGER.error("Could not find entity id matching %s (%s)", name,
+                          entity_name)
+            return response
+
+        yield from hass.services.async_call(
+            self.domain, self.service, {
+                ATTR_ENTITY_ID: entity_id
+            }, blocking=True)
+
+        response.async_set_speech(
+            self.response.format(name))
+        return response
+
+
 class Intent:
     """Hold the intent."""
 

requirements_all.txt

@@ -294,9 +294,6 @@ fritzhome==1.0.4
 # homeassistant.components.media_player.frontier_silicon
 fsapi==0.0.7
 
-# homeassistant.components.conversation
-fuzzywuzzy==0.16.0
-
 # homeassistant.components.tts.google
 gTTS-token==1.1.1
 

requirements_test_all.txt

@@ -56,9 +56,6 @@ evohomeclient==0.2.5
 # homeassistant.components.sensor.geo_rss_events
 feedparser==5.2.1
 
-# homeassistant.components.conversation
-fuzzywuzzy==0.16.0
-
 # homeassistant.components.tts.google
 gTTS-token==1.1.1
 

script/gen_requirements_all.py

@@ -66,6 +66,7 @@
     'pushbullet.py',
     'py-canary',
     'pydispatcher',
+    'python-Levenshtein',
     'PyJWT',
     'pylitejet',
     'pymonoprice',