main.py

#!/usr/bin/env python3
import os
import click
from utils import TaskList, error, success

VERSION = "1.4"


def get_task_list():
    current_dir = os.getcwd()
    todo_file = os.path.join(current_dir, ".todo.json")
    return TaskList(todo_file)


@click.group()
def cli():
    pass


@cli.command()
@click.argument('task')
def add(task):
    """Add a new task"""
    tasks = get_task_list()
    tasks.add(task)
    success(f'"{task}" was added successfully!')


@cli.command()
@click.argument('task_id')
def rm(task_id):
    """Remove a task by ID"""
    tasks = get_task_list()
    if tasks.rm(task_id):
        success("Remove was successful.")
    else:
        raise SystemExit(1)


@cli.command()
def ls():
    """List all tasks"""
    tasks = get_task_list()
    if tasks.is_empty():
        error("No tasks created yet!")
        raise SystemExit(1)
    for id, task in tasks:
        if task.checked():
            success(f"- [X] {task.name} id: {id}")
        else:
            click.echo(f"- [ ] {task.name} id: {id}")


@cli.command()
@click.argument('task_id')
def check(task_id):
    """Mark a task as completed"""
    tasks = get_task_list()
    if tasks.check(task_id):
        success("Checked task successfully.")
    else:
        raise SystemExit(1)


@cli.command()
@click.argument('task_id')
def uncheck(task_id):
    """Mark a task as open"""
    tasks = get_task_list()
    match tasks.uncheck(task_id):
        case 0:
            success("Unchecked task successfully.")
        case 1:
            raise SystemExit(1)


@cli.command()
def version():
    """Show version"""
    click.echo(f"todo {VERSION}")


@cli.command()
def help():
    """Show this help message"""
    click.echo("Usage:")
    click.echo('  todo add "task"      Add a new task')
    click.echo("  todo rm <id>         Remove a task by ID")
    click.echo("  todo ls              List all tasks")
    click.echo("  todo check <id>      Mark a task as completed")
    click.echo("  todo uncheck <id>    Mark a task as open")
    click.echo("  todo --version, -v   Show version")
    click.echo("  todo --help          Show this help message")


if __name__ == "__main__":
    cli()