From 8eede504c09dc831fdbcf3ca282e81ad49acaa20 Mon Sep 17 00:00:00 2001
From: Arnas Udovicius <arnas.u@estina.lt>
Date: Thu, 15 Feb 2018 09:16:20 +0200
Subject: [PATCH] coloors, unicode in search, back from manage task, error on
 wrong key

---
 ztm.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/ztm.py b/ztm.py
index 63214f5..cb5572d 100644
--- a/ztm.py
+++ b/ztm.py
@@ -48,6 +48,10 @@ Task manager from Zordsdavini (2018)
         elif menu == '+':
             self.add()
 
+        else:
+            print(bcolors.FAIL + 'This is not implemented...\n' + bcolors.ENDC)
+            self.menu()
+
     def about(self):
         print(bcolors.WARNING + '''
 Short instruction
@@ -99,7 +103,7 @@ q - exit
         self.manage_task_menu(aid)
 
     def manage_task_menu(self, aid):
-        menu = input(bcolors.OKGREEN + 'What you want to do? (?e*+-v&q) ' + bcolors.ENDC)
+        menu = input(bcolors.OKGREEN + 'What you want to do? (?e*+-v&><q) ' + bcolors.ENDC)
 
         if menu == 'q':
             self.bye()
@@ -110,6 +114,13 @@ q - exit
         elif menu == 'e':
             self.edit_task(aid)
 
+        elif menu == '<':
+            self.menu()
+
+        else:
+            print(bcolors.FAIL + 'This is not implemented...\n' + bcolors.ENDC)
+            self.manage_task_menu(aid)
+
     def manage_task_about(self, aid):
         print(bcolors.WARNING + '''
 Short instruction
@@ -121,6 +132,8 @@ e - edit content
 - - remove tag
 v - mark done
 & - add child task
+> - go to child
+< - back
 q - exit
         ''' + bcolors.ENDC)
         self.manage_task_menu(aid)
@@ -162,7 +175,7 @@ Created:     %s
                 found = True
 
         self.model.save_content(aid, content)
-        print(bcolors.OKBLUE + '[content has been saved]/n' + bcolors.ENDC)
+        print(bcolors.OKBLUE + '\n[content has been saved]' + bcolors.ENDC)
         self.manage_task(aid)
 
     def bye(self):