From 85007ced67fbdec20b9dbd659737b4cfff63be79 Mon Sep 17 00:00:00 2001
From: Gaetan Delannay <gaetan.delannay@gmail.com>
Date: Mon, 2 Mar 2015 12:19:19 +0100
Subject: [PATCH] [gen] Calendar widget: more work on timeslots.

---
 fields/calendar.py | 86 ++++++++++++++++++++++++++++++----------------
 gen/ui/calendar.js | 65 +++++++++++++++++++++--------------
 2 files changed, 96 insertions(+), 55 deletions(-)

diff --git a/fields/calendar.py b/fields/calendar.py
index 4239947..c27e42c 100644
--- a/fields/calendar.py
+++ b/fields/calendar.py
@@ -81,6 +81,11 @@ class Event(Persistent):
         if self.timeslot != 'main': res += ' ' + self.timeslot
         return res
 
+    def sameAs(self, other):
+        '''Is p_self the same as p_other?'''
+        return (self.eventType == other.eventType) and \
+               (self.timeslot == other.timeslot)
+
 # ------------------------------------------------------------------------------
 class Calendar(Field):
     '''This field allows to produce an agenda (monthly view) and view/edit
@@ -182,10 +187,9 @@ class Calendar(Field):
      <x>:field.pxTimelineLegend</x>''')
 
     # Popup for adding an event in the month view
-    pxAddEvent = Px('''
+    pxAddPopup = Px('''
      <div var="prefix='%s_newEvent' % field.name;
-               popupId=prefix + 'Popup';
-               showTimeslots=len(field.timeslots) &gt; 2"
+               popupId=prefix + 'Popup'"
           id=":popupId" class="popup" align="center">
       <form id=":prefix + 'Form'" method="post">
        <input type="hidden" name="fieldName" value=":field.name"/>
@@ -212,12 +216,10 @@ class Calendar(Field):
         </select>
        </div>
        <!-- Span the event on several days -->
-       <x if="not showTimeslots">
-        <div align="center" class="discreet" style="margin-bottom: 3px">
-         <span>:_('event_span')</span>
-         <input type="text" size="3" name="eventSpan"/>
-        </div>
-       </x>
+       <div align="center" class="discreet" style="margin-bottom: 3px">
+        <span>:_('event_span')</span>
+        <input type="text" size="3" name="eventSpan"/>
+       </div>
        <input type="button"
               value=":_('object_save')"
               onclick=":'triggerCalendarEvent(%s, %s, %s, %s, \
@@ -230,7 +232,7 @@ class Calendar(Field):
      </div>''')
 
     # Popup for removing events in the month view
-    pxDelEvent = Px('''
+    pxDelPopup = Px('''
      <div var="prefix='%s_delEvent' % field.name;
                popupId=prefix + 'Popup'"
           id=":popupId" class="popup" align="center">
@@ -299,18 +301,20 @@ class Calendar(Field):
           <!-- Icon for adding an event -->
           <x if="mayCreate">
            <img class="clickable" style="visibility:hidden"
-                var="info=field.getApplicableEventsTypesAt(zobj, date, \
+                var="info=field.getApplicableEventTypesAt(zobj, date, \
                            eventTypes, preComputed, True)"
                 if="info and info.eventTypes" src=":url('plus')"
-                onclick=":'openEventPopup(%s, %s, %s, null, null, %s, %s)' % \
-                 (q('new'), q(field.name), q(dayString), q(info.eventTypes),\
-                  q(info.message))"/>
+                var2="freeSlots=field.getFreeSlotsAt(date, events, slotIds,\
+                                                     slotIdsStr, True)"
+                onclick=":'openEventPopup(%s,%s,%s,null,null,%s,%s,%s)' % \
+                 (q('new'), q(field.name), q(dayString), q(info.eventTypes), \
+                  q(info.message), q(freeSlots))"/>
           </x>
           <!-- Icon for deleting event(s) -->
           <img if="mayDelete" class="clickable" style="visibility:hidden"
                src=":url(single and 'delete' or 'deleteMany')"
-               onclick=":'openEventPopup(%s, %s, %s, %s, %s, null, null)' % \
-             (q('del'), q(field.name), q(dayString), q('main'), q(spansDays))"/>
+               onclick=":'openEventPopup(%s,%s,%s,%s,%s)' %  (q('del'), \
+                        q(field.name), q(dayString), q('main'), q(spansDays))"/>
           <!-- Events -->
           <x if="events">
           <div for="event in events" style="color: grey">
@@ -318,8 +322,8 @@ class Calendar(Field):
            <!-- Icon for delete this particular event -->
             <img if="mayDelete and not single" class="clickable"
                  src=":url('delete')"  style="visibility:hidden"
-                 onclick=":'openEventPopup(%s, %s, %s, %s, null, null, null)'% \
-                   (q('del'), q(field.name), q(dayString), q(event.timeslot))"/>
+                 onclick=":'openEventPopup(%s,%s,%s,%s)' % (q('del'), \
+                            q(field.name), q(dayString), q(event.timeslot))"/>
           </div>
           </x>
           <!-- Events from other calendars -->
@@ -339,7 +343,7 @@ class Calendar(Field):
 
       <!-- Popups for creating and deleting a calendar event -->
       <x if="mayEdit and eventTypes">
-       <x>:field.pxAddEvent</x><x>:field.pxDelEvent</x></x>''')
+       <x>:field.pxAddPopup</x><x>:field.pxDelPopup</x></x>''')
 
     pxView = pxCell = Px('''
      <div var="defaultDate=field.getDefaultDate(zobj);
@@ -363,7 +367,10 @@ class Calendar(Field):
                allEventTypes=events[0];
                allEventNames=events[1];
                colors=field.getColors(zobj);
-               namesOfDays=field.getNamesOfDays(_)"
+               namesOfDays=field.getNamesOfDays(_);
+               showTimeslots=len(field.timeslots) &gt; 1;
+               slotIds=[slot.id for slot in field.timeslots];
+               slotIdsStr=','.join(slotIds)"
           id=":ajaxHookId">
       <script>:'var %s_maxEventLength = %d;' % \
                 (field.name, field.maxEventLength)</script>
@@ -658,8 +665,8 @@ class Calendar(Field):
         tooLate = endDate and not tooEarly and (date > endDate)
         return not tooEarly and not tooLate
 
-    def getApplicableEventsTypesAt(self, obj, date, eventTypes, preComputed,
-                                   forBrowser=False):
+    def getApplicableEventTypesAt(self, obj, date, eventTypes, preComputed,
+                                  forBrowser=False):
         '''Returns the event types that are applicable at a given p_date. More
            precisely, it returns an object with 2 attributes:
            * "events" is the list of applicable event types;
@@ -681,6 +688,20 @@ class Calendar(Field):
             if not res.message: res.message = ''
         return res
 
+    def getFreeSlotsAt(self, date, events, slotIds, slotIdsStr,
+                       forBrowser=False):
+        '''Gets the free timeslots in this calendar for some p_date. As a
+           precondition, we know that the day is not full (so timeslot "main"
+           cannot be taken). p_events are those already defined at p_date.
+           p_slotIds is the precomputed list of timeslot ids.'''
+        if not events: return forBrowser and slotIdsStr or slotIds
+        # Remove any taken slot
+        res = slotIds[1:] # "main" cannot be chosen: p_events is not empty
+        for event in events: res.remove(event.timeslot)
+        # Return the result
+        if not forBrowser: return res
+        return ','.join(res)
+
     def getEventsAt(self, obj, date):
         '''Returns the list of events that exist at some p_date (=day).'''
         obj = obj.o # Ensure p_obj is not a wrapper.
@@ -808,13 +829,18 @@ class Calendar(Field):
                 i -= 1
         return res
 
-    def hasEventsAt(self, obj, date, otherEvents):
-        '''Returns True if, at p_date, an event is found of the same type as
-           p_otherEvents.'''
-        if not otherEvents: return
-        events = self.getEventsAt(obj, date)
+    def hasEventsAt(self, obj, date, events):
+        '''Returns True if, at p_date, events are exactly of the same type as
+           p_events.'''
         if not events: return
-        return events[0].eventType == otherEvents[0].eventType
+        others = self.getEventsAt(obj, date)
+        if not others: return
+        if len(events) != len(others): return
+        i = 0
+        while i < len(events):
+            if not events[i].sameAs(others[i]): return
+            i += 1
+        return True
 
     def getOtherEventsAt(self, obj, date, others, eventNames, render, colors):
         '''Gets events that are defined in p_others at some p_date. If p_single
@@ -925,12 +951,12 @@ class Calendar(Field):
         # Create and store the event
         events.append(Event(eventType, timeslot))
         # Sort events in the order of timeslots
-        timeslots = [timeslot.id for timeslot in self.timeslots]
+        timeslots = [slot.id for slot in self.timeslots]
         if len(events) > 1:
             events.data.sort(key=lambda e: timeslots.index(e.timeslot))
             events._p_changed = 1
         # Span the event on the successive days if required
-        if handleEventSpan and eventSpan and (timeslot != 'main'):
+        if handleEventSpan and eventSpan:
             nbOfDays = min(int(eventSpan), self.maxEventLength)
             for i in range(nbOfDays):
                 date = date + 1
diff --git a/gen/ui/calendar.js b/gen/ui/calendar.js
index df0e739..aab9913 100644
--- a/gen/ui/calendar.js
+++ b/gen/ui/calendar.js
@@ -14,8 +14,40 @@ function askCalendar(hookId, objectUrl, render, fieldName, month) {
   askAjaxChunk(hookId, 'GET', objectUrl, fieldName+':pxView', params);
 }
 
+function enableOptions(select, enabled, selectFirst, message){
+  /* This function disables, in p_select, all options that are not in p_enabled.
+     p_enabled is a string containing a comma-separated list of option names.
+     If p_selectFirst is True, the first option from p_enabled will be selected
+     by default. p_message will be shown (as "title") for disabled options. */
+  // Get p_enabled as a dict
+  var l = enabled.split(',');
+  var d = {};
+  for (var i=0; i < l.length; i++) d[l[i]] = true;
+  // Remember if we have already selected the first enabled option
+  var isSelected = false;
+  var options = select.options;
+  // Disable options not being p_enabled
+  for (var i=0; i<options.length; i++) {
+    options[i].selected = false;
+    if (!options[i].value) continue;
+    if (options[i].value in d) {
+      options[i].disabled = false;
+      options[i].title = '';
+      // Select it?
+      if (selectFirst && !isSelected) {
+        options[i].selected = true;
+        isSelected = true;
+      }
+    }
+    else {
+      options[i].disabled = true;
+      options[i].title = message;
+    }
+  }
+}
+
 function openEventPopup(action, fieldName, day, timeslot, spansDays,
-                        applicableEventTypes, message) {
+                        applicableEventTypes, message, freeSlots) {
   /* Opens the popup for creating (or deleting, depending on p_action) a
      calendar event at some p_day. When action is "del", we need to know the
      p_timeslot where the event is assigned and if the event spans more days
@@ -23,12 +55,13 @@ function openEventPopup(action, fieldName, day, timeslot, spansDays,
      events for those successive days. When action is "new", a possibly
      restricted list of applicable event types for this day is given in
      p_applicableEventTypes; p_message contains an optional message explaining
-     why not applicable types are not applicable. */
+     why not applicable types are not applicable. When "new", p_freeSlots may
+     list the available timeslots at p_day. */
   var prefix = fieldName + '_' + action + 'Event';
   var f = document.getElementById(prefix + 'Form');
   f.day.value = day;
   if (action == 'del') {
-    f.timeslot.value = timeslot;
+    if (f.timeslot) f.timeslot.value = timeslot;
     // Show or hide the checkbox for deleting the event for successive days
     var elem = document.getElementById(prefix + 'DelNextEvent');
     var cb = elem.getElementsByTagName('input');
@@ -38,30 +71,12 @@ function openEventPopup(action, fieldName, day, timeslot, spansDays,
     else elem.style.display = 'none';
   }
   else if (action == 'new') {
-    // First: reinitialise input fields
+    // Reinitialise field backgrounds
     f.eventType.style.background = '';
-    var allOptions = f.eventType.options;
-    for (var i=0; i < allOptions.length; i++) {
-      allOptions[i].selected = false;
-    }
     if (f.eventSpan) f.eventSpan.style.background = '';
-    // Among all event types, show applicable ones and hide the others
-    var applicable = applicableEventTypes.split(',');
-    var applicableDict = {};
-    for (var i=0; i < applicable.length; i++) {
-      applicableDict[applicable[i]] = true;
-    }
-    for (var i=0; i < allOptions.length; i++) {
-      if (!allOptions[i].value) continue;
-      if (allOptions[i].value in applicableDict) {
-        allOptions[i].disabled = false;
-        allOptions[i].title = '';
-      }
-      else {
-        allOptions[i].disabled = true;
-        allOptions[i].title = message;
-      }
-    }
+    // Disable unapplicable events and non-free timeslots
+    enableOptions(f.eventType, applicableEventTypes, false, message);
+    if (f.timeslot) enableOptions(f.timeslot, freeSlots, true, 'Not free');
   }
   openPopup(prefix + 'Popup');
 }