Skip to content

Commit 93ee73a

Browse files
committed
Break management forms into mixins
1 parent 79a6e45 commit 93ee73a

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

warehouse/manage/forms.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,7 @@
1515
from warehouse import forms
1616

1717

18-
class CreateRoleForm(forms.Form):
19-
username = wtforms.StringField(
20-
validators=[
21-
wtforms.validators.DataRequired(message="Must specify a username"),
22-
]
23-
)
18+
class RoleNameMixin:
2419

2520
role_name = wtforms.SelectField(
2621
'Select a role',
@@ -33,6 +28,15 @@ class CreateRoleForm(forms.Form):
3328
]
3429
)
3530

31+
32+
class UsernameMixin:
33+
34+
username = wtforms.StringField(
35+
validators=[
36+
wtforms.validators.DataRequired(message="Must specify a username"),
37+
]
38+
)
39+
3640
def validate_username(self, field):
3741
userid = self.user_service.find_userid(field.data)
3842

@@ -41,6 +45,9 @@ def validate_username(self, field):
4145
"No user found with that username. Please try again."
4246
)
4347

48+
49+
class CreateRoleForm(RoleNameMixin, UsernameMixin, forms.Form):
50+
4451
def __init__(self, *args, user_service, **kwargs):
4552
super().__init__(*args, **kwargs)
4653
self.user_service = user_service

0 commit comments

Comments
 (0)