Add task difficulty and priority fields

1 files changed, 9 insertions, 3 deletions

diff --git a/src/model/task.py b/src/model/task.py
index cc7758d..f5e0ae5 100644
--- a/src/model/task.py
+++ b/src/model/task.py
@@ -1,17 +1,23 @@
 from typing import NamedTuple
-from datetime import datetime
+
+from model.difficulty import Difficulty
+from model.priority import Priority
 
 class Task(NamedTuple):
     id: int
     created_at: int
     modified_at: int
     name: str
-    description: str
     duration: int
     tag: str
+    difficulty: Difficulty
+    priority: Priority
+    description: str
 
 class ValidTaskForm(NamedTuple):
     name: str
-    description: str
     duration: int
     tag: str
+    difficulty: Difficulty
+    priority: Priority
+    description: str