mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 02:42:26 -05:00
de6ed99006
When the template content was renamed in
9ee8610da0 I missed doing the same change for the
delete template route.
This commit does the same fix, so that template content is still visible when
you’re about to delete a template (so you can make sure it’s the right one).
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
from flask import request, render_template, redirect, url_for, flash, abort
|
||
from flask_login import login_required
|
||
|
||
from app.main import main
|
||
from app.main.forms import TemplateForm
|
||
from app import job_api_client
|
||
from app.main.dao.services_dao import get_service_by_id
|
||
from app.main.dao import templates_dao as tdao
|
||
from app.main.dao import services_dao as sdao
|
||
from client.errors import HTTPError
|
||
|
||
|
||
@main.route("/services/<service_id>/templates")
|
||
@login_required
|
||
def manage_service_templates(service_id):
|
||
try:
|
||
jobs = job_api_client.get_job(service_id)['data']
|
||
templates = tdao.get_service_templates(service_id)['data']
|
||
except HTTPError as e:
|
||
if e.status_code == 404:
|
||
abort(404)
|
||
else:
|
||
raise e
|
||
return render_template(
|
||
'views/manage-templates.html',
|
||
service_id=service_id,
|
||
has_jobs=bool(jobs),
|
||
templates=[tdao.TemplatesBrowsableItem(x) for x in templates])
|
||
|
||
|
||
@main.route("/services/<service_id>/templates/add", methods=['GET', 'POST'])
|
||
@login_required
|
||
def add_service_template(service_id):
|
||
try:
|
||
service = sdao.get_service_by_id(service_id)['data']
|
||
except HTTPError as e:
|
||
if e.status_code == 404:
|
||
abort(404)
|
||
else:
|
||
raise e
|
||
|
||
form = TemplateForm()
|
||
|
||
if form.validate_on_submit():
|
||
tdao.insert_service_template(
|
||
form.name.data, form.template_type.data, form.template_content.data, service_id)
|
||
return redirect(url_for(
|
||
'.manage_service_templates', service_id=service_id))
|
||
return render_template(
|
||
'views/edit-template.html',
|
||
h1='Add template',
|
||
form=form,
|
||
service_id=service_id)
|
||
|
||
|
||
@main.route("/services/<service_id>/templates/<int:template_id>", methods=['GET', 'POST'])
|
||
@login_required
|
||
def edit_service_template(service_id, template_id):
|
||
try:
|
||
template = tdao.get_service_template(service_id, template_id)['data']
|
||
except HTTPError as e:
|
||
if e.status_code == 404:
|
||
abort(404)
|
||
else:
|
||
raise e
|
||
|
||
template['template_content'] = template['content']
|
||
form = TemplateForm(**template)
|
||
|
||
if form.validate_on_submit():
|
||
tdao.update_service_template(
|
||
template_id, form.name.data, form.template_type.data,
|
||
form.template_content.data, service_id)
|
||
return redirect(url_for('.manage_service_templates', service_id=service_id))
|
||
|
||
return render_template(
|
||
'views/edit-template.html',
|
||
h1='Edit template',
|
||
form=form,
|
||
service_id=service_id,
|
||
template_id=template_id)
|
||
|
||
|
||
@main.route("/services/<service_id>/templates/<int:template_id>/delete", methods=['GET', 'POST'])
|
||
@login_required
|
||
def delete_service_template(service_id, template_id):
|
||
try:
|
||
template = tdao.get_service_template(service_id, template_id)['data']
|
||
except HTTPError as e:
|
||
if e.status_code == 404:
|
||
abort(404)
|
||
else:
|
||
raise e
|
||
|
||
template['template_content'] = template['content']
|
||
form = TemplateForm(**template)
|
||
|
||
if request.method == 'POST':
|
||
tdao.delete_service_template(service_id, template_id)
|
||
return redirect(url_for('.manage_service_templates', service_id=service_id))
|
||
|
||
flash('Are you sure you want to delete ‘{}’?'.format(form.name.data), 'delete')
|
||
return render_template(
|
||
'views/edit-template.html',
|
||
h1='Edit template',
|
||
form=form,
|
||
service_id=service_id,
|
||
template_id=template_id)
|