[module] New module pomodoro timer

bumblebee/modules/pomodoro.py

@@ -0,0 +1,67 @@
+# pylint: disable=C0111,R0903
+
+"""Display and run a Pomodoro timer.
+Left click to start timer, left click again to pause.
+Right click will cancel the timer.
+"""
+
+import datetime
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        widgets = bumblebee.output.Widget(full_text=self.text)
+        self.remaining_time = datetime.timedelta(minutes=25)
+        self.remaining_time_str = "{}min{}s ".format(int((self.remaining_time.seconds / 60)),
+                                                     round((self.remaining_time.seconds/60) % 1*60))
+        self.time = None
+        self.pomodoro = { "state":"OFF", "type": "n/a"}
+        self._text = self.remaining_time_str + self.pomodoro["type"] + " " + self.pomodoro["state"]
+        super(Module, self).__init__(engine, config, widgets)
+        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
+                                       cmd=self.timer_play_pause)
+        engine.input.register_callback(self, button=bumblebee.input.RIGHT_MOUSE,
+                                       cmd=self.timer_reset)
+        
+    def text(self, widget):
+        return "{}".format(self._text) 
+        
+    def update(self, widget):
+        if self.pomodoro["state"] == "ON":
+            timediff = (datetime.datetime.now() - self.time)
+            if timediff.seconds >= 0:
+                self.remaining_time -= timediff
+                self.time = datetime.datetime.now()
+
+            if self.remaining_time.seconds <= 0:
+                if self.pomodoro["type"] == "WORK":
+                    self.pomodoro["type"] = "PLAY"
+                    self.remaining_time = datetime.timedelta(minutes=25)
+                elif self.pomodoro["type"] == "PLAY":
+                    self.pomodoro["type"] = "WORK"
+                    self.remaining_time = datetime.timedelta(minutes=5)
+
+        self.remaining_time_str = "{}min{}s ".format(int((self.remaining_time.seconds / 60)),
+                                                    round((self.remaining_time.seconds / 60) % 1 * 60))
+        self._text = self.remaining_time_str + self.pomodoro["type"] + " " + self.pomodoro["state"]
+    
+    def timer_play_pause(self, widget):
+        if self.pomodoro["state"] == "OFF":
+            self.pomodoro = {"state": "ON", "type": "WORK"}
+            self.remaining_time = datetime.timedelta(minutes=25)
+            self.time = datetime.datetime.now()
+        elif self.pomodoro["state"] == "ON":
+            self.pomodoro["state"] = "PAUSED"
+            self.remaining_time -= (datetime.datetime.now() - self.time)
+            self.time = datetime.datetime.now()
+        elif self.pomodoro["state"] == "PAUSED":
+            self.pomodoro["state"] = "ON"
+            self.time = datetime.datetime.now()
+
+    def timer_reset(self, widget):
+        if self.pomodoro["state"] == "ON" or self.pomodoro["state"] == "PAUSED":
+            self.pomodoro = {"state":"OFF", "type": "n/a" }
+            self.remaining_time = datetime.timedelta(minutes=25)