citationextractor/main.py

#!/usr/bin/env python3
import config
import html
import os
import tempfile

from bottle import redirect, request, route, run, view
import libbmc.doi as doi
import libbmc.citations.pdf as pdf


def format_citation(doi_url):
    if doi_url is None:
        return {
            "doi": None,
            "oa": None,
            "sharable": None
        }

    canonical_doi = doi.to_canonical(doi_url)
    sharable = doi.get_oa_policy(canonical_doi)
    if sharable is not None:
        sharable = "can" in sharable.values()

    return {
        "doi": canonical_doi,
        "oa": doi.get_oa_version(canonical_doi),
        "sharable": sharable
    }


@route("/upload", method="POST")
@view("template.tpl")
def do_upload():
    """
    Handle uploaded PDF files and process them for citations.
    """
    # Check uploaded file has been provided
    upload = request.files.get('upload')
    if upload is None:
        return redirect("/")

    # Check file is a PDF file
    _, ext = os.path.splitext(upload.filename)
    if ext not in (".pdf"):
        return {
            "params": {
                "error": "File does not seem to be a valid PDF file."
            }
        }

    # Process citations
    with tempfile.NamedTemporaryFile() as fh:
        upload.save(fh)
        raw_citations = pdf.cermine_dois(fh.name,
                                         override_local=config.CERMINE_PATH)
    citations = {
        html.unescape(k): format_citation(v)
        for k, v in raw_citations.items()
    }

    return {
        "params": {
            "citations": citations,
            "upload_name": upload.filename
        }
    }


@route("/<:re:.+>")
def redirect_catchall():
    """
    Redirect everything else to /.
    """
    return redirect("/")


@route("/")
@view("template.tpl")
def index():
    """
    Main index view, upload form.
    """
    return {
        "params": {}
    }


if __name__ == "__main__":
    run(host=config.HOST, port=config.PORT, debug=config.DEBUG)