def get_homepage(request, draft_id, obj_pk=None):
data = {"pk": draft_id}
if obj_pk:
data["obj_pk"] = obj_pk
try:
url = reverse(document_switch(request.path)["homepage"], kwargs=data)
except NoReverseMatch:
url = reverse(document_switch(request.path)["homepage"], kwargs={"pk": draft_id})
return redirect(url)
def post(self, request, **kwargs):
draft_id = str(kwargs["pk"])
option = request.POST.get("delete_document_confirmation")
if option is None:
return form_page(
request,
get_delete_confirmation_page(request.path, str(kwargs["pk"])),
errors={
"delete_document_confirmation":
["Select yes to confirm you want to delete the document"]
},
)
else:
if option == "yes":
action = document_switch(request.path)["delete"]
if len(signature(action).parameters) == 2:
status_code = action(request, draft_id)
else:
status_code = action(request, draft_id, kwargs["obj_pk"])
if status_code == HTTPStatus.NO_CONTENT:
return get_homepage(request, draft_id)
else:
return error_page(
request, strings.applications.DeleteDocument.
DOCUMENT_DELETE_GENERIC_ERROR)
else:
return get_homepage(request, draft_id)
def get_upload_page(path, draft_id, is_permanent_application=False):
paths = document_switch(path=path)
is_document_optional = paths["optional"]
# For standard permanent only - upload is mandatory
if "/end-user" in path and is_permanent_application:
is_document_optional = False
return attach_document_form(
application_id=draft_id, strings=paths["strings"], back_link=paths["homepage"], is_optional=is_document_optional
)
def get(self, request, **kwargs):
draft_id = str(kwargs["pk"])
action = document_switch(request.path)["download"]
if len(signature(action).parameters) == 2:
document, _ = action(request, draft_id)
else:
document, _ = action(request, draft_id, kwargs["obj_pk"])
document = document["document"]
if document["safe"]:
return download_document_from_s3(document["s3_key"], document["name"])
else:
return error_page(request, strings.applications.AttachDocumentPage.DOWNLOAD_GENERIC_ERROR)
def post(self, request, **kwargs):
draft_id = str(kwargs["pk"])
application = get_application(request, draft_id)
is_permanent_application = is_application_export_type_permanent(
application)
form = get_upload_page(
request.path,
draft_id,
is_permanent_application=is_permanent_application)
try:
files = request.FILES
except Exception: # noqa
return error_page(
request,
strings.applications.AttachDocumentPage.UPLOAD_FAILURE_ERROR)
# Only validate documents if there are any present or are mandatory in the following cases:
# standard permanent application end user section, additional documents section
if (files or ("/end-user" in request.path
and is_application_export_type_permanent(application))
or "additional-document" in request.path):
logging.info(self.request)
data, error = add_document_data(request)
if error:
return form_page(request,
form,
extra_data={"draft_id": draft_id},
errors={"documents": [error]})
action = document_switch(request.path)["attach"]
if len(signature(action).parameters) == 3:
_, status_code = action(request, draft_id, data)
if status_code == HTTPStatus.CREATED:
return get_homepage(request, draft_id)
else:
_, status_code = action(request, draft_id, kwargs["obj_pk"],
data)
if status_code == HTTPStatus.CREATED:
return get_homepage(request, draft_id, kwargs["obj_pk"])
return error_page(
request,
strings.applications.AttachDocumentPage.UPLOAD_FAILURE_ERROR)
return get_homepage(request, draft_id)
def get_delete_confirmation_page(path, pk):
paths = document_switch(path)
return delete_document_confirmation_form(
overview_url=reverse(paths["homepage"], kwargs={"pk": pk}), strings=paths["strings"],
)