Add JQueryUI date picker to event start/end date fields

events/forms.py

@@ -1,9 +1,13 @@
 from django.utils.safestring import mark_safe
-from django.forms import ModelForm, Field
-from django.forms.widgets import TextInput, Select, Media
+from django.forms import ModelForm, Field, SplitDateTimeWidget, DateInput, MultiWidget, Select
+from django.forms.widgets import TextInput, Media
+from django.utils.translation import ugettext_lazy as _
+
 from .models.profiles import Team
 from .models.events import Event, Place
 
+from datetime import time
+
 class LookupMedia(Media):
     def render(self):
         return mark_safe('''<script type="text/javascript"><script>
@@ -42,6 +46,96 @@ class Lookup(TextInput):
         context['widget']['label'] = self.label
         return context
 
+class DateWidget(DateInput):
+    """A more-friendly date widget with a pop-up calendar.
+    """
+    template_name = 'forms/widgets/date.html'
+    def __init__(self, attrs=None):
+        self.date_class = 'datepicker'
+        if not attrs:
+            attrs = {}
+        if 'date_class' in attrs:
+            self.date_class = attrs.pop('date_class')
+        if 'class' not in attrs:
+            attrs['class'] = 'date'
+
+        super(DateWidget, self).__init__(attrs=attrs)
+
+
+class TimeWidget(MultiWidget):
+    """A more-friendly time widget.
+    """
+    def __init__(self, attrs=None):
+        self.time_class = 'timepicker'
+        if not attrs:
+            attrs = {}
+        if 'time_class' in attrs:
+            self.time_class = attrs.pop('time_class')
+        if 'class' not in attrs:
+            attrs['class'] = 'time'
+
+        widgets = (
+            Select(attrs=attrs, choices=[(i + 1, "%02d" % (i + 1)) for i in range(0, 12)]),
+            Select(attrs=attrs, choices=[(i, "%02d" % i) for i in range(00, 60, 15)]),
+            Select(attrs=attrs, choices=[('AM', _('AM')), ('PM', _('PM'))])
+        )
+
+        super(TimeWidget, self).__init__(widgets, attrs)
+
+    def decompress(self, value):
+        if isinstance(value, str):
+            try:
+                value = strptime(value, '%I:%M %p')
+            except:
+                value = strptime(value, '%H:%M:%S')
+            hour = int(value.tm_hour)
+            minute = int(value.tm_min)
+            if hour < 12:
+                meridian = 'AM'
+            else:
+                meridian = 'PM'
+                hour -= 12
+            return (hour, minute, meridian)
+        elif isinstance(value, time):
+            hour = int(value.strftime("%I"))
+            minute = int(value.strftime("%M"))
+            meridian = value.strftime("%p")
+            return (hour, minute, meridian)
+        return (None, None, None)
+
+    def value_from_datadict(self, data, files, name):
+        value = super(TimeWidget, self).value_from_datadict(data, files, name)
+        t = strptime("%02d:%02d %s" % (int(value[0]), int(value[1]), value[2]), "%I:%M %p")
+        return strftime("%H:%M:%S", t)
+
+    def format_output(self, rendered_widgets):
+        return '<span class="%s">%s%s%s</span>' % (
+            self.time_class,
+            rendered_widgets[0], rendered_widgets[1], rendered_widgets[2]
+        )
+
+class DateTimeWidget(SplitDateTimeWidget):
+    """
+    A more-friendly date/time widget.
+    """
+    def __init__(self, attrs=None, date_format=None, time_format=None):
+        super(DateTimeWidget, self).__init__(attrs, date_format, time_format)
+        self.widgets = (
+            DateWidget(attrs=attrs),
+            TimeWidget(attrs=attrs),
+        )
+
+    def decompress(self, value):
+        if value:
+            d = strftime("%Y-%m-%d", value.timetuple())
+            t = strftime("%I:%M %p", value.timetuple())
+            return (d, t)
+        else:
+            return (None, None)
+
+    def format_output(self, rendered_widgets):
+        return '%s %s' % (rendered_widgets[0], rendered_widgets[1])
+
 class TeamForm(ModelForm):
     class Meta:
         model = Team
@@ -74,6 +168,8 @@ class NewTeamEventForm(ModelForm):
         fields = ['name', 'start_time', 'end_time', 'summary', 'place']
         widgets = {
             'place': Lookup(source='/api/places/', label='name'),
+            'start_time': DateTimeWidget,
+            'end_time': DateTimeWidget
         }
 
 class NewPlaceForm(ModelForm):

events/templates/forms/widgets/date.html

@@ -0,0 +1 @@
+<span class="datepicker">{% include "django/forms/widgets/input.html" %}</span>

get_together/templates/get_together/create_event.html

@@ -13,9 +13,9 @@
 {% endblock %}
 
 {% block javascript %}
-{% with event_form.fields.place.widget as widget %}
 <script type="text/javascript">
 $(document).ready(function(){
+{% with event_form.fields.place.widget as widget %}
     $("#{{ widget.name }}_search").keyup(function() {
 	var searchText = this.value;
 	$.getJSON("{{ widget.source }}?q="+searchText, function(data) {
@@ -26,7 +26,18 @@ $(document).ready(function(){
 	    });
 	});
     });
-});
-</script>
 {% endwith %}
+
+    $.datepicker.setDefaults({
+        showOn: 'focus',
+        dateFormat: 'yy-mm-dd',
+    });
+
+    $("#id_start_time_0").datepicker({altField: "#id_end_time_0", altFormat: "yy-mm-dd"});
+
+    $("#id_end_time_0").datepicker();
+
+});
+
+</script>
 {% endblock %}