127 lines
5.1 KiB
Python
127 lines
5.1 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.
|
|
|
|
Parameters:
|
|
* pomodoro.work: The work duration of timer in minutes (defaults to 25)
|
|
* pomodoro.break: The break duration of timer in minutes (defaults to 5)
|
|
* pomodoro.format: Timer display format with '%m' and '%s' for minutes and seconds (defaults to '%m:%s')
|
|
Examples: '%m min %s sec', '%mm', '', 'timer'
|
|
* pomodoro.notify: Notification command to run when timer ends/starts (defaults to nothing)
|
|
Example: 'notify-send 'Time up!''. If you want to chain multiple commands,
|
|
please use an external wrapper script and invoke that. The module itself does
|
|
not support command chaining (see https://github.com/tobi-wan-kenobi/bumblebee-status/issues/532
|
|
for a detailled explanation)
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
import datetime
|
|
from math import ceil
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
|
|
import util.cli
|
|
|
|
|
|
class Module(core.module.Module):
|
|
def __init__(self, config, theme):
|
|
super().__init__(config, theme, core.widget.Widget(self.text))
|
|
|
|
# Parameters
|
|
self.__work_period = int(self.parameter("work", 25))
|
|
self.__break_period = int(self.parameter("break", 5))
|
|
self.__time_format = self.parameter("format", "%m:%s")
|
|
self.__notify_cmd = self.parameter("notify", "")
|
|
|
|
# TODO: Handle time formats more gracefully. This is kludge.
|
|
self.display_seconds_p = False
|
|
self.display_minutes_p = False
|
|
if "%s" in self.__time_format:
|
|
self.display_seconds_p = True
|
|
if "%m" in self.__time_format:
|
|
self.display_minutes_p = True
|
|
|
|
self.remaining_time = datetime.timedelta(minutes=self.__work_period)
|
|
|
|
self.time = None
|
|
self.pomodoro = {"state": "OFF", "type": ""}
|
|
self.__text = self.remaining_time_str() + self.pomodoro["type"]
|
|
|
|
core.input.register(
|
|
self, button=core.input.LEFT_MOUSE, cmd=self.timer_play_pause
|
|
)
|
|
core.input.register(self, button=core.input.RIGHT_MOUSE, cmd=self.timer_reset)
|
|
|
|
def remaining_time_str(self):
|
|
if self.display_seconds_p and self.display_minutes_p:
|
|
minutes, seconds = divmod(self.remaining_time.seconds, 60)
|
|
if not self.display_seconds_p:
|
|
minutes = ceil(self.remaining_time.seconds / 60)
|
|
seconds = 0
|
|
if not self.display_minutes_p:
|
|
minutes = 0
|
|
seconds = self.remaining_time.seconds
|
|
|
|
minutes = "{:2d}".format(minutes)
|
|
seconds = "{:02d}".format(seconds)
|
|
return self.__time_format.replace("%m", minutes).replace("%s", seconds) + " "
|
|
|
|
def text(self, widget):
|
|
return "{}".format(self.__text)
|
|
|
|
def update(self):
|
|
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.total_seconds() <= 0:
|
|
self.notify()
|
|
if self.pomodoro["type"] == "Work":
|
|
self.pomodoro["type"] = "Break"
|
|
self.remaining_time = datetime.timedelta(
|
|
minutes=self.__break_period
|
|
)
|
|
elif self.pomodoro["type"] == "Break":
|
|
self.pomodoro["type"] = "Work"
|
|
self.remaining_time = datetime.timedelta(minutes=self.__work_period)
|
|
|
|
self.__text = self.remaining_time_str() + self.pomodoro["type"]
|
|
|
|
def notify(self):
|
|
if self.__notify_cmd:
|
|
util.cli.execute(self.__notify_cmd)
|
|
|
|
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
|
|
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|