From 54efba8e3b8fb20e506a31b3b9e452e1eccffe34 Mon Sep 17 00:00:00 2001 From: Robin Krahl Date: Sat, 24 Mar 2018 17:22:29 +0100 Subject: Add --doi option for explicit DOIs --- bibtool/extract.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'bibtool/extract.py') diff --git a/bibtool/extract.py b/bibtool/extract.py index 6955dbe..cefbe8f 100644 --- a/bibtool/extract.py +++ b/bibtool/extract.py @@ -42,8 +42,9 @@ def _get_data_for_doi(doi): return bibtexparser.loads(bibtex) -def _handle_pdf(filename): - doi = _extract_doi(filename) +def _handle_pdf(filename, doi): + if not doi: + doi = _extract_doi(filename) if not doi: raise Exception('Could not extract a DOI from the PDF file') @@ -51,8 +52,8 @@ def _handle_pdf(filename): return _get_data_for_doi(doi) -def get_bibtex_data(filename): +def get_bibtex_data(filename, doi): mtype, _ = mimetypes.guess_type(filename) if mtype == 'application/pdf': - return _handle_pdf(filename) + return _handle_pdf(filename, doi) raise Exception('Unsupported file type ' + str(mtype)) -- cgit v1.2.1