Making status manageable via contextual classes.

specialized/plugin_status_led.py

 from plugins.base import Process, PluginProcessBase
+from plugins.processes_host import find_plugin
 from plugins.decorators import make_plugin
 from collections import namedtuple
 from math import isinf, isnan
@@ -91,6 +92,26 @@ class BlinkingStatus(namedtuple('BlinkingStatusBase',
             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)
 class StatusLEDPlugin(PluginProcessBase):
     def __init__(self):
@@ -219,6 +240,43 @@ class StatusLEDPlugin(PluginProcessBase):
             raise TypeError('on_color')
         if not isinstance(off_color, (tuple, list)) or len(off_color) != 3:
             raise TypeError('off_color')
-        return self._push(BlinkingStatus(on_color=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))
+        return ContextualStatus(self._push(BlinkingStatus(
+            on_color=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
+        )))
+
+
+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)