bumblebee-status/bumblebee/modules/pomodoro.py

# 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
from math import ceil

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._work_period = 25
        self._play_period = 5
        self.remaining_time = datetime.timedelta(minutes=self._work_period)
        self.remaining_time_str = "{}m ".format(ceil((self.remaining_time.seconds / 60)))
        self.time = None
        self.pomodoro = { "state":"OFF", "type": ""}
        self._text = self.remaining_time_str + self.pomodoro["type"]
        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"] = "Break"
                    self.remaining_time = datetime.timedelta(minutes=self._work_period)
                elif self.pomodoro["type"] == "Break":
                    self.pomodoro["type"] = "Work"
                    self.remaining_time = datetime.timedelta(minutes=self._play_period)

        self.remaining_time_str = "{}m ".format(ceil((self.remaining_time.seconds / 60)))
        self._text = self.remaining_time_str + self.pomodoro["type"]
    
    def timer_play_pause(self, widget):
        if self.pomodoro["state"] == "OFF":
            self.pomodoro = {"state": "ON", "type": "Work"}
            self.remaining_time = datetime.timedelta(minutes=self._work_period)
            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": "" }
            self.remaining_time = datetime.timedelta(minutes=self._work_period)

    def state(self, widget):
        state = [];
        state.append(self.pomodoro["state"].lower())
        if self.pomodoro["state"] == "ON" or self.pomodoro["state"] == "OFF":
            state.append(self.pomodoro["type"].lower())

        return state