time-management/interface.py at master · Kpacka/time-management · 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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
def print_menu_and_get_input():
    print_ascii_banner(parse_ascii_banner("banner.txt"))
    return input('''
1: Make a note.
2: Make a TODO item and set a completion goal.
3: Print your notes and TODOs.
4: Complete a TODO.
5: Print overdue TODOs.
6: Delete your time management history.
7: Quit.
''')


def divide():
    return '-' * 150


def parse_ascii_banner(file):
    f = open(file, 'r')
    lines = f.readlines()
    f.close()
    return lines


def print_ascii_banner(lines):
    for line in lines:
        print(line.rstrip("\n"))


def run_menu_loop(proxy):
    while True:
        choice = print_menu_and_get_input()
        map_user_menu_choice_to_function(choice, proxy)


def map_user_menu_choice_to_function(choice, proxy):
    if choice == "1":
        make_a_note(proxy)
    elif choice == "2":
        set_a_todo_item_with_goal(proxy)
    elif choice == "3":
        print_contents(proxy)
    elif choice == "4":
        complete_todo(proxy)
    elif choice == "5":
        print_overdue_items(proxy)
    elif choice == "6":
        delete_history(proxy)
    elif choice == "7":
        quit_program(proxy)
    else:
        print("Choice not recognized.")


def make_a_note(proxy):
    note = input("Make a note: ")
    proxy.update_table_with_note(note)


def set_a_todo_item_with_goal(proxy):
    note = input("Make a TODO item: ")
    days_until_completion = input("Set number of days to complete: ")
    proxy.update_table_with_todo_and_goal(note, days_until_completion)


def print_contents(proxy):
    table_rows = proxy.get_all_items()
    for row in table_rows:
        print(divide())
        print(row)
    print(divide())


def complete_todo(proxy):
    row_id = input("Select note id for completion: ")
    proxy.update_completion(row_id)


def print_overdue_items(proxy):
    table_rows = proxy.get_overdue_items()
    for row in table_rows:
        print(divide())
        print(row)
    print(divide())


def delete_history(proxy):
    choice = input(
        "Are you sure you want to delete your history?\nSubmit 'y' to drop table or any other key to return the menu: ")
    if choice == "y":
        print("Deleting history...")
        proxy.delete_history()


def quit_program(proxy):
    print("Get outta here!")
    proxy.disconnect()
    quit()