def test_does_not_delete_non_temp_letter_file(mocker, fake_uuid): mocked_delete_s3_object = mocker.patch( 'app.s3_client.s3_logo_client.delete_s3_object') with pytest.raises(ValueError) as error: delete_letter_temp_file(svg_filename) mocked_delete_s3_object.assert_not_called assert str(error.value) == 'Not a temp file: {}'.format(svg_filename)
def test_delete_single_temp_letter_file(mocker, fake_uuid, upload_filename): mocked_delete_s3_object = mocker.patch( 'app.s3_client.s3_logo_client.delete_s3_object') upload_filename = LETTER_TEMP_TAG.format(user_id=fake_uuid) + svg_filename delete_letter_temp_file(upload_filename) mocked_delete_s3_object.assert_called_with(upload_filename)
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, ) upload_letter_logos(logo, db_filename, png_file, session["user_id"]) return redirect(url_for("main.letter_branding")) except HTTPError as e: if "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 update_letter_branding(branding_id, logo=None): letter_branding = letter_branding_client.get_letter_branding(branding_id) file_upload_form = SVGFileUpload() letter_branding_details_form = ServiceLetterBrandingDetails( name=letter_branding['name'], ) file_upload_form_submitted = file_upload_form.file.data details_form_submitted = request.form.get( 'operation') == 'branding-details' logo = logo if logo else permanent_letter_logo_name( letter_branding['filename'], 'svg') 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.startswith(LETTER_TEMP_TAG.format(user_id=session['user_id'])): delete_letter_temp_file(logo) return redirect( url_for('.update_letter_branding', branding_id=branding_id, logo=upload_filename)) if details_form_submitted and letter_branding_details_form.validate_on_submit( ): db_filename = letter_filename_for_db(logo, session['user_id']) try: if db_filename == letter_branding['filename']: letter_branding_client.update_letter_branding( branding_id=branding_id, filename=db_filename, name=letter_branding_details_form.name.data, ) return redirect(url_for('main.letter_branding')) else: letter_branding_client.update_letter_branding( branding_id=branding_id, filename=db_filename, name=letter_branding_details_form.name.data, ) upload_letter_svg_logo(logo, db_filename, session['user_id']) return redirect(url_for('main.letter_branding')) except HTTPError as e: if 'name' in e.message: letter_branding_details_form.name.errors.append( e.message['name'][0]) else: raise e except BotoClientError: # we had a problem saving the file - rollback the db changes letter_branding_client.update_letter_branding( branding_id=branding_id, filename=letter_branding['filename'], name=letter_branding['name'], ) file_upload_form.file.errors = [ 'Error saving uploaded file - try uploading again' ] 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, is_update=True)
def update_letter_branding(branding_id, logo=None): letter_branding = letter_branding_client.get_letter_branding(branding_id) file_upload_form = SVGFileUpload() letter_branding_details_form = ServiceLetterBrandingDetails( name=letter_branding["name"], ) file_upload_form_submitted = file_upload_form.file.data details_form_submitted = request.form.get( "operation") == "branding-details" logo = logo if logo else permanent_letter_logo_name( letter_branding["filename"], "svg") 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.startswith(LETTER_TEMP_TAG.format(user_id=session["user_id"])): delete_letter_temp_file(logo) return redirect( url_for(".update_letter_branding", branding_id=branding_id, logo=upload_filename)) if details_form_submitted and letter_branding_details_form.validate_on_submit( ): db_filename = letter_filename_for_db(logo, session["user_id"]) try: if db_filename == letter_branding["filename"]: letter_branding_client.update_letter_branding( branding_id=branding_id, filename=db_filename, name=letter_branding_details_form.name.data, ) return redirect(url_for("main.letter_branding")) else: png_file = get_png_file_from_svg(logo) letter_branding_client.update_letter_branding( branding_id=branding_id, filename=db_filename, name=letter_branding_details_form.name.data, ) upload_letter_logos(logo, db_filename, png_file, session["user_id"]) return redirect(url_for("main.letter_branding")) except HTTPError as e: if "name" in e.message: letter_branding_details_form.name.errors.append( e.message["name"][0]) else: raise e except BotoClientError: # we had a problem saving the file - rollback the db changes letter_branding_client.update_letter_branding( branding_id=branding_id, filename=letter_branding["filename"], name=letter_branding["name"], ) file_upload_form.file.errors = [ "Error saving uploaded file - try uploading again" ] 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, is_update=True, )