Update atst to atat

atat/forms/portfolio_member.py

@@ -0,0 +1,64 @@
+from wtforms.validators import Required
+from wtforms.fields import BooleanField, FormField
+
+from .forms import BaseForm
+from .member import NewForm as BaseNewMemberForm
+from atat.domain.permission_sets import PermissionSets
+from atat.forms.fields import SelectField
+from atat.utils.localization import translate
+
+
+class PermissionsForm(BaseForm):
+    perms_app_mgmt = BooleanField(
+        translate("forms.new_member.app_mgmt.label"),
+        default=False,
+        description=translate("forms.new_member.app_mgmt.description"),
+    )
+    perms_funding = BooleanField(
+        translate("forms.new_member.funding.label"),
+        default=False,
+        description=translate("forms.new_member.funding.description"),
+    )
+    perms_reporting = BooleanField(
+        translate("forms.new_member.reporting.label"),
+        default=False,
+        description=translate("forms.new_member.reporting.description"),
+    )
+    perms_portfolio_mgmt = BooleanField(
+        translate("forms.new_member.portfolio_mgmt.label"),
+        default=False,
+        description=translate("forms.new_member.portfolio_mgmt.description"),
+    )
+
+    @property
+    def data(self):
+        _data = super().data
+        _data.pop("csrf_token", None)
+        perm_sets = []
+
+        if _data["perms_app_mgmt"]:
+            perm_sets.append(PermissionSets.EDIT_PORTFOLIO_APPLICATION_MANAGEMENT)
+
+        if _data["perms_funding"]:
+            perm_sets.append(PermissionSets.EDIT_PORTFOLIO_FUNDING)
+
+        if _data["perms_reporting"]:
+            perm_sets.append(PermissionSets.EDIT_PORTFOLIO_REPORTS)
+
+        if _data["perms_portfolio_mgmt"]:
+            perm_sets.append(PermissionSets.EDIT_PORTFOLIO_ADMIN)
+
+        _data["permission_sets"] = perm_sets
+        return _data
+
+
+class NewForm(PermissionsForm):
+    user_data = FormField(BaseNewMemberForm)
+
+
+class AssignPPOCForm(PermissionsForm):
+    role_id = SelectField(
+        label=translate("forms.assign_ppoc.dod_id"),
+        validators=[Required()],
+        choices=[("", "- Select -")],
+    )