# 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. """ from __future__ import absolute_import 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)