mirror of https://github.com/logos-co/open-law.git
38 lines
986 B
Python
38 lines
986 B
Python
|
from flask_wtf import FlaskForm
|
||
|
|
from wtforms import StringField, SubmitField, SelectField
|
||
|
|
from wtforms.validators import DataRequired
|
||
|
|
|
||
|
|
from app import models as m
|
||
|
|
|
||
|
|
|
||
|
|
class AddContributorForm(FlaskForm):
|
||
|
|
user_id = StringField("User ID", [DataRequired()])
|
||
|
|
role = SelectField(
|
||
|
|
"Role",
|
||
|
|
choices=[
|
||
|
|
(member.value, name.capitalize())
|
||
|
|
for name, member in m.BookContributor.Roles.__members__.items()
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
submit = SubmitField("Add Contributor")
|
||
|
|
|
||
|
|
|
||
|
|
class DeleteContributorForm(FlaskForm):
|
||
|
|
user_id = StringField("User ID", [DataRequired()])
|
||
|
|
|
||
|
|
submit = SubmitField("Delete Contributor")
|
||
|
|
|
||
|
|
|
||
|
|
class EditContributorRoleForm(FlaskForm):
|
||
|
|
user_id = StringField("User ID", [DataRequired()])
|
||
|
|
role = SelectField(
|
||
|
|
"Role",
|
||
|
|
choices=[
|
||
|
|
(member.value, name.capitalize())
|
||
|
|
for name, member in m.BookContributor.Roles.__members__.items()
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
submit = SubmitField("Edit Contributor")
|