notifications-admin/app/main/views/add_service.py

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,
    )