Show duration with a green orange red color

1 files changed, 7 insertions, 4 deletions

diff --git a/src/gui/tasks/table/model.py b/src/gui/tasks/table/model.py
index 062ded6..8374d03 100644
--- a/src/gui/tasks/table/model.py
+++ b/src/gui/tasks/table/model.py
@@ -48,10 +48,13 @@ class TableModel(QtCore.QAbstractTableModel):
                 return priority.format(task.priority)
             elif index.column() == 5:
                 return task.tag
-        elif role == Qt.ForegroundRole and index.column() == 3:
-            return QtGui.QBrush(difficulty_color(task.difficulty))
-        elif role == Qt.ForegroundRole and index.column() == 4:
-            return QtGui.QBrush(priority_color(task.priority))
+        elif role == Qt.ForegroundRole:
+            if index.column() == 2:
+                return QtGui.QBrush(gui.tasks.duration.color(task.duration))
+            elif index.column() == 3:
+                return QtGui.QBrush(difficulty_color(task.difficulty))
+            elif index.column() == 4:
+                return QtGui.QBrush(priority_color(task.priority))
         else:
             return QtCore.QVariant()