notes cmd
This commit is contained in:
parent
f92430de33
commit
1ad64d7859
@ -8,6 +8,7 @@ import list_cmd
|
||||
import attach_cmd
|
||||
import open_cmd
|
||||
import tag_cmd
|
||||
import note_cmd
|
||||
# bulk
|
||||
import export_cmd
|
||||
import import_cmd
|
||||
|
27
papers/commands/note_cmd.py
Normal file
27
papers/commands/note_cmd.py
Normal file
@ -0,0 +1,27 @@
|
||||
from .. import repo
|
||||
from .. import content
|
||||
from ..configs import config
|
||||
from ..uis import get_ui
|
||||
|
||||
def parser(subparsers):
|
||||
parser = subparsers.add_parser('note',
|
||||
help='edit the note attached to a paper')
|
||||
parser.add_argument('citekey',
|
||||
help='citekey of the paper')
|
||||
return parser
|
||||
|
||||
|
||||
def command(args):
|
||||
"""
|
||||
"""
|
||||
|
||||
ui = get_ui()
|
||||
|
||||
|
||||
rp = repo.Repository(config())
|
||||
if not rp.databroker.exists(args.citekey):
|
||||
ui.error("citekey {} not found".format(args.citekey))
|
||||
ui.exit(1)
|
||||
|
||||
notepath = rp.databroker.real_notepath('notesdir://{}.txt'.format(args.citekey))
|
||||
content.edit_file(config().edit_cmd, notepath, temporary=False)
|
@ -12,7 +12,8 @@ class DataBroker(object):
|
||||
def __init__(self, directory, create=False):
|
||||
self.filebroker = filebroker.FileBroker(directory, create=create)
|
||||
self.endecoder = endecoder.EnDecoder()
|
||||
self.docbroker = filebroker.DocBroker(directory)
|
||||
self.docbroker = filebroker.DocBroker(directory, scheme='docsdir', subdir='doc')
|
||||
self.notebroker = filebroker.DocBroker(directory, scheme='notesdir', subdir='notes')
|
||||
|
||||
# filebroker+endecoder
|
||||
|
||||
@ -52,8 +53,8 @@ class DataBroker(object):
|
||||
|
||||
# docbroker
|
||||
|
||||
def is_pubsdir_doc(self, docpath):
|
||||
return self.docbroker.is_pubsdir_doc(docpath)
|
||||
def in_docsdir(self, docpath):
|
||||
return self.docbroker.in_docsdir(docpath)
|
||||
|
||||
def copy_doc(self, citekey, source_path, overwrite=False):
|
||||
return self.docbroker.copy_doc(citekey, source_path, overwrite=overwrite)
|
||||
@ -63,3 +64,12 @@ class DataBroker(object):
|
||||
|
||||
def real_docpath(self, docpath):
|
||||
return self.docbroker.real_docpath(docpath)
|
||||
|
||||
|
||||
# notesbroker
|
||||
|
||||
def in_notesdir(self, docpath):
|
||||
return self.notebroker.in_docsdir(docpath)
|
||||
|
||||
def real_notepath(self, docpath):
|
||||
return self.notebroker.real_docpath(docpath)
|
@ -60,10 +60,10 @@ class DataCache(object):
|
||||
"""Will return None if bibdata_raw can't be decoded"""
|
||||
return self.databroker.verify(bibdata_raw)
|
||||
|
||||
# docbroker
|
||||
# docbroker
|
||||
|
||||
def is_pubsdir_doc(self, docpath):
|
||||
return self.databroker.is_pubsdir_doc(docpath)
|
||||
def in_docsdir(self, docpath):
|
||||
return self.databroker.in_docsdir(docpath)
|
||||
|
||||
def copy_doc(self, citekey, source_path, overwrite=False):
|
||||
return self.databroker.copy_doc(citekey, source_path, overwrite=overwrite)
|
||||
@ -74,6 +74,15 @@ class DataCache(object):
|
||||
def real_docpath(self, docpath):
|
||||
return self.databroker.real_docpath(docpath)
|
||||
|
||||
# notesbroker
|
||||
|
||||
def in_notesdir(self, docpath):
|
||||
return self.databroker.in_notesdir(docpath)
|
||||
|
||||
def real_notepath(self, docpath):
|
||||
return self.databroker.real_notepath(docpath)
|
||||
|
||||
|
||||
# class ChangeTracker(object):
|
||||
|
||||
# def __init__(self, cache, directory):
|
||||
|
@ -107,38 +107,37 @@ class DocBroker(object):
|
||||
|
||||
* only one document can be attached to a paper (might change in the future)
|
||||
* this document can be anything, the content is never processed.
|
||||
* these document have an adress of the type "pubsdir://doc/citekey.pdf"
|
||||
* these document have an adress of the type "docsdir://citekey.pdf"
|
||||
* docsdir:// correspond to /path/to/pubsdir/doc (configurable)
|
||||
* document outside of the repository will not be removed.
|
||||
* deliberately, there is no move_doc method.
|
||||
"""
|
||||
|
||||
def __init__(self, directory):
|
||||
self.docdir = os.path.join(directory, 'doc')
|
||||
def __init__(self, directory, scheme='docsdir', subdir='doc'):
|
||||
self.scheme = scheme
|
||||
self.docdir = os.path.join(directory, subdir)
|
||||
if not check_directory(self.docdir, fail = False):
|
||||
os.mkdir(self.docdir)
|
||||
|
||||
def is_pubsdir_doc(self, docpath):
|
||||
def in_docsdir(self, docpath):
|
||||
try:
|
||||
parsed = urlparse.urlparse(docpath)
|
||||
except Exception:
|
||||
return False
|
||||
if parsed.scheme == 'pubsdir':
|
||||
assert parsed.netloc == 'doc'
|
||||
assert parsed.path[0] == '/'
|
||||
return parsed.scheme == 'pubsdir'
|
||||
return parsed.scheme == self.scheme
|
||||
|
||||
def copy_doc(self, citekey, source_path, overwrite=False):
|
||||
""" Copy a document to the pubsdir/doc, and return the location
|
||||
|
||||
The document will be named {citekey}.{ext}.
|
||||
The location will be pubsdir://doc/{citekey}.{ext}.
|
||||
The location will be docsdir://{citekey}.{ext}.
|
||||
:param overwrite: will overwrite existing file.
|
||||
:return: the above location
|
||||
"""
|
||||
full_source_path = self.real_docpath(source_path)
|
||||
check_file(full_source_path)
|
||||
|
||||
target_path = 'pubsdir://' + os.path.join('doc', citekey + os.path.splitext(source_path)[-1])
|
||||
target_path = '{}://{}'.format(self.scheme, citekey + os.path.splitext(source_path)[-1])
|
||||
full_target_path = self.real_docpath(target_path)
|
||||
if not overwrite and check_file(full_target_path, fail=False):
|
||||
raise IOError('{} file exists.'.format(full_target_path))
|
||||
@ -147,11 +146,11 @@ class DocBroker(object):
|
||||
return target_path
|
||||
|
||||
def remove_doc(self, docpath, silent=True):
|
||||
""" Will remove only file hosted in pubsdir://doc/
|
||||
""" Will remove only file hosted in docsdir://
|
||||
|
||||
:raise ValueError: for other paths, unless :param silent: is True
|
||||
"""
|
||||
if not self.is_pubsdir_doc(docpath):
|
||||
if not self.in_docsdir(docpath):
|
||||
if not silent:
|
||||
raise ValueError(('the file to be removed {} is set as external. '
|
||||
'you should remove it manually.').format(docpath))
|
||||
@ -165,7 +164,10 @@ class DocBroker(object):
|
||||
Essentially transform pubsdir://doc/{citekey}.{ext} to /path/to/pubsdir/doc/{citekey}.{ext}.
|
||||
Return absoluted paths of regular ones otherwise.
|
||||
"""
|
||||
if self.is_pubsdir_doc(docpath):
|
||||
if self.in_docsdir(docpath):
|
||||
parsed = urlparse.urlparse(docpath)
|
||||
docpath = os.path.join(self.docdir, parsed.path[1:])
|
||||
if parsed.path == '':
|
||||
docpath = os.path.join(self.docdir, parsed.netloc)
|
||||
else:
|
||||
docpath = os.path.join(self.docdir, parsed.netloc, parsed.path[1:])
|
||||
return os.path.normpath(os.path.abspath(docpath))
|
||||
|
@ -22,6 +22,7 @@ CORE_CMDS = collections.OrderedDict([
|
||||
('attach', commands.attach_cmd),
|
||||
('open', commands.open_cmd),
|
||||
('tag', commands.tag_cmd),
|
||||
('note', commands.note_cmd),
|
||||
|
||||
('export', commands.export_cmd),
|
||||
('import', commands.import_cmd),
|
||||
|
@ -109,7 +109,7 @@ class Repository(object):
|
||||
paper.bibdata = new_bibdata
|
||||
|
||||
# move doc file if necessary
|
||||
if self.databroker.is_pubsdir_doc(paper.docpath):
|
||||
if self.databroker.in_docsdir(paper.docpath):
|
||||
new_docpath = self.databroker.copy_doc(new_citekey, paper.docpath)
|
||||
self.databroker.remove_doc(paper.docpath)
|
||||
paper.docpath = new_docpath
|
||||
|
@ -67,8 +67,8 @@ class TestDataBroker(TestFakeFs):
|
||||
with self.assertRaises(IOError):
|
||||
db.pull_metadata('citekey')
|
||||
|
||||
db.copy_doc('Larry99', 'pubsdir://doc/Page99.pdf')
|
||||
db.copy_doc('Larry99', 'docsdir://Page99.pdf')
|
||||
self.assertTrue(content.check_file('repo/doc/Page99.pdf', fail=False))
|
||||
self.assertTrue(content.check_file('repo/doc/Larry99.pdf', fail=False))
|
||||
|
||||
db.remove_doc('pubsdir://doc/Page99.pdf')
|
||||
db.remove_doc('docsdir://Page99.pdf')
|
||||
|
@ -101,9 +101,9 @@ class TestDocBroker(TestFakeFs):
|
||||
docpath = docb.copy_doc('Page99', 'data/pagerank.pdf')
|
||||
self.assertTrue(content.check_file(os.path.join('tmpdir', 'doc/Page99.pdf')))
|
||||
|
||||
self.assertTrue(docb.is_pubsdir_doc(docpath))
|
||||
self.assertEqual(docpath, 'pubsdir://doc/Page99.pdf')
|
||||
docb.remove_doc('pubsdir://doc/Page99.pdf')
|
||||
self.assertTrue(docb.in_docsdir(docpath))
|
||||
self.assertEqual(docpath, 'docsdir://Page99.pdf')
|
||||
docb.remove_doc('docsdir://Page99.pdf')
|
||||
|
||||
self.assertFalse(content.check_file(os.path.join('tmpdir', 'doc/Page99.pdf'), fail=False))
|
||||
with self.assertRaises(IOError):
|
||||
|
@ -100,13 +100,13 @@ class TestInit(CommandTestCase):
|
||||
pubsdir = os.path.expanduser('~/papers_test2')
|
||||
papers_cmd.execute('papers init -p {}'.format(pubsdir).split())
|
||||
self.assertEqual(set(self.fs['os'].listdir(pubsdir)),
|
||||
{'bib', 'doc', 'meta'})
|
||||
{'bib', 'doc', 'meta', 'notes'})
|
||||
|
||||
def test_init2(self):
|
||||
pubsdir = os.path.expanduser('~/.papers')
|
||||
papers_cmd.execute('papers init'.split())
|
||||
self.assertEqual(set(self.fs['os'].listdir(pubsdir)),
|
||||
{'bib', 'doc', 'meta'})
|
||||
{'bib', 'doc', 'meta', 'notes'})
|
||||
|
||||
class TestAdd(DataCommandTestCase):
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user