65 lines
1.3 KiB
Python
Executable File
65 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import config
|
|
from libbmc.citations import pdf
|
|
import os
|
|
import tempfile
|
|
|
|
from bottle import redirect, request, route, run, view
|
|
|
|
|
|
@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)
|
|
citations = pdf.cermine_dois(fh.name,
|
|
override_local=config.CERMINE_PATH)
|
|
|
|
return {
|
|
"params": {
|
|
"citations": citations
|
|
}
|
|
}
|
|
|
|
|
|
@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)
|