aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--todo7
1 files changed, 3 insertions, 4 deletions
diff --git a/todo b/todo
index 56c6657..fbe51eb 100644
--- a/todo
+++ b/todo
@@ -16,7 +16,7 @@ from sys import exit
from collections import OrderedDict
from os import getenv, path
from docopt import docopt
-import pytoml
+import toml
def sort_toml(toml_dict, sort_key, rev=False):
@@ -60,10 +60,9 @@ def main():
config = parse_args(arguments)
print(arguments)
- with open("/home/kyrias/documents/notes/TODO.toml", "rt") as in_file:
- tasks = in_file.read()
+ todo_dict = toml.load(config.todo_file)
- unsorted = OrderedDict(sorted(pytoml.loads(tasks).items()))
+ unsorted = OrderedDict(sorted(todo_dict.items()))
sorted_toml = OrderedDict(sort_toml(unsorted, config.sort_key, config.reverse))
print(sorted_toml)