89 lines
No EOL
2.7 KiB
Python
89 lines
No EOL
2.7 KiB
Python
import os
|
|
import yaml
|
|
|
|
class OrvaniTag:
|
|
|
|
DB = dict()
|
|
FILE = ''
|
|
CTX = {
|
|
'contexts' : '@',
|
|
'projects' : '+',
|
|
'tags' : '#'
|
|
}
|
|
|
|
def __init__(self, path: str='context.yaml', contexts=CTX):
|
|
self.FILE = path
|
|
self.__cd()
|
|
|
|
def __cd(self):
|
|
"""change file context to subdirectory"""
|
|
if os.path.isfile(self.FILE):
|
|
try:
|
|
with open(self.FILE, 'r') as data:
|
|
self.DB = yaml.safe_load(data)
|
|
except Exception:
|
|
print('File does not exist')
|
|
else:
|
|
self.DB = dict()
|
|
|
|
def __save(self):
|
|
"""save context file to current directory"""
|
|
try:
|
|
with open(self.FILE, 'w+') as data:
|
|
yaml.dump(self.DB, data)
|
|
except Exception:
|
|
print('File write error')
|
|
|
|
def __context(self, argv) -> dict:
|
|
"""create context for given filename"""
|
|
context = dict()
|
|
for ctx in self.CTX.keys():
|
|
context.update( { ctx : []})
|
|
for arg in argv:
|
|
for ctx in self.CTX.keys():
|
|
if str(arg)[0] in self.CTX.values():
|
|
if str(arg).startswith(self.CTX[ctx]):
|
|
context[ctx].append(str(arg)[1:])
|
|
else:
|
|
context.update( { 'note' : str(arg)})
|
|
return context
|
|
|
|
def add(self, file: str, *argv: str):
|
|
if os.path.dirname(file):
|
|
self.FILE = os.path.join(os.path.dirname(file), self.FILE)
|
|
self.__cd()
|
|
file = os.path.basename(file)
|
|
context = self.__context(argv)
|
|
if file in self.DB:
|
|
for ctx in self.CTX.keys():
|
|
self.DB[file][ctx] = list(set( self.DB[file][ctx] + context[ctx]))
|
|
if 'note' in context.keys():
|
|
self.DB[file]['note'] = context['note']
|
|
else:
|
|
self.DB.update( { file : context } )
|
|
self.__save()
|
|
|
|
def remove(self, file: str, *argv: str):
|
|
if os.path.dirname(file):
|
|
self.FILE = os.path.join(os.path.dirname(file), self.FILE)
|
|
self.__cd()
|
|
file = os.path.basename(file)
|
|
context = self.__context(argv)
|
|
if file in self.DB:
|
|
for ctx in self.CTX.keys():
|
|
for item in context[ctx]:
|
|
if item in self.DB[file][ctx]:
|
|
self.DB[file][ctx].remove(item)
|
|
self.__save()
|
|
|
|
def purge(self, file: str):
|
|
if os.path.dirname(file):
|
|
self.FILE = os.path.join(os.path.dirname(file), self.FILE)
|
|
self.__cd()
|
|
file = os.path.basename(file)
|
|
if file in self.DB:
|
|
del self.DB[file]
|
|
self.__save()
|
|
|
|
o = OrvaniTag()
|
|
o.add('dir/file.txt', 'weird') |