Improve pomo script

main
Hektor Misplon 2025-10-26 22:46:15 +01:00
parent 84e0c7f700
commit 5a50aa540f
1 changed files with 4 additions and 7 deletions

View File

@ -14,11 +14,12 @@ from argparse import ArgumentParser
from time import sleep from time import sleep
from plyer import notification from plyer import notification
POMO_PATH = os.path.join(os.getenv("XDG_DATA_HOME", os.path.expanduser("~/.local/share")), "pomo")
@atexit.register @atexit.register
def clear(): def clear():
if os.path.exists('/home/h/.local/share/pomo'): if os.path.exists(POMO_PATH):
os.remove('/home/h/.local/share/pomo') os.remove(POMO_PATH)
def format_mins_secs(mins, secs): def format_mins_secs(mins, secs):
return f"{mins:02d}:{secs:02d}" return f"{mins:02d}:{secs:02d}"
@ -30,24 +31,20 @@ def make_countdown():
mins = duration // 60 mins = duration // 60
secs = duration % 60 secs = duration % 60
time_str = format_mins_secs(mins, secs) time_str = format_mins_secs(mins, secs)
os.system(f'echo -n "{time_str}" > /home/h/.local/share/pomo') os.system(f'echo -n "{time_str}" > {POMO_PATH}')
sleep(1) sleep(1)
duration -= 1 duration -= 1
return countdown return countdown
def main(args): def main(args):
prep_duration = args.prep_duration * 60
work_duration = args.work_duration * 60 work_duration = args.work_duration * 60
break_duration = args.break_duration * 60 break_duration = args.break_duration * 60
repeats = args.repeats repeats = args.repeats
prep_countdown = make_countdown()
work_countdown = make_countdown() work_countdown = make_countdown()
break_countdown = make_countdown() break_countdown = make_countdown()
prep_countdown(prep_duration)
while repeats != 0: while repeats != 0:
notification.notify(title="Get started") notification.notify(title="Get started")
work_countdown(work_duration) work_countdown(work_duration)