dc4e118c3c
This involved many changes, some side effects of the change include: - remove of all `u"abc"` forms, in favor of `from __future__ import unicode_literals`. Their usage was inconsistent anyway, leading to problems when mixing with unicode content. - improve the tests, to allow printing for usecase even when crashing. Should make future test easier. This is done with a rather hacky `StdIO` class in `p3`, but it works. - for some reason, the skipped test for Python 2 seems to work now. While the previous point might seem related, it is not clear that this is actually the case.
170 lines
6.8 KiB
Python
170 lines
6.8 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
from .. import repo
|
|
from .. import color
|
|
from ..uis import get_ui
|
|
from .. import content
|
|
from ..utils import resolve_citekey, resolve_citekey_list
|
|
from ..completion import CiteKeyCompletion
|
|
|
|
|
|
# doc --+- add $file $key [[-L|--link] | [-M|--move]] [-f|--force]
|
|
# +- remove $key [$key [...]] [-f|--force]
|
|
# +- export $key [$path]
|
|
# +- open $key [-w|--with $cmd]
|
|
# supplements attach, open
|
|
|
|
def parser(subparsers, conf):
|
|
doc_parser = subparsers.add_parser(
|
|
'doc',
|
|
help='manage the document relating to a publication')
|
|
doc_subparsers = doc_parser.add_subparsers(
|
|
title='document actions', dest='action',
|
|
help='actions to interact with the documents')
|
|
doc_subparsers.required = True
|
|
|
|
add_parser = doc_subparsers.add_parser('add', help='add a document to a publication')
|
|
add_parser.add_argument('-f', '--force', action='store_true', dest='force', default=False,
|
|
help='force overwriting an already assigned document')
|
|
add_parser.add_argument('document', nargs=1, help='document file to assign')
|
|
add_parser.add_argument('citekey', nargs=1, help='citekey of the publication'
|
|
).completer = CiteKeyCompletion(conf)
|
|
add_exclusives = add_parser.add_mutually_exclusive_group()
|
|
add_exclusives.add_argument(
|
|
'-L', '--link', action='store_false', dest='link', default=False,
|
|
help='do not copy document files, just create a link')
|
|
add_exclusives.add_argument(
|
|
'-M', '--move', action='store_true', dest='move', default=False,
|
|
help='move document instead of of copying (ignored if --link)')
|
|
|
|
remove_parser = doc_subparsers.add_parser('remove', help='remove assigned documents from publications')
|
|
remove_parser.add_argument('citekeys', nargs='+', help='citekeys of the publications'
|
|
).completer = CiteKeyCompletion(conf)
|
|
remove_parser.add_argument('-f', '--force', action='store_true', dest='force',
|
|
default=False,
|
|
help='force removing assigned documents')
|
|
|
|
# favor key+ path over: key
|
|
export_parser = doc_subparsers.add_parser('export', help='export assigned documents to given path')
|
|
export_parser.add_argument('citekeys', nargs='+',
|
|
help='citekeys of the documents to export'
|
|
).completer = CiteKeyCompletion(conf)
|
|
export_parser.add_argument('path', nargs=1, help='directory to export the files to')
|
|
|
|
open_parser = doc_subparsers.add_parser('open', help='open an assigned document')
|
|
open_parser.add_argument('citekey', nargs=1, help='citekey of the document to open'
|
|
).completer = CiteKeyCompletion(conf)
|
|
open_parser.add_argument('-w', '--with', dest='cmd', help='command to open the file with')
|
|
|
|
return doc_parser
|
|
|
|
|
|
def command(conf, args):
|
|
|
|
ui = get_ui()
|
|
rp = repo.Repository(conf)
|
|
|
|
# print(args)
|
|
# ui.exit()
|
|
|
|
if args.action == 'add':
|
|
citekey = resolve_citekey(rp, args.citekey[0], ui=ui, exit_on_fail=True)
|
|
paper = rp.pull_paper(citekey)
|
|
|
|
if paper.docpath is not None and not args.force:
|
|
msg = ("The publication {} has already the document {} assigned.{newline}Overwrite? "
|
|
).format(color.dye_out(paper.citekey, 'citekey'),
|
|
color.dye_out(paper.docpath, 'filepath'),
|
|
newline=os.linesep)
|
|
if not ui.input_yn(question=msg, default='n'):
|
|
ui.exit(0)
|
|
else:
|
|
rp.remove_doc(paper.citekey)
|
|
|
|
document = args.document[0]
|
|
if args.link:
|
|
rp.push_doc(paper.citekey, document, copy=False)
|
|
else:
|
|
rp.push_doc(paper.citekey, document, copy=True)
|
|
if not args.link and args.move:
|
|
content.remove_file(document)
|
|
|
|
ui.message('{} added to {}'.format(
|
|
color.dye_out(document, 'filepath'),
|
|
color.dye_out(paper.citekey, 'citekey')))
|
|
|
|
elif args.action == 'remove':
|
|
|
|
for key in resolve_citekey_list(rp, args.citekeys, ui=ui, exit_on_fail=True):
|
|
paper = rp.pull_paper(key)
|
|
|
|
# if there is no document (and the user cares) -> inform + continue
|
|
if paper.docpath is None and not args.force:
|
|
ui.message('Publication {} has no assigned document. Not removed.'.format(
|
|
color.dye_out(paper.citekey, 'citekey')))
|
|
continue
|
|
|
|
if not args.force:
|
|
msg = 'Do you really want to remove {} from {} ?'.format(color.dye_out(paper.docpath, 'filepath'),
|
|
color.dye_out(paper.citekey, 'citekey'))
|
|
if not ui.input_yn(question=msg, default='n'):
|
|
continue
|
|
|
|
rp.remove_doc(paper.citekey)
|
|
|
|
elif args.action == 'export':
|
|
|
|
if os.path.isdir(args.path[0]):
|
|
path = args.path[0]
|
|
if not path.endswith('/'):
|
|
path += '/'
|
|
else:
|
|
ui.error('{} is not a directory.'.format(
|
|
color.dye_err(args.path[0], 'filepath')))
|
|
ui.exit(1)
|
|
|
|
for key in resolve_citekey_list(rp, args.citekeys, ui=ui, exit_on_fail=True):
|
|
try:
|
|
paper = rp.pull_paper(key)
|
|
doc = paper.docpath
|
|
|
|
if doc is None:
|
|
ui.message('Publication {} has no document assigned.'.format(
|
|
color.dye_out(paper.citekey, 'citekey')))
|
|
else:
|
|
real_doc_path = rp.pull_docpath(key)
|
|
dest_path = path + os.path.basename(real_doc_path)
|
|
content.copy_content(real_doc_path, dest_path)
|
|
except (ValueError, IOError) as e:
|
|
ui.error(str(e))
|
|
|
|
elif args.action == 'open':
|
|
with_command = args.cmd
|
|
citekey = resolve_citekey(rp, args.citekey[0], ui=ui, exit_on_fail=True)
|
|
paper = rp.pull_paper(citekey)
|
|
|
|
if paper.docpath is None:
|
|
ui.error('No document associated with the entry {}.'.format(
|
|
color.dye_err(citekey, 'citekey')))
|
|
ui.exit()
|
|
|
|
if with_command is None:
|
|
with_command = conf['main']['open_cmd']
|
|
if with_command is None: # default in conf have not been changed
|
|
pass # TODO platform specific
|
|
|
|
try:
|
|
docpath = content.system_path(rp.databroker.real_docpath(paper.docpath))
|
|
cmd = with_command.split()
|
|
cmd.append(docpath)
|
|
subprocess.Popen(cmd)
|
|
ui.message('{} opened.'.format(color.dye_out(docpath, 'filepath')))
|
|
except OSError:
|
|
ui.error("Command does not exist: %s." % with_command)
|
|
ui.exit(127)
|
|
|
|
rp.close()
|