summaryrefslogtreecommitdiffstats
path: root/todo
blob: 890213ba54ea9b53419e64f8e7003f90d587db8d (plain)
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
99
100
101
102
103
#!/usr/bin/env python3
"""
Usage: todo [--id | --date | --priority] [--reverse] [--file <file>]

Options:
  -i --id        Sort by ID
  -d --date      Sort by date
  -p --priority  Sort by priority
  -r --reverse   Reverse sort order
  -f --file      Path to todo file
  -h --help      Show this screen
  -v --version   Show version
"""

from sys import exit
from collections import OrderedDict
from os import getenv, path
from docopt import docopt
import toml


def sort_toml(toml_dict, sort_key, rev=False):
    return sorted(
            list(toml_dict.items()),
            key=lambda d: (sort_key not in d[1], d[1].get(sort_key, None)),
            reverse=rev)

class Config():
    def __init__(self):
        self.todo_file = None
        self.sort_key  = None
        self.reverse   = False

def parse_args(args):
    config = Config()
    home_dir  = path.expanduser('~')
    data_home = getenv('XDG_DATA_HOME')

    if args['--id']:
        config.sort_key = 'id'
    elif args['--date']:
        config.sort_key = 'date'
    else:
        config.sort_key = 'priority'

    config.reverse = args['--reverse']

    if args['<file>']:
        config.todo_file = args['<file>']
    else:
        if data_home:
            config.todo_file = data_home + '/todo.toml'
        else:
            config.todo_file = home_dir + '/.local/share/todo.toml'

    return config

def color_bold(text):
    color_string = '\x1b[1m{}\x1b[0m'
    return color_string.format(text)

def color_red(text):
    color_string = '\x1b[38;2;250;050;050m{}\x1b[0m'
    return color_string.format(text)

def color_yellow(text):
    color_string = '\x1b[38;2;250;150;050m{}\x1b[0m'
    return color_string.format(text)

def color_blue(text):
    color_string = '\x1b[38;2;050;150;250m{}\x1b[0m'
    return color_string.format(text)

def main():
    arguments = docopt(__doc__, version='todo 0.0.1.alpha')
    config = parse_args(arguments)
    print(arguments)

    todo_dict = toml.load(config.todo_file)

    unsorted = OrderedDict(sorted(todo_dict.items()))
    sorted_toml = OrderedDict(sort_toml(unsorted, config.sort_key, config.reverse))

    print(sorted_toml)

    for t_id in sorted_toml:
        task = sorted_toml[t_id]
        output = "{}. ".format(t_id)
        if 'priority' in task:
            output += "({}) ".format(task['priority'])
        if 'date' in task:
            output += "{}".format(task['date'])
        if 'description' in task:
            output += "\n    {}".format(task['description'])
        if 'url' in task:
            output += "\n    URL: {}".format(task['url'])
        if 'context' in task:
            output += "\n    Contexts: {}".format(task['context'])

        print(output)

if __name__ == '__main__':
    main()