stash

Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>

lg/forms.py

@@ -1,6 +1,21 @@
 from django import forms
+from django.utils.translation import gettext_lazy as _
 
-from .models import Action, CupidonAction, DoveAction, HackerAction, RavenAction, WerewolfAction, WitchAction
+from .models import Action, Cupidon, CupidonAction, DiscordUser, Dove, DoveAction, Hacker, HackerAction, \
+    Raven, RavenAction, Werewolf, WerewolfAction, Witch, WitchAction
+
+
+class PlayerForm(forms.Form):
+    user = forms.ModelChoiceField(
+        queryset=DiscordUser.objects.filter(is_superuser=False, player__isnull=True),
+        label=lambda: _("user").capitalize(),
+    )
+
+    player_type = forms.ChoiceField(
+        choices=[(model.__name__, model._meta.verbose_name)
+                 for model in [Werewolf, Cupidon, Witch, Raven, Dove, Hacker]],
+        label=lambda: _("player type").capitalize(),
+    )
 
 
 class ActionForm(forms.ModelForm):