masq/i3pystatus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor

Browse Source 
Use integer values for states instead of strings. Feels more optimal.
This commit is contained in:
Sergei Turukin 2016-05-17 14:01:37 +03:00
parent 575f682a61
commit 7ec7739055
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
i3pystatus/pomodoro.py
View File

@ -6,6 +6,10 @@ from datetime import datetime, timedelta
from i3pystatus import IntervalModule from i3pystatus import IntervalModule
STOPPED = 0
RUNNING = 1
BREAK = 2
class Pomodoro(IntervalModule): class Pomodoro(IntervalModule):
""" """
@ -44,14 +48,14 @@ class Pomodoro(IntervalModule):
def init(self): def init(self):
# state could be either running/break or stopped # state could be either running/break or stopped
self.state = 'stopped' self.state = STOPPED
self.breaks = 0 self.breaks = 0
self.time = None self.time = None
def run(self): def run(self):
if self.time and datetime.utcnow() >= self.time: if self.time and datetime.utcnow() >= self.time:
if self.state == 'running': if self.state == RUNNING:
self.state = 'break' self.state = BREAK
if self.breaks == self.short_break_count: if self.breaks == self.short_break_count:
self.time = datetime.utcnow() + \ self.time = datetime.utcnow() + \
timedelta(seconds=self.long_break_duration) timedelta(seconds=self.long_break_duration)
@ -62,19 +66,19 @@ class Pomodoro(IntervalModule):
self.breaks += 1 self.breaks += 1
text = 'Go for a break!' text = 'Go for a break!'
else: else:
self.state = 'running' self.state = RUNNING
self.time = datetime.utcnow() + \ self.time = datetime.utcnow() + \
timedelta(seconds=self.pomodoro_duration) timedelta(seconds=self.pomodoro_duration)
text = 'Back to work!' text = 'Back to work!'
self._alarm(text) self._alarm(text)
if self.state == 'running' or self.state == 'break': if self.state == RUNNING or self.state == BREAK:
min, sec = divmod((self.time - datetime.utcnow()).total_seconds(), 60) min, sec = divmod((self.time - datetime.utcnow()).total_seconds(), 60)
text = '{:02}:{:02}'.format(int(min), int(sec)) text = '{:02}:{:02}'.format(int(min), int(sec))
color = self.color_running if self.state == 'running' else self.color_break color = self.color_running if self.state == RUNNING else self.color_break
else: else:
self.output = { self.output = {
'full_text': 'Stopped', 'full_text': 'Start pomodoro',
'color': self.color_stopped 'color': self.color_stopped
} }
return return
@ -91,12 +95,12 @@ class Pomodoro(IntervalModule):
} }
def start(self): def start(self):
self.state = 'running' self.state = RUNNING
self.time = datetime.utcnow() + timedelta(seconds=self.pomodoro_duration) self.time = datetime.utcnow() + timedelta(seconds=self.pomodoro_duration)
self.breaks = 0 self.breaks = 0
def stop(self): def stop(self):
self.state = 'stopped' self.state = STOPPED
self.time = None self.time = None
def _alarm(self, text): def _alarm(self, text):