Commit 43cc5536 authored by Pietro Saccardi's avatar Pietro Saccardi
Browse files

Making status manageable via contextual classes.

parent 668c0c0a

specialized/plugin_status_led.py

+61 −3
Original line number Original line Diff line number Diff line 
from plugins.base import Process, PluginProcessBase

from plugins.base import Process, PluginProcessBase

from plugins.processes_host import find_plugin

from plugins.decorators import make_plugin

from plugins.decorators import make_plugin

from collections import namedtuple

from collections import namedtuple

from math import isinf, isnan

from math import isinf, isnan
@@ -91,6 +92,26 @@ class BlinkingStatus(namedtuple('BlinkingStatusBase', 
            already_faded_in = False

            already_faded_in = False









class ContextualStatus:

    def __init__(self, blinking_status):

        self._blinking_status = blinking_status

        self._entered = False



    def cancel(self):

        if self._entered:

            self._entered = False

            plugin = find_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

            if plugin:

                plugin.cancel(self._blinking_status)



    def __enter__(self):

        self._entered = True

        return self



    def __exit__(self, exc_type, exc_val, exc_tb):

        self.cancel()





@make_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

@make_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

class StatusLEDPlugin(PluginProcessBase):

class StatusLEDPlugin(PluginProcessBase):

    def __init__(self):

    def __init__(self):
@@ -219,6 +240,43 @@ class StatusLEDPlugin(PluginProcessBase): 
            raise TypeError('on_color')

            raise TypeError('on_color')

        if not isinstance(off_color, (tuple, list)) or len(off_color) != 3:

        if not isinstance(off_color, (tuple, list)) or len(off_color) != 3:

            raise TypeError('off_color')

            raise TypeError('off_color')

        return self._push(BlinkingStatus(on_color=on_color, off_color=off_color, fade_in_time=fade_in_time,

        return ContextualStatus(self._push(BlinkingStatus(

                                         fade_out_time=fade_out_time, persist_on_time=persist_on_time,

            on_color=on_color, off_color=off_color, fade_in_time=fade_in_time, fade_out_time=fade_out_time,

                                         persist_off_time=persist_off_time, n=n))
 
            persist_on_time=persist_on_time, persist_off_time=persist_off_time, n=n

        )))





class Status:

    @staticmethod

    def set(color, fade_in_time=0.5, persist_until_canceled=False):

        plugin = find_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

        if plugin:

            return plugin.set(color, fade_in_time=fade_in_time, persist_until_canceled=persist_until_canceled)

        else:

            return ContextualStatus(None)



    @staticmethod

    def blink(color, n=float('inf'), duty_cycle=0.5, frequency=1.):

        plugin = find_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

        if plugin:

            return plugin.blink(color, n=n, duty_cycle=duty_cycle, frequency=frequency)

        else:

            return ContextualStatus(None)



    @staticmethod

    def pulse(color, n=float('inf'), persist_time=0., frequency=1.):

        plugin = find_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

        if plugin:

            return plugin.pulse(color, n=n, persist_time=persist_time, frequency=frequency)

        else:

            return ContextualStatus(None)



    @staticmethod

    def custom(on_color, off_color=(0., 0., 0.), fade_in_time=0., fade_out_time=0.5, persist_on_time=0.,

               persist_off_time=0., n=float('inf')):

        plugin = find_plugin(STATUS_LED_PLUGIN_NAME, Process.MAIN)

        if plugin:

            return plugin.blink(on_color, off_color=off_color, fade_in_time=fade_in_time, fade_out_time=fade_out_time,

                                persist_on_time=persist_on_time, persist_off_time=persist_off_time, n=n)

        else:

            return ContextualStatus(None)