.bin/pomo

@@ -1,12 +1,11 @@
-#!/usr/bin/env python3
-# vim: set filetype=python:
-"""
-Pomodoro timer
+#!/usr/bin/env python
 
-- Writes pomodoro timer to temporary file so statusbar can read it
-- Notification on session finish
-- Notification on break finish
-"""
+# Pomodoro timer
+#
+# - 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,37 +18,47 @@ def clear():
     os.system('echo -n "" > /tmp/pomo')
 
 
-def format_mins_secs(mins, secs):
-    return f"{mins:02d}:{secs:02d}"
+atexit.register(clear)
+
+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(f'echo -n "{time_str}" > /tmp/pomo')
+            # os.system('echo -n "{:s} {:02d}:{:02d} \x01" > /tmp/pomo'.format(color, mins, secs))
+            os.system(
+                'echo -n "<fc={:s}> {:02d}:{:02d} </fc>" > /tmp/pomo'.format(color, mins, secs))
             sleep(1)
             duration -= 1
     return countdown
 
 
-def main(args):
-    atexit.register(clear)
+prep_countdown = make_countdown("#0000aa")
+work_countdown = make_countdown("#aa0000")
+break_countdown = make_countdown("#00bb00")
 
-    prep_duration = args.prep_duration * 60
-    work_duration = args.work_duration * 60
-    break_duration = args.break_duration * 60
-    repeats = args.repeats
+prep_countdown(prep_duration)
 
-    prep_countdown = make_countdown()
-    work_countdown = make_countdown()
-    break_countdown = make_countdown()
-
-    prep_countdown(prep_duration)
-
-    while repeats != 0:
+while repeats != 0:
     notification.notify(title="Get started")
     work_countdown(work_duration)
     if break_duration != 0:
@@ -57,20 +66,3 @@ def main(args):
         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)