from flask import current_app, redirect, render_template, session, url_for from flask_login import current_user from notifications_python_client.errors import HTTPError from app import service_api_client from app.formatters import email_safe from app.main import main from app.main.forms import CreateServiceForm from app.utils.user import user_is_gov_user, user_is_logged_in def _create_service(service_name, organization_type, email_from, form): try: service_id = service_api_client.create_service( service_name=service_name, organization_type=organization_type, message_limit=current_app.config["DEFAULT_SERVICE_LIMIT"], restricted=True, user_id=session["user_id"], email_from=email_from, ) session["service_id"] = service_id return service_id, None except HTTPError as e: if e.status_code == 400 and e.message["name"]: form.name.errors.append("This service name is already in use") return None, e else: raise e def _create_example_template(service_id): example_sms_template = service_api_client.create_service_template( "Example text message template", "sms", "Hi, I’m trying out Notify.gov. Today is ((day of week)) and my favorite color is ((color)).", service_id, ) return example_sms_template @main.route("/add-service", methods=["GET", "POST"]) @user_is_gov_user @user_is_logged_in def add_service(): default_organization_type = current_user.default_organization_type if default_organization_type is None: default_organization_type = "federal" form = CreateServiceForm( # This value is currently not useful but if it is not set it will result in a bug organization_type=default_organization_type ) if form.validate_on_submit(): email_from = email_safe(form.name.data) service_name = form.name.data service_id, error = _create_service( service_name, default_organization_type or form.organization_type.data, email_from, form, ) if error: return _render_add_service_page(form, default_organization_type) if ( len( service_api_client.get_active_services( {"user_id": session["user_id"]} ).get("data", []) ) > 1 ): return redirect(url_for("main.service_dashboard", service_id=service_id)) example_sms_template = _create_example_template(service_id) return redirect( url_for( "main.begin_tour", service_id=service_id, template_id=example_sms_template["data"]["id"], ) ) else: return _render_add_service_page(form, default_organization_type) def _render_add_service_page(form, default_organization_type): heading = "About your service" if default_organization_type == "local": return render_template( "views/add-service-local.html", form=form, heading=heading, default_organization_type=default_organization_type, ) return render_template( "views/add-service.html", form=form, heading=heading, default_organization_type=default_organization_type, )