task.py

import time
from pathlib import Path
import argparse
from task_class import Task

def handle_add_task(args):
    task = Task(args.task, args.notes)
    task.create_task()

def handle_mark_tasks(args):
    tasks = Task.load_tasks()
    for task in tasks:
        if task["id"] == args.id:
            t = {
                "id": args.id,
                "title": task["task"],
                "status": args.state,
                "createdAt": task["createdAt"],
                "updatedAt": time.ctime(),
                "notes": task["notes"]
            }
            tasks[args.id -1] = t
            Task.save_tasks(tasks)

def handle_list_tasks(args):
    tasks = Task.load_tasks()
    match args.task:
        case "all":
            for task in tasks:
                print(task)
        case "to-do":
            for task in tasks:
                if task["status"] == "to-do":
                    print(task)
        case "in-progress":
            for task in tasks:
                if task["status"] == "in-progress":
                    print(task)
        case "done":
            for task in tasks:
                if task["status"] == "done":
                    print(task)

def handle_delete_task(args):
    tasks = Task.load_tasks()
    new_task_list = []
    for i in range(len(tasks)):
        if tasks[i]["id"] == args.id:
            continue
        new_task_list.append(tasks[i])
    Task.save_tasks(new_task_list)

def handle_update_task(args):
    tasks = Task.load_tasks()
    for task in tasks:
        if task["id"] == args.id:
            if task['notes'] is not None:
                task["notes"] += f'\n {args.notes}'
            else:
                task['notes'] = args.notes
            Task.save_tasks(tasks)

def main():
    parser =  argparse.ArgumentParser(prog="task-cli")
    subparser = parser.add_subparsers(help='subcommand help')
    
    # Add Subcommand
    parser_add = subparser.add_parser('add', help="add a task to the list.")
    parser_add.add_argument('task', help="Name of the task.")
    parser_add.add_argument('-n', '--notes', help='Add some comments/notes to the task.')
    parser_add.set_defaults(func=handle_add_task)
    
    # Mark Subcommand
    parser_mark = subparser.add_parser('mark', help="mark a task as: 'to-do, in-progress or done.'")
    parser_mark.add_argument('id', help="ID of the task to mark.", type=int)
    parser_mark.add_argument("state", choices=["to-do", "in-progress", "done"], type=str)
    parser_mark.set_defaults(func=handle_mark_tasks)
    
    # List Subcommand
    parser_list = subparser.add_parser("list", help="List tasks.")
    parser_list.add_argument("task", choices=["all", "to-do", "in-progress", "done"])
    parser_list.set_defaults(func=handle_list_tasks)
    
    # Delete Subcommand
    parser_delete = subparser.add_parser("delete", help="Delete a task by its ID.")
    parser_delete.add_argument("id", type=int)
    parser.set_defaults(func=handle_delete_task)
    
    # Update Subcommand
    parser_update = subparser.add_parser("update", help="Update the description of a task.")
    parser_update.add_argument("id", type=int)
    parser_update.add_argument("notes", help="The new notes of the task.")
    parser_update.set_defaults(func=handle_update_task)
    
    # Parse Args
    args = parser.parse_args()
    try:
        args.func(args)
    except AttributeError:
        print(parser.print_help())

if __name__ == "__main__":    
    main()