Merge pull request hack4impact#156 from hack4impact/custom_select

Add customselect

Author: sbue

app/templates/macros/form_macros.html

@@ -75,6 +75,35 @@
     {% endif %}
 {% endmacro %}
 
+{# Render a field for the form #}
+{% macro render_custom_select(field) %}
+    {{ field.label }}
+    {% if field.description %}
+        <div class="ui pointing below label">
+            {{ field.description }}
+        </div>
+    {% endif %}
+    <div id="{{field.name}}" class="ui fluid {% if field.multiple %}multiple {% endif %}search selection dropdown">
+        {{ field() }}
+        <i class="dropdown icon"></i>
+        <div class="default text">{{field.label.text}}</div>
+        <div class="menu">
+            {% for choice in field.choices %}
+                <div class="item">{{choice}}</div>
+            {% endfor %}
+        </div>
+    </div>
+    {% if field.allow_custom %}
+        <script>
+            $(document).ready(function() {
+                $('div#{{field.name}}').dropdown({
+                  allowAdditions: true
+                })
+            });
+        </script>
+    {% endif %}
+{% endmacro %}
+
 {# Render a field for the form #}
 {% macro render_form_field(field, extra_classes='') %}
     {% if field.type == 'Radio Field' %}
@@ -97,8 +126,8 @@
             {{ field.label }}
         </div>
     {% elif field.type == 'RadioField' %}
+        {{ field.label }}
         {% for item in field %}
-            {{ field.label }}
             <div class="ui radio checkbox">
                 {{ item }}
                 {{ item.label }}
@@ -108,13 +137,15 @@
         {{ field(class='ui button') }}
     {% elif field.type == 'FormField' %}
         {{ render_form(field) }}
+    {% elif field.type == 'CustomSelectField' %}
+        {{ render_custom_select(field) }}
     {% else %}
         {{ field.label }}
-            {% if field.description %}
-                <div class="ui pointing below label">
-                    {{ field.description }}
-                </div>
-            {% endif %}
+        {% if field.description %}
+            <div class="ui pointing below label">
+                {{ field.description }}
+            </div>
+        {% endif %}
         {{ field(placeholder=field.label.text) }}
     {% endif %}
 {% endmacro %}

app/utils.py

@@ -1,4 +1,7 @@
 from flask import url_for
+from wtforms.fields import Field
+from wtforms.widgets import HiddenInput
+from wtforms.compat import text_type
 
 
 def register_template_utils(app):
@@ -18,3 +21,24 @@ def is_hidden_field(field):
 
 def index_for_role(role):
     return url_for(role.index)
+
+
+class CustomSelectField(Field):
+    widget = HiddenInput()
+
+    def __init__(self, label='', validators=None, multiple=False,
+                 choices=[], allow_custom=True, **kwargs):
+        super(CustomSelectField, self).__init__(label, validators, **kwargs)
+        self.multiple = multiple
+        self.choices = choices
+        self.allow_custom = allow_custom
+
+    def _value(self):
+        return text_type(self.data) if self.data is not None else ''
+
+    def process_formdata(self, valuelist):
+        if valuelist:
+            self.data = valuelist[1]
+            self.raw_data = [valuelist[1]]
+        else:
+            self.data = ''