refactor: nixify / clean remaining dots scripts

home/modules/bin/scripts/pomo

@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+# vim: set filetype=python:
+"""
+Pomodoro timer
+
+- Writes pomodoro timer to temporary file so statusbar can read it
+- Notification on session finish
+- Notification on break finish
+"""
+
+import atexit
+import os
+import signal
+import subprocess
+from argparse import ArgumentParser
+from time import sleep
+
+POMO_PATH = os.path.join(
+    os.getenv("XDG_DATA_HOME", os.path.expanduser("~/.local/share")), "pomo"
+)
+
+
+@atexit.register
+def clear():
+    if os.path.exists(POMO_PATH):
+        os.remove(POMO_PATH)
+
+
+def format_mins_secs(mins, secs):
+    return f"{mins:02d}:{secs:02d}"
+
+
+def countdown(duration):
+    while duration != 0:
+        mins = duration // 60
+        secs = duration % 60
+        with open(POMO_PATH, "w") as f:
+            f.write(format_mins_secs(mins, secs))
+        sleep(1)
+        duration -= 1
+
+
+def main(args):
+    work_duration = args.work_duration * 60
+    break_duration = args.break_duration * 60
+    repeats = args.repeats
+
+    while repeats != 0:
+        subprocess.run(["notify-send", "Get started"])
+        countdown(work_duration)
+        if break_duration != 0:
+            subprocess.run(["notify-send", "Time for a break"])
+            countdown(break_duration)
+            subprocess.run(["notify-send", "Break is over, back to work"])
+        repeats -= 1
+
+
+def handle_signal(signal, frame):
+    # Wait for clear to finish
+    clear()
+    print("Exiting")
+    exit(0)
+
+
+if __name__ == "__main__":
+
+    parser = ArgumentParser()
+    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
+    )
+    parser.add_argument("-c", "--clear", action="store_true", help="Clear timer")
+
+    args = parser.parse_args()
+
+    signal.signal(signal.SIGINT, handle_signal)
+    signal.signal(signal.SIGTERM, handle_signal)
+
+    if args.clear:
+        clear()
+    else:
+        main(args)