Add the ability to quit gracefgully if no longer tracking time

This commit is contained in:
benjamin melançon 2020-07-05 21:45:53 -04:00
parent 501784c562
commit 57ec262f7d

View file

@ -24,6 +24,22 @@ def pomodoro_prompt_report(whatnext):
whatdid = '' whatdid = ''
return whatdid return whatdid
def quit_prompt():
print('\nPress p to start a new pomodoro, q to quit')
choice = input('p/q: ').strip()
if choice.lower() in ('p', 'pomodoro'):
return
elif choice.lower() in ('q', 'quit', 'exit'):
quit()
else:
quit_prompt()
def log_step(text):
# timelog.touch(exist_ok=True)
# with timelog.open('r+') as f:
# pp = PomodoroPrompt(timelog=f)
return
def record_task(whatdid, start, end=None): def record_task(whatdid, start, end=None):
if end is None: if end is None:
end = datetime.now(pytz.utc) end = datetime.now(pytz.utc)
@ -55,15 +71,13 @@ def main():
if choice.lower() in ('y', 'yes', ''): if choice.lower() in ('y', 'yes', ''):
whatdid = pomodoro_prompt_report(whatnext) whatdid = pomodoro_prompt_report(whatnext)
record_task(whatdid, start) record_task(whatdid, start)
else:
quit_prompt()
else: else:
print('What did you break?') print('What did you break?')
# If we somehow end up here, try a last-ditch effort to save. # If we somehow end up here, try a last-ditch effort to save.
record_task('Incomplete, interrupted task:' + whatnext, start) record_task('Incomplete, interrupted task:' + whatnext, start)
# timelog.touch(exist_ok=True)
# with timelog.open('r+') as f:
# pp = PomodoroPrompt(timelog=f)
if __name__ == '__main__': if __name__ == '__main__':
main() main()