86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
import csv
|
|
import time
|
|
from datetime import datetime, timedelta
|
|
from pathlib import Path
|
|
|
|
import pytz
|
|
import zenipy
|
|
|
|
WORK_MIN = .1
|
|
SHORT_MIN = .1
|
|
LONG_MIN = 15
|
|
|
|
|
|
def pomodoro_prompt_plan(whatdid = ''):
|
|
whatnext = zenipy.zenipy.entry(text="What're you gonna do?", placeholder=whatdid, title='Pomodoro Prompt: Plan')
|
|
# Pressing cancel returns None, but we want to just treat it as an empty string.
|
|
if whatnext is None:
|
|
whatnext = ''
|
|
return whatnext
|
|
|
|
def pomodoro_prompt_report(whatnext):
|
|
whatdid = zenipy.zenipy.entry(text="What'd you do?", placeholder=whatnext, title='Pomodoro Prompt: Report')
|
|
# Pressing cancel returns None, but we want to just treat it as an empty string.
|
|
if whatdid is None:
|
|
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, start):
|
|
timelog_path = 'log/' + str(start.year) + '-' + format(start.month, '02') + '-' + format(start.day, '02') + '.log'
|
|
timelog_file = Path(timelog_path)
|
|
timelog_file.touch(exist_ok=True)
|
|
with timelog_file.open('a') as timelog:
|
|
|
|
return
|
|
|
|
def record_task(whatdid, start, end=None):
|
|
if end is None:
|
|
end = datetime.now(pytz.utc)
|
|
with open('timelog.csv', 'a', newline='') as csvfile:
|
|
timewriter = csv.writer(csvfile)
|
|
timewriter.writerow([start, end, whatdid])
|
|
|
|
|
|
def main():
|
|
whatdid = ''
|
|
while True:
|
|
whatnext = pomodoro_prompt_plan(whatdid)
|
|
|
|
start = datetime.now(pytz.utc)
|
|
end = start + timedelta(minutes=WORK_MIN, seconds=0)
|
|
try:
|
|
print('Task -', whatnext)
|
|
while datetime.now(pytz.utc) <= end:
|
|
to_go = end-datetime.now(pytz.utc)
|
|
print('\r', str(to_go).split('.')[0], sep='', end='')
|
|
time.sleep(1)
|
|
whatdid = pomodoro_prompt_report(whatnext)
|
|
record_task(whatdid, start)
|
|
continue
|
|
except KeyboardInterrupt:
|
|
time_spent = datetime.now(pytz.utc) - start
|
|
print('\n{} time spent, save?'.format(str(time_spent).split('.')[0]))
|
|
choice = input('[y]/n: ').strip()
|
|
if choice.lower() in ('y', 'yes', ''):
|
|
whatdid = pomodoro_prompt_report(whatnext)
|
|
record_task(whatdid, start)
|
|
else:
|
|
quit_prompt()
|
|
else:
|
|
print('What did you break?')
|
|
# If we somehow end up here, try a last-ditch effort to save.
|
|
record_task('Incomplete, interrupted task:' + whatnext, start)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|