134 lines
3.3 KiB
Python
Executable File
134 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# From: https://gist.github.com/varunagrawal/2b93c5dc721520ff6876e940c420ab05
|
|
# This hooks script logs any finished tasks currently in the todo.txt file
|
|
# pointed to by its settings.
|
|
# The on-exit event is triggered once, after all processing is complete.
|
|
|
|
import re
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import argparse
|
|
|
|
TODOTXT_FILE = "/home/marty/documents/records/todo.md"
|
|
TODO_BLOCK_TITLE = "todotxt"
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"-n", "--dryrun", help="only simulate and print changes", action="store_true"
|
|
)
|
|
parser.add_argument(
|
|
"-L",
|
|
"--nologging",
|
|
help="don't log done todo items to taskwarrior",
|
|
action="store_true",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
# set options
|
|
def is_dryrun():
|
|
if args.dryrun:
|
|
return True
|
|
return False
|
|
|
|
|
|
def is_logging_done_to_tw():
|
|
if args.nologging:
|
|
return False
|
|
return True
|
|
|
|
|
|
def get_todo_block_date(line):
|
|
result = re.search(
|
|
rf"^\[(\d{{4}}-\d{{2}}-\d{{2}}).*\] ({re.escape(TODO_BLOCK_TITLE)}$)?",
|
|
line,
|
|
)
|
|
# we are entering todotxt data block
|
|
if result and result[2]:
|
|
return result[1]
|
|
# we are in block but not one for todotxt
|
|
elif result and not result[2]:
|
|
return False
|
|
# nothing changed, we didn't cross a date line
|
|
return "same"
|
|
|
|
|
|
def parse_file(fname):
|
|
curdate = "1970-01-01"
|
|
lines_to_delete = []
|
|
with open(fname) as file:
|
|
line_number = 0
|
|
for line in file:
|
|
line_number += 1
|
|
|
|
todo_block = get_todo_block_date(line)
|
|
if todo_block == False:
|
|
continue
|
|
if todo_block != "same":
|
|
curdate = todo_block
|
|
continue
|
|
|
|
completed_task = re.search(r"^x ((?:\([A-D]\))? ?.*)$", line)
|
|
if completed_task and is_logging_done_to_tw():
|
|
log_done_to_taskwarrior(completed_task[1], curdate)
|
|
lines_to_delete.append(line_number)
|
|
|
|
if lines_to_delete:
|
|
delete_logged_tasks_from_file(TODOTXT_FILE, lines_to_delete)
|
|
|
|
|
|
def get_prio_string(task_string):
|
|
prio = re.search(r"^\(([ABC])\) (.*)$", task_string)
|
|
prio_string = ""
|
|
if prio:
|
|
task_string = prio[2]
|
|
if prio[1] == "A":
|
|
prio_string = "prio:H"
|
|
elif prio[1] == "B":
|
|
prio_string = "prio:M"
|
|
elif prio[1] == "C":
|
|
prio_string = "prio:L"
|
|
return prio_string
|
|
|
|
|
|
def log_done_to_taskwarrior(task_string, date):
|
|
overrides = "rc.context=none rc.hooks=0"
|
|
|
|
cmd = [
|
|
"task",
|
|
overrides,
|
|
"log",
|
|
f'"{task_string}"',
|
|
f"entry:{date}",
|
|
f"end:{date}",
|
|
get_prio_string(task_string),
|
|
]
|
|
|
|
if is_dryrun():
|
|
print(cmd)
|
|
return
|
|
|
|
subprocess.Popen(cmd)
|
|
|
|
|
|
def delete_logged_tasks_from_file(fname, lines):
|
|
if is_dryrun():
|
|
print(f"deleting lines: {lines}")
|
|
return
|
|
|
|
cur_line = 0
|
|
repl_file = fname + ".bak"
|
|
with open(fname) as read_file, open(repl_file, "w") as write_file:
|
|
for line in read_file:
|
|
cur_line += 1
|
|
if lines and cur_line == lines[0]:
|
|
lines.pop(0)
|
|
continue
|
|
write_file.write(line)
|
|
os.rename(repl_file, fname)
|
|
|
|
|
|
parse_file(TODOTXT_FILE)
|
|
sys.exit(0)
|