TaskTracker/TaskTracker.py at main · xerchion/TaskTracker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import sys
from typing import List

from Action import Action
from constants import NAME_FILE, USER_MSG
from File import File
from TodoList import TodoList
from View import View


def catch_intro() -> List[str]:
    return sys.argv[1:] if len(sys.argv) > 1 else []


def run_user_action(action, todo_list, file):
    task_id = None
    if action.get_name() in ["delete", "mark", "update"]:
        task_id = int(action.get_args()[1])

    if action.get_name() == "add":
        description = action.get_args()[1]
        if not todo_list.task_exists(description):
            todo_list.add_task(description)
            file.save_data(todo_list.to_dict())
            view.ok(USER_MSG["TASK_ADDED_OK"])
        else:
            view.alert(USER_MSG["TASK_ALREADY_EXISTS"])
            todo_list.prepare_view(todo_list.get_task_by_description(description))

    elif action.get_name() == "delete":
        if todo_list.delete_task(task_id):
            file.save_data(todo_list.to_dict())
            view.ok(USER_MSG["TASK_DELETED_OK"])
        else:
            view.alert(USER_MSG["ID_NO_EXISTS"] + ":", task_id)

    elif action.get_name() == "update":
        if todo_list.get_task_by_id(task_id):
            task_new_name = action.get_args()[2]
            todo_list.update_task(task_id, task_new_name)
            file.save_data(todo_list.to_dict())
            view.ok(USER_MSG["TASK_UPDATED_OK"])
        else:
            view.alert(USER_MSG["ID_NO_EXISTS"] + ":", task_id)

    elif action.get_name()[:4] == "mark":
        task_id = int(action.get_args()[1])
        if todo_list.get_task_by_id(task_id):
            todo_list.mark_task(task_id, action.get_name()[5:])
            file.save_data(todo_list.to_dict())
            view.ok(USER_MSG["TASK_UPDATED_OK"])

    elif action.get_name() == "list":
        filter = "all"
        if action.get_n_args() == 1:
            filter = action.get_args()[1]
        tasks = todo_list.filter_by(filter)
        if len(tasks) != 0:
            view.info(USER_MSG["FILTERED_TASKS"] + filter + "\n")
            view.info(todo_list.prepare_view(tasks))

        else:
            view.alert(USER_MSG["NO_TASK_IN_STATUS"] + filter)


view = View()

# Load data
file = File(NAME_FILE)
if not file.exists():
    file.save_data([])
todo_list = TodoList(file.extract_data())

# Load user action
arguments = catch_intro()
action = Action(arguments)


if action.is_valid():
    run_user_action(action, todo_list, file)
else:
    view.alert(action.get_error_message())
    view.info(USER_MSG["HELP"])