create task, list
This commit is contained in:
Arnas Udovicius
parent
b068ff0b42
commit
2b48db5b62
6 changed files with 258 additions and 2 deletions
33
abcex.py
Normal file
33
abcex.py
Normal file
|
|
@ -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
|
||||
|
||||
12
bcolors.py
Normal file
12
bcolors.py
Normal file
|
|
@ -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'
|
||||
BIN
default.db
BIN
default.db
Binary file not shown.
57
model.py
Normal file
57
model.py
Normal file
|
|
@ -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()
|
||||
35
tags
35
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
|
||||
|
|
|
|||
123
ztm.py
123
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()
|
||||
|
|
|
|||
Reference in a new issue