bibtex.py

import os, sys, yaml, jinja2, utils

def parser(cmd_name, subparsers, config):
    parser = subparsers.add_parser(
        cmd_name, help="generate BibTeX from a paper's metadata")
    parser.add_argument(
        "paper_folder_name",
        help="the target paper's folder within your paper pile")

    return parser

def command(config, paper_folder_name):
    metadata_path = utils.paper_metadata_path(paper_folder_name, config)

    if metadata_path is None:
        return 1

    with open(metadata_path, 'r') as fp:
        metadata_file_contents = fp.read()

        try:
            metadata = yaml.load(metadata_file_contents)
        except yaml.scanner.ScannerError as e:
            print("Error loading metadata: %s" % (e))
            return 1

    for key in metadata:
        if metadata[key] is None:
            metadata[key] = ""

    jinja_env = jinja2.Environment(
        loader = jinja2.FileSystemLoader(os.path.abspath(
                os.path.join(os.path.dirname(__file__), 'templates'))),
        trim_blocks = True,
        variable_start_string="<<",
        variable_end_string=">>")

    bibtex_template = jinja_env.get_template("bibtex.jinja2")

    print(bibtex_template.render(
        paper_name = paper_folder_name,
        **metadata))

    return 0