129 lines
No EOL
4.1 KiB
Python
129 lines
No EOL
4.1 KiB
Python
import os
|
|
import yaml
|
|
import shutil
|
|
|
|
class OrvaniTag:
|
|
|
|
DB = dict()
|
|
FILENAME = ''
|
|
FILE = ''
|
|
CTX = {
|
|
'contexts' : '@',
|
|
'projects' : '+',
|
|
'tags' : '#'
|
|
}
|
|
|
|
def __init__(self, path: str='context.yaml', contexts=CTX):
|
|
self.FILENAME = os.path.basename(path)
|
|
self.__cd(path)
|
|
|
|
def __cd(self, path: str):
|
|
"""change file context to subdirectory"""
|
|
self.FILE = os.path.join(os.path.dirname(path), self.FILENAME)
|
|
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.isfile(file):
|
|
if os.path.dirname(file):
|
|
self.__cd(file)
|
|
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.isfile(file):
|
|
if os.path.dirname(file):
|
|
self.__cd(file)
|
|
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 delete(self, file: str, fop=False):
|
|
if os.path.dirname(file):
|
|
self.__cd(file)
|
|
if fop:
|
|
os.remove(file)
|
|
file = os.path.basename(file)
|
|
if file in self.DB:
|
|
del self.DB[file]
|
|
self.__save()
|
|
|
|
def move(self, source: str, dest: str, retain=False, fop=True):
|
|
if os.path.isfile(source) and os.path.isdir(os.path.dirname(dest)):
|
|
sourcefolder = os.path.dirname(source)
|
|
sourcefile = os.path.basename(source)
|
|
destfolder = os.path.dirname(dest)
|
|
destfile = os.path.basename(dest)
|
|
if not destfile:
|
|
destfile = sourcefile
|
|
sourcedata = {}
|
|
self.__cd(source)
|
|
if self.DB[sourcefile]:
|
|
sourcedata = self.DB[sourcefile]
|
|
if not retain:
|
|
self.purge(source)
|
|
self.__cd(dest)
|
|
self.DB.update( { destfile : sourcedata } )
|
|
self.__save()
|
|
if fop:
|
|
if retain:
|
|
shutil.copy2(source, dest)
|
|
else:
|
|
shutil.move(source, dest, copy_function=shutil.copy2)
|
|
|
|
def purge(self, path: str=''):
|
|
if os.path.dirname(path):
|
|
self.__cd(path)
|
|
for file in self.DB.keys():
|
|
if not os.path.isfile(file):
|
|
if file in self.DB:
|
|
del self.DB[file]
|
|
self.__save()
|
|
if not self.DB:
|
|
os.remove(self.FILE)
|
|
|
|
|
|
o = OrvaniTag()
|
|
#o.add('TODO.md', '#tagx', '+projectX')
|
|
o.purge() |