Custom choice fields for epic and assignees

main/forms.py

@@ -5,10 +5,11 @@
 from django import forms
 from django.contrib.auth.forms import UserCreationForm
 from django.contrib.auth.models import User
+from django.forms.fields import DateField, ChoiceField, MultipleChoiceField
 from martor.fields import MartorFormField
 
 from main.models import Story, Epic, Comment, Sprint
-from django.forms.fields import DateField
+
 
 class RegisterForm(UserCreationForm):
                             
@@ -44,7 +45,9 @@ class EpicForm(forms.ModelForm):
 class StoryForm(forms.ModelForm):
     
     description = MartorFormField(label="Description")
- 
+    epic = ChoiceField(choices=[(epic.id, epic.name) for epic in Epic.objects.filter(closed=False).order_by('-value')])
+    assignees = MultipleChoiceField(choices=[(user.id, user.get_full_name() or user.username) for user in User.objects.all()])
+    
     class Meta:
         model = Story
         widgets = {'author': forms.HiddenInput()}