From fca5ab57513adbdbca677f0fb1d453db1285775f Mon Sep 17 00:00:00 2001
From: Johannes Löthberg <johannes@kyriasis.com>
Date: Wed, 17 Dec 2014 00:22:00 +0100
Subject: todo: add format_priority to color priority

---
 todo | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'todo')

diff --git a/todo b/todo
index 289f5e5..eaecf64 100755
--- a/todo
+++ b/todo
@@ -85,6 +85,17 @@ def color_blue(text):
     color_string = '\x1b[38;2;050;150;250m{}\x1b[0m'
     return color_string.format(text)
 
+def format_priority(priority):
+    text = '({})'.format(priority)
+    if priority == 'A':
+        return color_red(text)
+    elif priority == 'B':
+        return color_yellow(text)
+    elif priority == 'C':
+        return color_blue(text)
+    else:
+        return text
+
 def main():
     arguments = docopt(__doc__, version='todo 0.0.1.alpha')
     config = parse_args(arguments)
@@ -101,7 +112,7 @@ def main():
         task = todo_dict[t_id]
         output = "{}. ".format(t_id)
         if 'priority' in task:
-            output += "({}) ".format(task['priority'])
+            output += '{} '.format(format_priority(task['priority']))
         if 'date' in task:
             output += "{}".format(task['date'])
         if 'description' in task:
-- 
cgit v1.2.3-70-g09d2