Fetch reference citations in BibTeX, JSON or YAML from DOI

doi2bib

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import subprocess
import tempfile
import requests
import argparse
import textwrap
import sys


def main(doi, format):
    """
    Fetch a citation from a DOI using the doi.org service.
    Prints either BibTeX, JSON or YAML to stdout.

    Requires
    --------
    * requests package for Python
    * pandoc, pandoc-citeproc for JSON or YAML output

    """
    if args['doi'].startswith('http://'):
        url = args['doi']
    elif args['doi'].startswith('dx.doi.org') or args['doi'].startswith('doi.org'):
        url = 'http://' + args['doi']
    else:
        url = 'http://doi.org/' + args['doi']

    with tempfile.NamedTemporaryFile(suffix='.bib') as f:

        # Fetch BibTeX from dx.doi.org
        r = requests.get(
            url,
            headers={'Accept': 'text/bibliography; style=bibtex'})
        r.raise_for_status()
        f.write(r.content)
        f.seek(0)

        # Convert to JSON or YAML
        if args['format'] in ('json', 'yaml'):
            cmd = ['pandoc-citeproc', '--bib2' + args['format'], f.name]
            subprocess.run(cmd, stdout=sys.stdout)
        else:
            stdout = getattr(sys.stdout, 'buffer', sys.stdout)
            stdout.write(f.read())


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        description=textwrap.dedent(main.__doc__),
        formatter_class=argparse.RawTextHelpFormatter)
    parser.add_argument(
        'doi',
        type=str,
        help="DOI or http://doi.org URL")
    parser.add_argument(
        '--format', '-f',
        default='bib', 
        choices=['bib', 'json', 'yaml'],
        help="Output format (default: bib)")
    args = vars(parser.parse_args())
    main(**args)