mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-19 05:59:44 -04:00
f78b3bff81
This is a platform admin page which lists all letter brands and their domain. Once the page to edit a letter brand has been created, each brand will link to its edit page, but for now this is just a list of brand names and domains.
134 lines
4.3 KiB
Python
134 lines
4.3 KiB
Python
from flask import (
|
|
current_app,
|
|
redirect,
|
|
render_template,
|
|
request,
|
|
session,
|
|
url_for,
|
|
)
|
|
from flask_login import login_required
|
|
from notifications_python_client.errors import HTTPError
|
|
from requests import get as requests_get
|
|
|
|
from app import letter_branding_client
|
|
from app.main import main
|
|
from app.main.forms import (
|
|
SearchByNameForm,
|
|
ServiceLetterBrandingDetails,
|
|
SVGFileUpload,
|
|
)
|
|
from app.s3_client.s3_logo_client import (
|
|
LETTER_TEMP_TAG,
|
|
delete_letter_temp_file,
|
|
delete_letter_temp_files_created_by,
|
|
get_letter_filename_with_no_path_or_extension,
|
|
letter_filename_for_db,
|
|
permanent_letter_logo_name,
|
|
persist_logo,
|
|
upload_letter_png_logo,
|
|
upload_letter_temp_logo,
|
|
)
|
|
from app.utils import get_logo_cdn_domain, user_is_platform_admin
|
|
|
|
|
|
@main.route("/letter-branding", methods=['GET'])
|
|
@login_required
|
|
@user_is_platform_admin
|
|
def letter_branding():
|
|
|
|
brandings = letter_branding_client.get_all_letter_branding()
|
|
|
|
return render_template(
|
|
'views/letter-branding/select-letter-branding.html',
|
|
letter_brandings=brandings,
|
|
search_form=SearchByNameForm()
|
|
)
|
|
|
|
|
|
@main.route("/letter-branding/create", methods=['GET', 'POST'])
|
|
@main.route("/letter-branding/create/<path:logo>", methods=['GET', 'POST'])
|
|
@login_required
|
|
@user_is_platform_admin
|
|
def create_letter_branding(logo=None):
|
|
file_upload_form = SVGFileUpload()
|
|
letter_branding_details_form = ServiceLetterBrandingDetails()
|
|
|
|
file_upload_form_submitted = file_upload_form.file.data
|
|
details_form_submitted = request.form.get('operation') == 'branding-details'
|
|
|
|
if file_upload_form_submitted and file_upload_form.validate_on_submit():
|
|
upload_filename = upload_letter_temp_logo(
|
|
file_upload_form.file.data.filename,
|
|
file_upload_form.file.data,
|
|
current_app.config['AWS_REGION'],
|
|
user_id=session["user_id"]
|
|
)
|
|
|
|
if logo and logo.startswith(LETTER_TEMP_TAG.format(user_id=session['user_id'])):
|
|
delete_letter_temp_file(logo)
|
|
|
|
return redirect(url_for('.create_letter_branding', logo=upload_filename))
|
|
|
|
if details_form_submitted and letter_branding_details_form.validate_on_submit():
|
|
if logo:
|
|
db_filename = letter_filename_for_db(logo, session['user_id'])
|
|
png_file = get_png_file_from_svg(logo)
|
|
|
|
try:
|
|
letter_branding_client.create_letter_branding(
|
|
filename=db_filename,
|
|
name=letter_branding_details_form.name.data,
|
|
domain=letter_branding_details_form.domain.data,
|
|
)
|
|
|
|
upload_letter_logos(logo, db_filename, png_file, session['user_id'])
|
|
|
|
return redirect(url_for('main.letter_branding'))
|
|
|
|
except HTTPError as e:
|
|
if 'domain' in e.message:
|
|
letter_branding_details_form.domain.errors.append(e.message['domain'][0])
|
|
elif 'name' in e.message:
|
|
letter_branding_details_form.name.errors.append(e.message['name'][0])
|
|
else:
|
|
raise e
|
|
else:
|
|
# Show error on upload form if trying to submit with no logo
|
|
file_upload_form.validate()
|
|
|
|
return render_template(
|
|
'views/letter-branding/manage-letter-branding.html',
|
|
file_upload_form=file_upload_form,
|
|
letter_branding_details_form=letter_branding_details_form,
|
|
cdn_url=get_logo_cdn_domain(),
|
|
logo=logo
|
|
)
|
|
|
|
|
|
def get_png_file_from_svg(filename):
|
|
filename_for_template_preview = get_letter_filename_with_no_path_or_extension(filename)
|
|
|
|
template_preview_svg_endpoint = '{}/{}.svg.png'.format(
|
|
current_app.config['TEMPLATE_PREVIEW_API_HOST'],
|
|
filename_for_template_preview
|
|
)
|
|
|
|
response = requests_get(
|
|
template_preview_svg_endpoint,
|
|
headers={'Authorization': 'Token {}'.format(current_app.config['TEMPLATE_PREVIEW_API_KEY'])}
|
|
)
|
|
|
|
return response.content
|
|
|
|
|
|
def upload_letter_logos(old_filename, new_filename, png_file, user_id):
|
|
persist_logo(old_filename, permanent_letter_logo_name(new_filename, 'svg'))
|
|
|
|
upload_letter_png_logo(
|
|
permanent_letter_logo_name(new_filename, 'png'),
|
|
png_file,
|
|
current_app.config['AWS_REGION'],
|
|
)
|
|
|
|
delete_letter_temp_files_created_by(user_id)
|