Small improvements

This commit is contained in:
Yehuda Deutsch 2024-11-11 16:39:47 -05:00
parent c9fdf7b839
commit 030baf137e
Signed by: uda
GPG key ID: 8EF44B89374262A5

16
todo.py
View file

@ -8,11 +8,12 @@ todo_file = os.path.join(base_dir, "todo.md")
def add_task(task: str, parent=None):
print(f"Adding task '{task}' to '{parent if parent else 'root'}'")
todo = parse_todo()
if parent and parent not in todo:
print(f"Could not find parent task '{parent}' in root task list")
exit(1)
print(f"Adding task '{task}' to '{parent if parent else 'root'}'")
if parent:
todo[parent]["sub_tasks"][task] = {"completed": False}
else:
@ -21,18 +22,18 @@ def add_task(task: str, parent=None):
def list_tasks(parent=None):
print(f"Listing tasks in '{parent if parent else 'root'}'")
todo = parse_todo()
if parent and parent not in todo:
print(f"Could not find parent task '{parent}' in root task list")
exit(1)
print(f"Listing tasks in '{parent if parent else 'root'}'")
if parent:
todo = {parent: todo[parent]}
print(build_todo(todo))
def complete_task(task, parent=None):
print(f"Marking task '{task}' in '{parent if parent else 'root'}' as completed")
todo = parse_todo()
if parent:
if parent not in todo:
@ -44,6 +45,8 @@ def complete_task(task, parent=None):
if task not in todo:
print(f"Could not find task '{task}' in root task list")
exit(1)
print(f"Marking task '{task}' in '{parent if parent else 'root'}' as completed")
if parent:
task = todo[parent]["sub_tasks"][task]
else:
@ -53,7 +56,6 @@ def complete_task(task, parent=None):
def remove_task(task, parent=None):
print(f"Removing task '{task}' from '{parent if parent else 'root'}'")
todo = parse_todo()
if parent:
if parent not in todo:
@ -65,6 +67,8 @@ def remove_task(task, parent=None):
if task not in todo:
print(f"Could not find task '{task}' in root task list")
exit(1)
print(f"Removing task '{task}' from '{parent if parent else 'root'}'")
if parent:
del todo[parent]["sub_tasks"][task]
else:
@ -125,7 +129,7 @@ def main():
print_help()
case "list":
match sys.argv[2:]:
case [parent]:
case [parent, *_]:
list_tasks(parent)
case _:
list_tasks()
@ -134,7 +138,7 @@ def main():
match sys.argv[2:]:
case [task]:
globals()[task_func](task)
case [task, parent]:
case [parent, task, *_]:
globals()[task_func](task, parent)
case _:
print_help()