Update atst to atat

atat/routes/applications/new.py

@@ -0,0 +1,167 @@
+from flask import redirect, render_template, request as http_request, url_for
+
+from .blueprint import applications_bp
+from atat.domain.applications import Applications
+from atat.forms.application import NameAndDescriptionForm, EnvironmentsForm
+from atat.domain.authz.decorator import user_can_access_decorator as user_can
+from atat.models.permissions import Permissions
+from atat.utils.flash import formatted_flash as flash
+from atat.routes.applications.settings import (
+    get_members_data,
+    get_new_member_form,
+    handle_create_member,
+    handle_update_member,
+    handle_update_application,
+)
+
+
+def get_new_application_form(form_data, form_class, application_id=None):
+    if application_id:
+        application = Applications.get(application_id)
+        return form_class(form_data, obj=application)
+    else:
+        return form_class(form_data)
+
+
+def render_new_application_form(
+    template, form_class, portfolio_id=None, application_id=None, form=None
+):
+    render_args = {"application_id": application_id}
+    if application_id:
+        application = Applications.get(application_id)
+        render_args["form"] = form or form_class(obj=application)
+        render_args["application"] = application
+    else:
+        render_args["form"] = form or form_class()
+
+    return render_template(template, **render_args)
+
+
+@applications_bp.route("/portfolios/<portfolio_id>/applications/new")
+@applications_bp.route("/applications/<application_id>/new/step_1")
+@user_can(Permissions.CREATE_APPLICATION, message="view create new application form")
+def view_new_application_step_1(portfolio_id=None, application_id=None):
+    return render_new_application_form(
+        "applications/new/step_1.html",
+        NameAndDescriptionForm,
+        portfolio_id=portfolio_id,
+        application_id=application_id,
+    )
+
+
+@applications_bp.route(
+    "/portfolios/<portfolio_id>/applications/new",
+    endpoint="create_new_application_step_1",
+    methods=["POST"],
+)
+@applications_bp.route(
+    "/applications/<application_id>/new/step_1",
+    endpoint="update_new_application_step_1",
+    methods=["POST"],
+)
+@user_can(Permissions.CREATE_APPLICATION, message="view create new application form")
+def create_or_update_new_application_step_1(portfolio_id=None, application_id=None):
+    form = get_new_application_form(
+        {**http_request.form}, NameAndDescriptionForm, application_id
+    )
+    application = handle_update_application(form, application_id, portfolio_id)
+
+    if application:
+        return redirect(
+            url_for(
+                "applications.update_new_application_step_2",
+                application_id=application.id,
+            )
+        )
+    else:
+        return (
+            render_new_application_form(
+                "applications/new/step_1.html",
+                NameAndDescriptionForm,
+                portfolio_id,
+                application_id,
+                form,
+            ),
+            400,
+        )
+
+
+@applications_bp.route("/applications/<application_id>/new/step_2")
+@user_can(Permissions.CREATE_APPLICATION, message="view create new application form")
+def view_new_application_step_2(application_id):
+    application = Applications.get(application_id)
+    render_args = {
+        "form": EnvironmentsForm(
+            data={
+                "environment_names": [
+                    environment.name for environment in application.environments
+                ]
+            }
+        ),
+        "application": application,
+    }
+
+    return render_template("applications/new/step_2.html", **render_args)
+
+
+@applications_bp.route("/applications/<application_id>/new/step_2", methods=["POST"])
+@user_can(Permissions.CREATE_APPLICATION, message="view create new application form")
+def update_new_application_step_2(application_id):
+    form = get_new_application_form(
+        {**http_request.form}, EnvironmentsForm, application_id
+    )
+    if form.validate():
+        application = Applications.get(application_id)
+        application = Applications.update(application, form.data)
+        flash("application_environments_updated")
+        return redirect(
+            url_for(
+                "applications.update_new_application_step_3",
+                application_id=application_id,
+            )
+        )
+    else:
+        return (
+            render_new_application_form(
+                "applications/new/step_2.html",
+                EnvironmentsForm,
+                application_id=application_id,
+                form=form,
+            ),
+            400,
+        )
+
+
+@applications_bp.route("/applications/<application_id>/new/step_3")
+@user_can(Permissions.CREATE_APPLICATION, message="view create new application form")
+def view_new_application_step_3(application_id):
+    application = Applications.get(application_id)
+    members = get_members_data(application)
+    new_member_form = get_new_member_form(application)
+
+    return render_template(
+        "applications/new/step_3.html",
+        application_id=application_id,
+        application=application,
+        members=members,
+        new_member_form=new_member_form,
+    )
+
+
+@applications_bp.route("/applications/<application_id>/new/step_3", methods=["POST"])
+@applications_bp.route(
+    "/applications/<application_id>/new/step_3/member/<application_role_id>",
+    methods=["POST"],
+)
+@user_can(Permissions.CREATE_APPLICATION, message="view create new application form")
+def update_new_application_step_3(application_id, application_role_id=None):
+    if application_role_id:
+        handle_update_member(application_id, application_role_id, http_request.form)
+    else:
+        handle_create_member(application_id, http_request.form)
+
+    return redirect(
+        url_for(
+            "applications.view_new_application_step_3", application_id=application_id
+        )
+    )