basic task order pdf downloads

This commit is contained in:
dandds
2018-09-05 17:45:12 -04:00
parent e8aa905a99
commit 0391348b5d
6 changed files with 81 additions and 5 deletions

View File

@@ -1,7 +1,9 @@
from flask import render_template, g
from flask import render_template, g, Response
from flask import current_app as app
from . import requests_bp
from atst.domain.requests import Requests
from atst.domain.exceptions import NotFoundError
def task_order_dictionary(task_order):
@@ -19,4 +21,15 @@ def approval(request_id):
if request.task_order:
data["task_order"] = task_order_dictionary(request.task_order)
return render_template("requests/approval.html", data=data, financial_review=True)
return render_template("requests/approval.html", data=data, request_id=request.id, financial_review=True)
@requests_bp.route("/requests/task_order_download/<string:request_id>", methods=["GET"])
def task_order_pdf_download(request_id):
request = Requests.get(g.current_user, request_id)
if request.task_order and request.task_order.pdf:
object_name = request.task_order.pdf.object_name
generator = app.uploader.download_stream(object_name)
return Response(generator, mimetype="application/pdf")
else:
raise NotFoundError("task_order pdf")