diff --git a/pomodoroprompt.py b/pomodoroprompt.py index e08eda0..ccb8b76 100644 --- a/pomodoroprompt.py +++ b/pomodoroprompt.py @@ -87,6 +87,11 @@ def main(): whatdid = pomodoro_prompt_report(whatnext) record_task(whatdid, start) log_step('Completed pomodoro: ' + whatdid, start, True) + print('Worked on: ' + whatdid) + + print('Break time!') + # We're taking a break. Clear out task start/end times. + start = None end = datetime.now(pytz.utc) + timedelta(minutes=SHORT_MIN) while datetime.now(pytz.utc) <= end: to_go = end-datetime.now(pytz.utc) @@ -94,15 +99,20 @@ def main(): time.sleep(1) continue except KeyboardInterrupt: + if start is None: + quit_prompt() + continue time_spent = datetime.now(pytz.utc) - start - if time_spent > timedelta(minutes=INTERRUPTED_MIN): - time_spent_str = str_minutes(time_spent) - print('\n{} time spent, save?'.format(time_spent_str)) - choice = input('[y]/n: ').strip() - if choice.lower() in ('y', 'yes', ''): - whatdid = pomodoro_prompt_report(whatnext) - record_task(whatdid, start) - log_step('Incomplete (' + time_spent_str + ') pomodoro: ' + whatdid, start, True) + if time_spent < timedelta(minutes=INTERRUPTED_MIN): + quit_prompt() + continue + time_spent_str = str_minutes(time_spent) + print('\n{} time spent, save?'.format(time_spent_str)) + choice = input('[y]/n: ').strip() + if choice.lower() in ('y', 'yes', ''): + whatdid = pomodoro_prompt_report(whatnext) + record_task(whatdid, start) + log_step('Incomplete (' + time_spent_str + ') pomodoro: ' + whatdid, start, True) quit_prompt() else: print('What did you break?')