.bin/pomo

@@ -1,12 +1,11 @@
-#!/usr/bin/env python3
+#!/usr/bin/env python
-# vim: set filetype=python:
-"""
-Pomodoro timer
 
-- Writes pomodoro timer to temporary file so statusbar can read it
+# Pomodoro timer
-- Notification on session finish
+#
-- Notification on break finish
+# - Writes pomodoro timer to temporary file so dwmblocks
-"""
+#   statusbar can read it
+# - Notification on session finish
+# - Notification on break finish
 
 import os
 import atexit
@@ -19,58 +18,51 @@ def clear():
     os.system('echo -n "" > /tmp/pomo')
 
 
-def format_mins_secs(mins, secs):
+atexit.register(clear)
-    return f"{mins:02d}:{secs:02d}"
+
+parser = ArgumentParser()
+
+parser.add_argument('-p', '--prep-duration', type=int,
+                    help='Pre session duration', default=0)
+parser.add_argument('-w', '--work-duration', type=int,
+                    help='Session duration', default=25)
+parser.add_argument('-b', '--break-duration', type=int,
+                    help='Break duration', default=5)
+parser.add_argument('-r', '--repeats', type=int,
+                    help='Numer of sessions', default=1)
+
+args = parser.parse_args()
+
+prep_duration = args.prep_duration * 60
+work_duration = args.work_duration * 60
+break_duration = args.break_duration * 60
+repeats = args.repeats
 
 
-def make_countdown():
+def make_countdown(color="#000000"):
     def countdown(duration):
         while duration != 0:
             mins = duration // 60
             secs = duration % 60
-            time_str = format_mins_secs(mins, secs)
+            # os.system('echo -n "{:s} {:02d}:{:02d} \x01" > /tmp/pomo'.format(color, mins, secs))
-            os.system(f'echo -n "{time_str}" > /tmp/pomo')
+            os.system(
+                'echo -n "<fc={:s}> {:02d}:{:02d} </fc>" > /tmp/pomo'.format(color, mins, secs))
             sleep(1)
             duration -= 1
     return countdown
 
 
-def main(args):
+prep_countdown = make_countdown("#0000aa")
-    atexit.register(clear)
+work_countdown = make_countdown("#aa0000")
+break_countdown = make_countdown("#00bb00")
 
-    prep_duration = args.prep_duration * 60
+prep_countdown(prep_duration)
-    work_duration = args.work_duration * 60
-    break_duration = args.break_duration * 60
-    repeats = args.repeats
 
-    prep_countdown = make_countdown()
+while repeats != 0:
-    work_countdown = make_countdown()
+    notification.notify(title="Get started")
-    break_countdown = make_countdown()
+    work_countdown(work_duration)
-
+    if break_duration != 0:
-    prep_countdown(prep_duration)
+        notification.notify(title="Time for a break")
-
+        break_countdown(break_duration)
-    while repeats != 0:
+        notification.notify(title="Break is over, back to work")
-        notification.notify(title="Get started")
+    repeats -= 1
-        work_countdown(work_duration)
-        if break_duration != 0:
-            notification.notify(title="Time for a break")
-            break_countdown(break_duration)
-            notification.notify(title="Break is over, back to work")
-        repeats -= 1
-
-
-if __name__ == '__main__':
-    parser = ArgumentParser()
-
-    parser.add_argument('-p', '--prep-duration', type=int,
-                        help='Pre session duration', default=0)
-    parser.add_argument('-w', '--work-duration', type=int,
-                        help='Session duration', default=25)
-    parser.add_argument('-b', '--break-duration', type=int,
-                        help='Break duration', default=5)
-    parser.add_argument('-r', '--repeats', type=int,
-                        help='Numer of sessions', default=1)
-
-    args = parser.parse_args()
-
-    main(args)