From 2b48db5b624ead079759415fd8ad3c2c290f8afa Mon Sep 17 00:00:00 2001
From: Arnas Udovicius <arnas.u@estina.lt>
Date: Wed, 14 Feb 2018 17:16:03 +0200
Subject: [PATCH] create task, list

---
 abcex.py   |  33 ++++++++++++++
 bcolors.py |  12 ++++++
 default.db | Bin 61440 -> 61440 bytes
 model.py   |  57 +++++++++++++++++++++++++
 tags       |  35 ++++++++++++++-
 ztm.py     | 123 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
 6 files changed, 258 insertions(+), 2 deletions(-)
 create mode 100644 abcex.py
 create mode 100644 bcolors.py
 create mode 100644 model.py

diff --git a/abcex.py b/abcex.py
new file mode 100644
index 0000000..5894864
--- /dev/null
+++ b/abcex.py
@@ -0,0 +1,33 @@
+# from https://github.com/zordsdavini/abcex
+
+class Abcex:
+
+    abc_map = {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9, 10: 'a',
+            11: 'b', 12: 'c', 13: 'd', 14: 'e', 15: 'f', 16: 'g', 17: 'h', 18:
+            'i', 19: 'j', 20: 'k', 21: 'l', 22: 'm', 23: 'n', 24: 'o', 25: 'p',
+            26: 'q', 27: 'r', 28: 's', 29: 't', 30: 'u', 31: 'v', 32: 'w', 33:
+            'x', 34: 'y', 35: 'z'}
+
+    inv_abc_map = {str(v): k for k, v in abc_map.items()}
+
+    def encode(self, number):
+        number = int(number)
+        result = ''
+
+        while number > 0:
+            result = "%s%s" % (self.abc_map[number % 36], result)
+            number //= 36
+
+        return result
+
+
+    def decode(self, string):
+        result = 0
+        string = str(string)
+        string = reversed(str(string))
+
+        for i, c in enumerate(string):
+            result = 36**i * self.inv_abc_map[c] + result
+
+        return result
+
diff --git a/bcolors.py b/bcolors.py
new file mode 100644
index 0000000..224364b
--- /dev/null
+++ b/bcolors.py
@@ -0,0 +1,12 @@
+# from https://stackoverflow.com/questions/287871/print-in-terminal-with-colors
+
+
+class bcolors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
diff --git a/default.db b/default.db
index 8de0e5e12c1c4c4c50d92c840f7947c3e1d581d9..3c7c613b9f8faa1c2d704e1d8922a1a66e8da523 100644
GIT binary patch
delta 464
zcmZp8z})bFd4jZ{5Ca1PKM=D6F(VL5Ow=*f6k^b0b>!vhVBqDDVBnv@@5c9)Z!?bz
z_X$2p-t}CJoQWI~yy?7hoQWGJPGHw=a^q(g7Zqh}$uCJv%1JFrEY1dD4(A|O#}HSA
z5Jx8;R|O@IkkaIhT!xdl?IzoEn@=w1*5=mXQUHR)<dV#?$?@Fc+FUvciJ2)1VU9u0
z9*#j8h6b7nexW`-Itt18c_pcNB?=*~5g}k9&CQ9N>Wrp5+ZY|dcE>guLtWuXsVjsq
zT`~C(_ZJpm=wwXZ#p9|7WNc6uWsJ_uOG&K&`71s%B_7V<a2K4sgGZK;ZSpl9O(8b{
zHgTX@gmnDmlRWa14S0FDAwi;`k(r{oc|C8a5E~=^cLx6Nn*{}4^7C;sGRv|vGK(^R
h%;(z7sPJEZVgNTQ8_;e}po;hWoBzu*DlAffA^?b1d-DJQ

delta 351
zcmZp8z})bFd4jYc9|HpeKM=zJ-$Wf_O+E%aR!3gm=?r|Fi46QR_}%!v@@?ko;C1Es
z&$Erkh5H1bB=34IM$W{|f&w==HYakbGio;3va*YdiZZsumLw+Sq?RNWXM-@CbC9cJ
zh^s<~qmz%T0!U<XAfF*`VrGg$m}8K$hhvb2;p7Q?ij)6w=`kuy*5Nh>k}o;5nYeT&
zOLEErIfuBvFu8L~R^XMLe3eIw!-btqTwRngdU8Im7$e8zHeR`Ukf{n9nJJn9Y;5A%
zqKx^Oc`2zCAklcBJ@KV*E<4mQp?;o$p{@#^elD&NU^gMDQ}Fi#3n^#-^=blxMFQ+A
sQHZbDc-g=qmBGU<ZZ6N*$PP4n^KM>Gq0KA?fAkj>C@gYNSX7Vz0F9<zvj6}9

diff --git a/model.py b/model.py
new file mode 100644
index 0000000..c2aac6c
--- /dev/null
+++ b/model.py
@@ -0,0 +1,57 @@
+import sqlite3
+
+from abcex import Abcex
+
+
+class Model:
+    def __init__(self):
+        self.conn = sqlite3.connect('ztm.db')
+        self.conn.row_factory = sqlite3.Row
+        self.abcex = Abcex()
+
+    def get_tasks_by_tag(self, tag_name):
+        query = '''
+            SELECT t.* FROM task t
+            LEFT JOIN tag_task tgt ON t.id = tgt.task_id
+            LEFT JOIN tag tg ON tg.id = tgt.tag_id
+            WHERE tg.name = ?
+        '''
+
+        return self.conn.execute(query, tag_name)
+
+    def get_all_tasks(self):
+        query = '''
+            SELECT t.* FROM task t
+        '''
+
+        return self.conn.execute(query)
+
+    def create_task_draft(self, description):
+        query = '''
+            INSERT INTO task
+            (description, created_at)
+            VALUES
+            (?, DATETIME('now'))
+        '''
+        cursor = self.conn.cursor()
+        cursor.execute(query, (description,))
+        task_id = cursor.lastrowid
+
+        aid = self.abcex.encode(task_id)
+        cursor.execute('UPDATE task SET aid=? WHERE id=?', (aid, task_id))
+
+        self.conn.commit()
+
+        return aid
+
+    def get_task(self, aid):
+        query = '''
+            SELECT t.* FROM task t
+            LEFT JOIN tag_task tgt ON t.id = tgt.task_id
+            LEFT JOIN tag tg ON tg.id = tgt.tag_id
+            WHERE t.aid = ?
+        '''
+        cursor = self.conn.cursor()
+        results = cursor.execute(query, (aid,))
+
+        return results.fetchone()
diff --git a/tags b/tags
index 45bb11e..a5ba879 100644
--- a/tags
+++ b/tags
@@ -1,2 +1,35 @@
 !_TAG_FILE_SORTED	2	/0=unsorted, 1=sorted, 2=foldcase/
-sqlite3	/home/arnas/Skleps/ztm/ztm.py	/^import sqlite3$/;"	i	language:Python
+a	/home/arnas/Skleps/ztm/ztm.py	/^    a = Main()$/;"	v	language:Python
+Abcex	/home/arnas/Skleps/ztm/model.py	/^from abcex import Abcex$/;"	i	language:Python
+about	/home/arnas/Skleps/ztm/ztm.py	/^    def about(self):$/;"	f	language:Python
+add	/home/arnas/Skleps/ztm/ztm.py	/^    def add(self):$/;"	f	language:Python
+add_tag	/home/arnas/Skleps/ztm/ztm.py	/^    def add_tag(self, task_id):$/;"	f	language:Python
+bcolors	/home/arnas/Skleps/ztm/bcolors.py	/^class bcolors:$/;"	c	language:Python
+bcolors	/home/arnas/Skleps/ztm/ztm.py	/^from bcolors import bcolors$/;"	i	language:Python
+BOLD	/home/arnas/Skleps/ztm/bcolors.py	/^    BOLD = '\\033[1m'$/;"	v	language:Python	class:bcolors
+create_task_draft	/home/arnas/Skleps/ztm/model.py	/^    def create_task_draft(self, description):$/;"	m	language:Python	class:Model
+ENDC	/home/arnas/Skleps/ztm/bcolors.py	/^    ENDC = '\\033[0m'$/;"	v	language:Python	class:bcolors
+FAIL	/home/arnas/Skleps/ztm/bcolors.py	/^    FAIL = '\\033[91m'$/;"	v	language:Python	class:bcolors
+fzfCall	/home/arnas/Skleps/ztm/ztm.py	/^    def fzfCall(self, elements):$/;"	f	language:Python
+get_all_tasks	/home/arnas/Skleps/ztm/model.py	/^    def get_all_tasks(self):$/;"	m	language:Python	class:Model
+get_task	/home/arnas/Skleps/ztm/model.py	/^    def get_task(self, aid):$/;"	m	language:Python	class:Model
+get_tasks_by_tag	/home/arnas/Skleps/ztm/model.py	/^    def get_tasks_by_tag(self, tag_name):$/;"	m	language:Python	class:Model
+HEADER	/home/arnas/Skleps/ztm/bcolors.py	/^    HEADER = '\\033[95m'$/;"	v	language:Python	class:bcolors
+Main	/home/arnas/Skleps/ztm/ztm.py	/^class Main:$/;"	c	language:Python
+manage_task	/home/arnas/Skleps/ztm/ztm.py	/^    def manage_task(self, aid):$/;"	f	language:Python
+manage_task_about	/home/arnas/Skleps/ztm/ztm.py	/^    def manage_task_about(self, aid):$/;"	f	language:Python
+manage_task_menu	/home/arnas/Skleps/ztm/ztm.py	/^    def manage_task_menu(self, aid):$/;"	f	language:Python
+menu	/home/arnas/Skleps/ztm/ztm.py	/^    def menu(self):$/;"	f	language:Python
+Model	/home/arnas/Skleps/ztm/model.py	/^class Model:$/;"	c	language:Python
+Model	/home/arnas/Skleps/ztm/ztm.py	/^from model import Model$/;"	i	language:Python
+OKBLUE	/home/arnas/Skleps/ztm/bcolors.py	/^    OKBLUE = '\\033[94m'$/;"	v	language:Python	class:bcolors
+OKGREEN	/home/arnas/Skleps/ztm/bcolors.py	/^    OKGREEN = '\\033[92m'$/;"	v	language:Python	class:bcolors
+run	/home/arnas/Skleps/ztm/ztm.py	/^    def run(self):$/;"	m	language:Python	class:Main
+search	/home/arnas/Skleps/ztm/ztm.py	/^    def search(self):$/;"	f	language:Python
+sqlite3	/home/arnas/Skleps/ztm/model.py	/^import sqlite3$/;"	i	language:Python
+subprocess	/home/arnas/Skleps/ztm/ztm.py	/^import subprocess$/;"	i	language:Python
+sys	/home/arnas/Skleps/ztm/ztm.py	/^import sys$/;"	i	language:Python
+UNDERLINE	/home/arnas/Skleps/ztm/bcolors.py	/^    UNDERLINE = '\\033[4m'$/;"	v	language:Python	class:bcolors
+WARNING	/home/arnas/Skleps/ztm/bcolors.py	/^    WARNING = '\\033[93m'$/;"	v	language:Python	class:bcolors
+__init__	/home/arnas/Skleps/ztm/model.py	/^    def __init__(self):$/;"	m	language:Python	class:Model
+__init__	/home/arnas/Skleps/ztm/ztm.py	/^    def __init__(self):$/;"	m	language:Python	class:Main
diff --git a/ztm.py b/ztm.py
index 3d58d1f..2830493 100644
--- a/ztm.py
+++ b/ztm.py
@@ -1,3 +1,124 @@
-import sqlite3
+import sys
+import subprocess
+
+from model import Model
+from bcolors import bcolors
 
 
+class Main:
+    def __init__(self):
+        self.model = Model()
+
+    def run(self):
+        print('''
+     _oo (o)__(o)  \/
+  >-(_  \(__  __) (OO)
+     / _/  (  ) ,'.--.)
+    / /     )( / /|_|_\\
+   / (     (  )| \_.--.
+  (   `-.   )/ '.   \) \\
+   `--.._) (     `-.(_.'
+
+Task manager from Zordsdavini (2018)
+                ''')
+
+        self.menu()
+
+    def menu(self):
+        menu = input('What you want to do? (?+/tcq) ')
+
+        if menu == 'q':
+            print('bye o/')
+            sys.exit(0)
+
+        elif menu == '?':
+            self.about()
+
+        elif menu == '/':
+            self.search()
+
+        elif menu == '+':
+            self.add()
+
+    def about(self):
+        print('''
+Short instruction
+-----------------
+? - help (this dialog)
++ - add
+/ - search
+t - tag manager
+c - configuration
+q - exit
+                ''')
+        self.menu()
+
+    def search(self):
+        tasks = []
+        tasksData = self.model.get_all_tasks()
+        if not tasksData:
+            self.add()
+            tasksData = self.model.get_all_tasks()
+
+        for t in self.model.get_all_tasks():
+            tasks.append('%s: %s' % (t[1], t[2]))
+
+        selected = self.fzfCall(tasks)
+
+    def add(self):
+        print('\n' + bcolors.HEADER + 'Adding task\n' + bcolors.ENDC)
+        description = input('Description: ')
+        aid = self.model.create_task_draft(description)
+        self.manage_task(aid)
+
+
+    def add_tag(self, task_id):
+        question = input('Do you want to add tag? ([y]/n) ')
+        if question == 'y' or question == '':
+            self.add_tag(task_id)
+        pass
+
+    def manage_task(self, aid):
+        task = self.model.get_task(aid)
+        print('\n' + bcolors.HEADER + 'Managing: [' + task['aid'] + '] ' + task['description'] + '\n' + bcolors.ENDC)
+
+        print()
+        self.manage_task_menu(aid)
+
+    def manage_task_menu(self, aid):
+        menu = input('What you want to do? (?e*+-v&q) ')
+
+        if menu == 'q':
+            print('bye o/')
+            sys.exit(0)
+
+        elif menu == '?':
+            self.manage_task_about(aid)
+
+    def manage_task_about(self, aid):
+        print('''
+Short instruction
+-----------------
+? - help (this dialog)
+e - edit content
+* - toggle long term
++ - add tag
+- - remove tag
+v - mark done
+& - add child task
+q - exit
+                ''')
+        self.manage_task_menu(aid)
+
+    def fzfCall(self, elements):
+        try:
+            p = subprocess.Popen(['echo', '\n'.join(elements)], stdout=subprocess.PIPE)
+            p2 = subprocess.check_output(('fzf'), stdin=p.stdout)
+            return p2
+        except:
+            return None
+
+
+if __name__ == '__main__':
+    a = Main()
+    a.run()