80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
from flask import g, render_template, redirect, url_for
|
|
from flask import request as http_request
|
|
|
|
from . import requests_bp
|
|
from atst.domain.requests import Requests
|
|
from atst.forms.financial import FinancialForm, ExtendedFinancialForm
|
|
|
|
|
|
def task_order_data(task_order):
|
|
data = task_order.to_dictionary()
|
|
data["task_order_number"] = task_order.number
|
|
data["funding_type"] = task_order.funding_type.value
|
|
return data
|
|
|
|
|
|
def financial_form(data):
|
|
if http_request.args.get("extended"):
|
|
return ExtendedFinancialForm(data=data)
|
|
else:
|
|
return FinancialForm(data=data)
|
|
|
|
|
|
@requests_bp.route("/requests/verify/<string:request_id>", methods=["GET"])
|
|
def financial_verification(request_id=None):
|
|
request = Requests.get(g.current_user, request_id)
|
|
form_data = request.body.get("financial_verification")
|
|
if request.task_order:
|
|
form_data.update(task_order_data(request.task_order))
|
|
|
|
form = financial_form(form_data)
|
|
return render_template(
|
|
"requests/financial_verification.html",
|
|
f=form,
|
|
request_id=request_id,
|
|
extended=http_request.args.get("extended"),
|
|
)
|
|
|
|
|
|
@requests_bp.route("/requests/verify/<string:request_id>", methods=["POST"])
|
|
def update_financial_verification(request_id):
|
|
post_data = http_request.form
|
|
existing_request = Requests.get(g.current_user, request_id)
|
|
form = financial_form(post_data)
|
|
rerender_args = dict(
|
|
request_id=request_id, f=form, extended=http_request.args.get("extended")
|
|
)
|
|
|
|
if form.validate():
|
|
valid = form.perform_extra_validation(
|
|
existing_request.body.get("financial_verification")
|
|
)
|
|
updated_request = Requests.update_financial_verification(request_id, form.data)
|
|
if valid:
|
|
submitted_request = Requests.submit_financial_verification(updated_request)
|
|
if submitted_request.is_financially_verified:
|
|
new_workspace = Requests.approve_and_create_workspace(submitted_request)
|
|
return redirect(
|
|
url_for(
|
|
"workspaces.new_project",
|
|
workspace_id=new_workspace.id,
|
|
newWorkspace=True,
|
|
)
|
|
)
|
|
else:
|
|
return redirect(
|
|
url_for(
|
|
"requests.requests_index", pendingFinancialVerification=True
|
|
)
|
|
)
|
|
|
|
else:
|
|
form.reset()
|
|
return render_template(
|
|
"requests/financial_verification.html", **rerender_args
|
|
)
|
|
|
|
else:
|
|
form.reset()
|
|
return render_template("requests/financial_verification.html", **rerender_args)
|