pubs/papers/commands/devadd_cmd.py

import os
import subprocess

from .. import repo
from .. import files
from ..paper import Paper, NoDocumentFile
from .. import configs

TMP_BIB_FILE = '/tmp/ref.bib'

def add_paper_with_docfile(repo, paper, docfile=None, copy=False):
    repo.add_paper(paper)
    if docfile is not None:
        if copy:
            repo.import_document(paper.citekey, docfile)
        else:
            paper.set_external_document(docfile)
            repo.add_or_update(paper)


def extract_doc_path_from_bibdata(paper, ui):
    try:
        file_path = paper.get_document_file_from_bibdata(remove=True)
        if files.check_file(file_path):
            return file_path
        else:
            ui.warning("File does not exist for %s (%s)."
                       % (paper.citekey, file_path))
    except NoDocumentFile:
        return None


def parser(subparsers, config):
    parser = subparsers.add_parser('devadd', help='devadd a paper to the repository')
    parser.add_argument('-b', '--bibfile', help='bibtex, bibtexml or bibyaml file', default=None)
    parser.add_argument('-d', '--docfile', help='pdf or ps file', default=None)
    parser.add_argument('-l', '--label', help='label associated to the paper', default=None)
    parser.add_argument('-c', '--copy', action='store_true', default=None,
            help="copy document files into library directory (default)")
    parser.add_argument('-C', '--nocopy', action='store_false', dest='copy',
            help="don't copy document files (opposite of -c)")
    return parser


def command(config, ui, bibfile, docfile, label, copy):
    """
    :param bibfile: bibtex file (in .bib, .bibml or .yaml format.
    :param docfile: path (no url yet) to a pdf or ps file
    """
    if copy is None:
        copy = config.get(configs.MAIN_SECTION, 'import-copy')
    rp = repo.Repository.from_directory(config)
    if bibfile is None:
        bibfile = fill_bib(config)
    p = Paper.load(bibfile)
    if label is not None:
        p.metadata['labels'] = label.split()
    # Check if another doc file is specified in bibtex
    docfile2 = extract_doc_path_from_bibdata(p, ui)
    if docfile is None:
        docfile = docfile2
    elif docfile2 is not None:
        ui.warning(
                "Skipping document file from bib file: %s, using %s instead."
                % (docfile2, docfile))
    add_paper_with_docfile(rp, p, docfile=docfile, copy=copy)

def fill_bib(config):
    if os.path.exists(TMP_BIB_FILE):
        os.remove(TMP_BIB_FILE)
    print "Fill the reference as a .bib and close the editor"
    proc = subprocess.Popen([config.get(configs.MAIN_SECTION, 'edit-cmd'), TMP_BIB_FILE])
    proc.wait()
    return TMP_BIB_FILE