77 lines
3.3 KiB
Python
77 lines
3.3 KiB
Python
# 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
|