43ea922218
Use ApplicationRole.id instead of User.id in forms. This eliminates the need for the function that checks whether a user is in a given application, because looking up the application role will raise an error if the user is not.
33 lines
872 B
Python
33 lines
872 B
Python
from flask_wtf import FlaskForm
|
|
from wtforms.fields import FieldList, FormField, HiddenField, RadioField, StringField
|
|
|
|
from .forms import BaseForm
|
|
from .data import ENV_ROLES, ENV_ROLE_NO_ACCESS as NO_ACCESS
|
|
|
|
|
|
class MemberForm(FlaskForm):
|
|
application_role_id = HiddenField()
|
|
user_name = StringField()
|
|
role_name = RadioField(choices=ENV_ROLES, default=NO_ACCESS)
|
|
|
|
@property
|
|
def data(self):
|
|
_data = super().data
|
|
if "role_name" in _data and _data["role_name"] == NO_ACCESS:
|
|
_data["role_name"] = None
|
|
return _data
|
|
|
|
|
|
class RoleForm(FlaskForm):
|
|
role = HiddenField()
|
|
members = FieldList(FormField(MemberForm))
|
|
|
|
|
|
class EnvironmentRolesForm(FlaskForm):
|
|
team_roles = FieldList(FormField(RoleForm))
|
|
env_id = HiddenField()
|
|
|
|
|
|
class AppEnvRolesForm(BaseForm):
|
|
envs = FieldList(FormField(EnvironmentRolesForm))
|