Initial commit

1 files changed, 119 insertions, 0 deletions

diff --git a/todoist b/todoist
new file mode 100755
index 0000000..5d16e14
--- /dev/null
+++ b/todoist
@@ -0,0 +1,119 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+import argparse
+import json
+
+import todoist
+
+from pprint import pprint
+
+TOKEN_FILE_NAME = ".todoist_token"
+DUMP_FILENAME = "dump.txt"
+
+def arg_parser():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--token",
+                        metavar="ACCESS_TOKEN_STRING",
+                        help="Store or change access token",
+                        type=str)
+    parser.add_argument("--dump",
+                        help="Dump data to file to " + DUMP_FILENAME + " and exit",
+                        action="store_true")
+    parser.add_argument("-p", "--project",
+                        metavar="PROJECT_NAME",
+                        help="Project name (lowercase!)",
+                        type=str)
+    parser.add_argument("-s", "--severity",
+                        help="Severity (or priority) of task",
+                        type=int,
+                        choices=[1, 2, 3, 4])
+    parser.add_argument("-d", "--date",
+                        metavar="DATE",
+                        help="Date string",
+                        type=str,
+                        default="today")
+    parser.add_argument("msg",
+                        metavar="MSG",
+                        nargs='*',
+                        help="Content of new Task - stdin if absent")
+    return parser.parse_args()
+
+def token_path():
+    return os.path.join(os.path.expanduser("~"), TOKEN_FILE_NAME)
+
+def get_token():
+    path = token_path()
+    with open(path, mode="r") as f:
+        for line in f:
+            token = line.strip()
+            ## Get first line only
+            break
+    return token
+
+def store_token(token):
+    path = token_path()
+    with open(path, mode="w+") as f:
+        f.write(token)
+
+def build_project_list(data):
+    projects = { proj["name"].lower(): proj["id"] for proj in data["Projects"] }
+    return projects
+
+def main():
+    args = arg_parser()
+
+    if args.token:
+        store_token(args.token)
+        print("Token stored")
+        sys.exit(0)
+
+    token = get_token()
+    if not token:
+        print("Couldn't read token. Is token stored?", file=sys.stderr)
+        sys.exit(1)
+
+    api = todoist.TodoistAPI(token)
+    data = api.sync(resource_types=['all'])
+    if "error" in data:
+        print("Failed to connect or authenticate to server:", file=sys.stderr)
+        print(data["error"], file=sys.stderr)
+        sys.exit(1)
+
+    if args.dump:
+        with open(DUMP_FILENAME, mode="w") as f:
+            pprint(data, f)
+        sys.exit(0)
+
+    projects = build_project_list(data)
+
+    ## Prepare data
+    if not args.msg:
+        msg = sys.stdin.read().strip()
+    else:
+        msg = " ".join(args.msg)
+
+    ## Prepare arguments
+    kwargs = {}
+    kwargs["date_string"] = args.date
+    if args.project:
+        if args.project in projects:
+            kwargs["project_id"] = projects[args.project]
+        else:
+            print("WARNING: Unknown project. Task will be stored in Inbox", file=sys.stderr)
+
+    if args.severity:
+        kwargs["priority"] = args.severity
+
+    ## Store task
+    status = api.add_item(msg, **kwargs)
+    if "error" in status:
+        print("Store failed:", file=sys.stderr)
+        print(status["error"], file=sys.stderr)
+        sys.exit(1)
+
+    api.commit()
+
+if __name__ == "__main__":
+    main()