You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.9 KiB
85 lines
2.9 KiB
from . import filebroker
|
|
from . import endecoder
|
|
|
|
|
|
class DataBroker(object):
|
|
""" DataBroker class
|
|
|
|
This is aimed at being a simple, high level interface to the content stored on disk.
|
|
Requests are optimistically made, and exceptions are raised if something goes wrong.
|
|
"""
|
|
|
|
def __init__(self, directory, create=False):
|
|
self.filebroker = filebroker.FileBroker(directory, create=create)
|
|
self.endecoder = endecoder.EnDecoder()
|
|
self.docbroker = filebroker.DocBroker(directory, scheme='docsdir', subdir='doc')
|
|
self.notebroker = filebroker.DocBroker(directory, scheme='notesdir', subdir='notes')
|
|
|
|
# filebroker+endecoder
|
|
|
|
def pull_metadata(self, citekey):
|
|
metadata_raw = self.filebroker.pull_metafile(citekey)
|
|
return self.endecoder.decode_metadata(metadata_raw)
|
|
|
|
def pull_bibdata(self, citekey):
|
|
bibdata_raw = self.filebroker.pull_bibfile(citekey)
|
|
return self.endecoder.decode_bibdata(bibdata_raw)
|
|
|
|
def push_metadata(self, citekey, metadata):
|
|
metadata_raw = self.endecoder.encode_metadata(metadata)
|
|
self.filebroker.push_metafile(citekey, metadata_raw)
|
|
|
|
def push_bibdata(self, citekey, bibdata):
|
|
bibdata_raw = self.endecoder.encode_bibdata(bibdata)
|
|
self.filebroker.push_bibfile(citekey, bibdata_raw)
|
|
|
|
def push(self, citekey, metadata, bibdata):
|
|
self.filebroker.push(citekey, metadata, bibdata)
|
|
|
|
def remove(self, citekey):
|
|
self.filebroker.remove(citekey)
|
|
|
|
def exists(self, citekey, both = True):
|
|
return self.filebroker.exists(citekey, both=both)
|
|
|
|
def listing(self, filestats=True):
|
|
return self.filebroker.listing(filestats=filestats)
|
|
|
|
def verify(self, bibdata_raw):
|
|
try:
|
|
return self.endecoder.decode_bibdata(bibdata_raw)
|
|
except ValueError:
|
|
return None
|
|
|
|
# docbroker
|
|
|
|
def in_docsdir(self, docpath):
|
|
return self.docbroker.in_docsdir(docpath)
|
|
|
|
def real_docpath(self, docpath):
|
|
return self.docbroker.real_docpath(docpath)
|
|
|
|
def add_doc(self, citekey, source_path, overwrite=False):
|
|
return self.docbroker.add_doc(citekey, source_path, overwrite=overwrite)
|
|
|
|
def remove_doc(self, docpath, silent=True):
|
|
return self.docbroker.remove_doc(docpath, silent=silent)
|
|
|
|
def rename_doc(self, docpath, new_citekey):
|
|
return self.docbroker.rename_doc(docpath, new_citekey)
|
|
|
|
# notesbroker
|
|
|
|
def real_notepath(self, citekey):
|
|
notepath = 'notesdir://{}.txt'.format(citekey)
|
|
return self.notebroker.real_docpath(notepath)
|
|
|
|
def remove_note(self, citekey, silent=True):
|
|
notepath = 'notesdir://{}.txt'.format(citekey)
|
|
return self.notebroker.remove_doc(notepath, silent=silent)
|
|
|
|
def rename_note(self, old_citekey, new_citekey):
|
|
notepath = 'notesdir://{}.txt'.format(old_citekey)
|
|
return self.notebroker.rename_doc(notepath, new_citekey)
|
|
|