示例#1
0
 def onJoin(self, details):
     self.hal = HAL(HALFS_ROOT)
     yield from self.register(self.tree, u'tree')
     yield from self.register_switchs()
     yield from self.register_triggers()
     yield from self.register_animations()
     yield from self.register_rgbs()
     self.hal.install_loop()
     yield from self.periodic_tasks()
示例#2
0
class Component(ApplicationSession):
    """
    An application component that publishes an event every second.
    """

    @asyncio.coroutine
    def tree(self):
        def authorized(resources, ignore_key):
            ignore = HAL_IGNORE.get(ignore_key, set())
            return list(filter(lambda x: x not in ignore, resources.keys()))

        h = self.hal
        return {
            resource: authorized(getattr(h, resource, []), resource)
            for resource in h.resource_mapping.keys()
        }

    @asyncio.coroutine
    def periodic_tasks(self):
        while True:
            for sensor in self.hal.sensors.values():
                self.publish('sensor.'+sensor.name, sensor.value)
            yield from asyncio.sleep(2.5)

    @asyncio.coroutine
    def register_triggers(self):
        @self.hal.on_trigger()
        def publish_trigger(name, state):
            key = 'trigger.%s' % name
            self.publish(key, state)

        for trig in self.hal.triggers.values():
            key = 'trigger.%s.state' % trig.name

            def current_state_trigger(trigger=trig):
                return trigger.on
            yield from self.register(current_state_trigger, key)

    @asyncio.coroutine
    def register_rgbs(self):
        for rgb in self.hal.rgbs.values():
            key = 'rgb.%s' % rgb.name

            def current_state_rgb(rgb_led=rgb):
                return rgb_led.css
            yield from self.register(current_state_rgb, key + '.state')

            def set_rgb(color, rgb_led=rgb):
                rgb_led.css = color
            yield from self.register(set_rgb, key + '.set')

            @rgb.on_change
            def publish_rgb(rgb_led):
                self.publish('rgb.%s' % rgb_led.name, rgb_led.css)

    @asyncio.coroutine
    def register_switchs(self):
        for sw in self.hal.switchs.values():
            key = 'switch.%s' % sw.name

            def current_state_switch(switch=sw):
                return switch.on
            yield from self.register(current_state_switch, key + '.state')

            def toggle_switch(switch=sw):
                switch.on = not switch.on
            yield from self.register(toggle_switch, key + '.toggle')

            @sw.on_change
            def publish_switch(switch):
                self.publish('switch.%s' % switch.name, switch.on)

    @asyncio.coroutine
    def register_animations(self):
        for anim in self.hal.animations.values():
            key = 'animation.%s' % anim.name

            def current_state_playing(animation=anim):
                return animation.playing
            yield from self.register(current_state_playing, key + '.play.state')

            def toggle_playing(animation=anim):
                animation.playing = not animation.playing
            yield from self.register(toggle_playing, key + '.play.toggle')

            def current_state_looping(animation=anim):
                return animation.looping
            yield from self.register(current_state_looping, key + '.loop.state')

            def toggle_looping(animation=anim):
                animation.looping = not animation.looping
            yield from self.register(toggle_looping, key + '.loop.toggle')

            def current_fps(animation=anim):
                return animation.fps
            yield from self.register(current_fps, key + '.fps.state')

            def set_fps(fps, animation=anim):
                animation.fps = int(fps)
            yield from self.register(set_fps, key + '.fps.set')

            def current_frames(animation=anim):
                return animation.frames
            yield from self.register(current_frames, key+'.frames.state')

            def set_frames(frames, animation=anim):
                clean = lambda x: 0 if x is None else max(0, min(255, int(x)))
                animation.frames = map(clean, frames)
            yield from self.register(set_frames, key+'.frames.set')

            @anim.on_change
            def publish_anim(animation):
                key = "animation.%s" % animation.name
                self.publish(key+'.loop', animation.looping)
                self.publish(key+'.play', animation.playing)
                self.publish(key+'.fps', animation.fps)
                self.publish(key+'.frames', animation.frames)

    @asyncio.coroutine
    def onJoin(self, details):
        self.hal = HAL(HALFS_ROOT)
        yield from self.register(self.tree, u'tree')
        yield from self.register_switchs()
        yield from self.register_triggers()
        yield from self.register_animations()
        yield from self.register_rgbs()
        self.hal.install_loop()
        yield from self.periodic_tasks()
示例#3
0
文件: test_hal.py 项目: UrLab/halpy
def test_rel_path():
    hal = HAL(ROOT)
    r = hal.map_path('animations/lolilol')
    assert isinstance(r, Animation)
    assert r.name == 'lolilol'
示例#4
0
文件: test_hal.py 项目: UrLab/halpy
def test_map_full_path():
    hal = HAL(ROOT)
    r = hal.map_path(path.join(ROOT, 'animations/lolilol'))
    assert isinstance(r, Animation)
    assert r.name == 'lolilol'