コード例 #1
0
class Skill_Hue:
    def __init__(self):
        config_dict = SnipsConfigParser.read_configuration_file()
        bridge_ip = config_dict.get(BRIDGE_IP_KEY, None)
        api_key = config_dict.get(API_KEY_KEY, None)
        (bridge_ip, api_key, config_changed) = HueSetup.validate_config(bridge_ip, api_key)
        if config_changed:
            SnipsConfigParser.write_configuration_file(bridge_ip, api_key)
        self.snipshue = SnipsHue(bridge_ip, api_key)

        with Hermes(MQTT_ADDR) as h:
            h.subscribe_intents(self.callback).start()

    # section -> extraction of slot value
    def _select_best_match(self, val, choices, min_match_percent):
        res = process.extractOne(val, choices)
        if res[1] > min_match_percent:
            return res[0]
        else:
            return None

    def extract_house_room(self, intent_message):
        available_house_rooms = self.snipshue.get_rooms()
        if intent_message.slots.house_room:
            best_match = self._select_best_match(intent_message.slots.house_room.first().value,
                                                 available_house_rooms.keys(), 90)
            if best_match:
                return best_match, available_house_rooms[best_match]
            else:
                return None, None

    def extract_percentage(self, intent_message, default_percentage):
        percentage = default_percentage
        if intent_message.slots.percent:
            percentage = intent_message.slots.percent.first().value
        if percentage < 0:
            percentage = 0
        if percentage > 100:
            percentage = 100
        return percentage

    def extract_scene(self, intent_message, house_room_id):
        available_scenes = self.snipshue.get_scenes_for_room(house_room_id)
        if intent_message.slots.scene:
            best_match = self._select_best_match(intent_message.slots.scene.first().value,
                                                 available_scenes.keys(), 90)
            if best_match:
                return best_match, available_scenes[best_match]
            else:
                return None, None

    # section -> handlers of intents

    def callback(self, hermes, intent_message):
        intent_name = intent_message.intent.intent_name
        if ':' in intent_name:
            intent_name = intent_name.split(":")[1]
        if intent_name not in ALL_INTENTS:
            return
        try:
            print("[HUE] Received {}".format(intent_message.intent.intent_name))
            # all the intents have to have a house_room slot, extract here
            room_name, room_id = self.extract_house_room(intent_message)
            if room_id:
                if intent_name == TURN_ON_INTENT:
                    self.turn_on(hermes, intent_message, room_id)
                if intent_name == TURN_OFF_INTENT:
                    self.turn_off(hermes, intent_message, room_id)
                if intent_name == SET_BRIGHTNESS_INTENT:
                    self.set_brightness(hermes, intent_message, room_id)
                if intent_name == SET_SCENE_INTENT:
                    self.set_scene(hermes, intent_message, room_id)
                if intent_name == SHIFT_UP_INTENT:
                    self.shift_up(hermes, intent_message, room_id)
                if intent_name == SHIFT_DOWN_INTENT:
                    self.shift_down(hermes, intent_message, room_id)
            else:
                self.terminate_feedback(hermes, intent_message, "Sorry no fitting room was detected.")
        except:
            traceback.print_exc()
            self.terminate_feedback(hermes, intent_message, "Unknown exception detected. Please consult log files.")

    def turn_on(self, hermes, intent_message, room_id):
        self.snipshue.light_on(room_id)
        self.terminate_feedback(hermes, intent_message)

    def turn_off(self, hermes, intent_message, room_id):
        self.snipshue.light_off(room_id)
        self.terminate_feedback(hermes, intent_message)

    def set_brightness(self, hermes, intent_message, room_id):
        percent = self.extract_percentage(intent_message, None)
        self.snipshue.light_brightness(percent, room_id)
        self.terminate_feedback(hermes, intent_message)

    def set_scene(self, hermes, intent_message, room_id):
        scene_name, scene_id = self.extract_scene(intent_message, room_id)
        if scene_id:
            self.snipshue.set_scene(scene_id, room_id)
            self.terminate_feedback(hermes, intent_message)
        else:
            self.terminate_feedback(hermes, intent_message, "Sorry, no matching seen found.")

    def shift_up(self, hermes, intent_message, room_id):
        percent = self.extract_percentage(intent_message, 20)
        self.snipshue.shift_brightness(room_id, percent, True)
        self.terminate_feedback(hermes, intent_message)

    def shift_down(self, hermes, intent_message, room_id):
        percent = self.extract_percentage(intent_message, 20)
        self.snipshue.shift_brightness(room_id, percent, False)
        self.terminate_feedback(hermes, intent_message)

    # section -> feedback reply // future function
    def terminate_feedback(self, hermes, intent_message, message=""):
        hermes.publish_end_session(intent_message.session_id, message)
コード例 #2
0
class Skill_Hue:
    def __init__(self):
        try:
            config = SnipsConfigParser.read_configuration_file(CONFIG_INI)
        except:
            config = None
        hostname = None
        code = None
        if config and config.get('secret', None) is not None:
            if config.get('secret').get('hostname', None) is not None:
                hostname = config.get('secret').get('hostname')
                if hostname == "":
                    hostname = None
            if config.get('secret').get(API_KEY, None) is not None:
                code = config.get('secret').get(API_KEY)
                if code == "":
                    code = None
            elif os.path.isfile(CACHE_INI):
                try:
                    cached_config = SnipsConfigParser.read_configuration_file(
                        CACHE_INI)
                except:
                    cached_config = None
                if cached_config and cached_config.get('secret',
                                                       None) is not None:
                    if cached_config.get('secret').get(API_KEY,
                                                       None) is not None:
                        code = cached_config.get('secret').get(API_KEY)
                        if code == "":
                            code = None
        if hostname is None or code is None:
            print('No configuration')
        self.snipshue = SnipsHue(hostname, code)
        hostname = self.snipshue.hostname
        code = self.snipshue.username
        self.update_config(CACHE_INI, config, hostname, code)
        self.queue = Queue.Queue()
        self.thread_handler = ThreadHandler()
        self.thread_handler.run(target=self.start_blocking)
        self.thread_handler.start_run_loop()

    def update_config(self, filename, data, hostname, code):
        if not os.path.exists(CACHE_INI_DIR):
            os.makedirs(CACHE_INI_DIR)
        if 'secret' not in data or data['secret'] is None:
            data['secret'] = {}
        data['secret']['hostname'] = hostname
        data['secret'][API_KEY] = code
        SnipsConfigParser.write_configuration_file(filename, data)

    def start_blocking(self, run_event):
        while run_event.is_set():
            try:
                self.queue.get(False)
            except Queue.Empty:
                with Hermes(MQTT_ADDR) as h:
                    h.subscribe_intents(self.callback).start()

    # section -> extraction of slot value
    def extract_house_rooms(self, intent_message):
        house_rooms = []
        if intent_message.slots.house_room:
            for room in intent_message.slots.house_room.all():
                print type(room.value)
                house_rooms.append(room.value)
        return house_rooms

    def extract_percentage(self, intent_message, default_percentage):
        percentage = default_percentage
        if intent_message.slots.percent:
            percentage = intent_message.slots.percent.first().value
        if percentage < 0:
            percentage = 0
        if percentage > 100:
            percentage = 100
        return percentage

    def extract_color(self, intent_message):
        color_code = None
        if intent_message.slots.color:
            color_code = intent_message.slots.color.first().value
        return color_code

    def extract_scene(self, intent_message):
        scene_code = None
        if intent_message.slots.scene:
            scene_code = intent_message.slots.scene.first().value
        return scene_code

    # section -> handlers of intents

    def callback(self, hermes, intent_message):
        print("[HUE] Received")
        # all the intents have a house_room slot, extract here
        rooms = self.extract_house_rooms(intent_message)
        intent_name = intent_message.intent.intent_name
        if ':' in intent_name:
            intent_name = intent_name.split(":")[1]
        if intent_name == 'turnOn':
            self.queue.put(self.turn_on(hermes, intent_message, rooms))
        if intent_name == 'turnOff':
            self.queue.put(self.turn_off(hermes, intent_message, rooms))
        if intent_name == 'setBrightness':
            self.queue.put(self.set_brightness(hermes, intent_message, rooms))
        if intent_name == 'setColor':
            self.queue.put(self.set_color(hermes, intent_message, rooms))
        if intent_name == 'setScene':
            self.queue.put(self.set_scene(hermes, intent_message, rooms))
        if intent_name == 'shiftUp':
            self.queue.put(self.shift_up(hermes, intent_message, rooms))
        if intent_name == 'shiftDown':
            self.queue.put(self.shift_down(hermes, intent_message, rooms))

    def turn_on(self, hermes, intent_message, rooms):
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_on(room.lower())
        else:
            self.snipshue.light_on_all()
        self.terminate_feedback(hermes, intent_message)

    def turn_off(self, hermes, intent_message, rooms):
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_off(room.lower())
        else:
            self.snipshue.light_off_all()
        self.terminate_feedback(hermes, intent_message)

    def set_brightness(self, hermes, intent_message, rooms):
        percent = self.extract_percentage(intent_message, None)
        if percent is None:
            self.terminate_feedback(hermes, intent_message)
            return
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_brightness(percent, room.lower())
        else:
            self.snipshue.light_brightness_all(percent)
        self.terminate_feedback(hermes, intent_message)

    def set_color(self, hermes, intent_message, rooms):
        color = self.extract_color(intent_message)
        if color is None:
            self.terminate_feedback(hermes, intent_message)
            return
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_color(color, room.lower())
        else:

            self.snipshue.light_color_all(color)
        self.terminate_feedback(hermes, intent_message)

    def set_scene(self, hermes, intent_message, rooms):
        scene = self.extract_scene(intent_message)
        if scene is None:
            self.terminate_feedback(hermes, intent_message)
            return
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_scene(scene, room.lower())
        else:
            self.snipshue.light_scene_all(scene)
        self.terminate_feedback(hermes, intent_message)

    def shift_up(self, hermes, intent_message, rooms):
        percent = self.extract_percentage(intent_message, 20)
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_up(percent, room.lower())
        else:
            self.snipshue.light_up_all(percent)
        self.terminate_feedback(hermes, intent_message)

    def shift_down(self, hermes, intent_message, rooms):
        percent = self.extract_percentage(intent_message, 20)
        if len(rooms) > 0:
            for room in rooms:
                self.snipshue.light_down(percent, room.lower())
        else:
            self.snipshue.light_down_all(percent)
        self.terminate_feedback(hermes, intent_message)

    # section -> feedback reply // future function
    def terminate_feedback(self, hermes, intent_message, mode='default'):
        if mode == 'default':
            hermes.publish_end_session(intent_message.session_id, "")
        else:
            # more design
            hermes.publish_end_session(intent_message.session_id, "")