Work on route form

1 files changed, 10 insertions, 1 deletions

diff --git a/forms.py b/forms.py
index a06d67e..77a9865 100644
--- a/forms.py
+++ b/forms.py
@@ -24,7 +24,8 @@ from django import forms
 from django.utils.translation import ugettext as _
 
 from chimere.models import SubCategory
-from chimere.forms import MarkerForm as BaseMarkerForm
+from chimere.forms import MarkerForm as BaseMarkerForm, \
+                          RouteForm as BaseRouteForm
 from chimere.widgets import TextareaWidget
 
 class MarkerForm(BaseMarkerForm):
@@ -36,3 +37,11 @@ class MarkerForm(BaseMarkerForm):
         self.fields['categories'].widget.choices = [('', _(u"Select"))] + \
                                                SubCategory.getAvailableTuples()
 
+class RouteForm(BaseRouteForm):
+    categories = forms.ChoiceField(choices=[], widget=forms.Select)
+    description = forms.CharField(widget=TextareaWidget, required=False)
+
+    def __init__(self, *args, **kwargs):
+        super(RouteForm, self).__init__(*args, **kwargs)
+        self.fields['categories'].widget.choices = [('', _(u"Select"))] + \
+                                               SubCategory.getAvailableTuples()