ReadJSShifts plugin updated to create rolling patterns for a given...

ReadJSShifts plugin updated to create rolling patterns for a given (hard-coded) period of time. To be tested

dream/plugins/JobShop/ReadJSShifts.py

 from copy import copy
 import json
+import math
 import time
 import random
 import copy
@@ -35,20 +36,24 @@ class ReadJSShifts(plugin.InputPreparationPlugin, TimeSupportMixin):
     # read the current date and define dateFormat from it
     try:
       now = strptime(data['general']['currentDate'], '%Y/%m/%d %H:%M')
+      # calculate the hours to end the first day
+      hoursToEndFirstDay = datetime.datetime.combine(now.date(), datetime.time(23,59,59)) - datetime.datetime.combine(now.date(), now.time())
       data['general']['dateFormat']='%Y/%m/%d %H:%M'
     except ValueError:
       now = strptime(data['general']['currentDate'], '%Y/%m/%d')
+      hoursToEndFirstDay = datetime.time(23,59,59)
       data['general']['dateFormat']='%Y/%m/%d'
     self.initializeTimeSupport(data)
     
     shiftData = data["input"].get("shift_spreadsheet",[])
     nodes = data["graph"]["node"]
 
-    shiftPattern = {} #shift pattern dictionary
+    defaultShiftPattern = {} #default shift pattern dictionary (if no pattern is defined for certain dates)
+    exceptionShiftPattern = {} # exceptions for shift pattern dictionary as defined in the spreadsheet
+    
     if shiftData:
       shiftData.pop(0)
       #iteration through the raw data to structure it into ManPy config
-      
       for line in shiftData:
         # if all the records of that line are none then continue
         toContinue = False
@@ -104,16 +109,21 @@ class ReadJSShifts(plugin.InputPreparationPlugin, TimeSupportMixin):
             end = timeEndList[index]
             if not start and not end:
               continue
-            shiftPattern[lastrec].append([start, end])
+            exceptionShiftPattern[lastrec].append([start, end])
         #if resource name is defined
-        elif str(entityID) not in shiftPattern:
+        elif str(entityID) not in exceptionShiftPattern:
           #take the name of the last entered resource from here
           lastrec = str(entityID)
+          exceptionShiftPattern[lastrec] = []
           for index, start in enumerate(timeStartList):
             end = timeEndList[index]
             if not start and not end:
               continue
-            shiftPattern[lastrec] = [[start, end]]
+            # if there is no other entry
+            if not len(exceptionShiftPattern[lastrec]):
+              exceptionShiftPattern[lastrec] = [[start, end]]
+            else:
+              exceptionShiftPattern[lastrec].append([start, end])
         #to avoid overwriting existing records, if there is another entry for a resource but does not follow it immediately (e.g. W2-FS)
         else:
           lastrec = str(entityID)
@@ -122,18 +132,78 @@ class ReadJSShifts(plugin.InputPreparationPlugin, TimeSupportMixin):
             end = timeEndList[index]
             if not start and not end:
               continue
-            shiftPattern[lastrec].append([start, end])
+            exceptionShiftPattern[lastrec].append([start, end])
 
       #sorts the list in case the records were not entered in correct ascending order
-      for info in shiftPattern:
-        shiftPattern[info].sort(key=itemgetter(0))
+      for info in exceptionShiftPattern:
+        exceptionShiftPattern[info].sort(key=itemgetter(0))
+      
+      # ================================================================
+      #create default pattern for all operators (10 days long)
+      timeStartList = []
+      timeEndList = []
+      for dayNumber in range(0,10):
+        startTime = "08:00"
+        endTime = "18:00"
+        upDate = now.date()+datetime.timedelta(days=dayNumber)
+        shiftStart = self.convertToSimulationTime(strptime("%s %s" % (upDate, startTime), '%Y-%m-%d %H:%M'))
+        shiftEnd = self.convertToSimulationTime(strptime("%s %s" % (upDate, endTime), '%Y-%m-%d %H:%M'))
+        timePair = self.correctTimePair(shiftStart, shiftEnd)
+        shiftStart, shiftEnd = timePair
+        timeStartList.append(shiftStart)
+        timeEndList.append(shiftEnd)
+      #for every operator (can be also machine) create an entry on the defaultShiftPattern 
+      for node, node_data in nodes.iteritems():
+        #if the node is an operator
+        if node_data.get('_class', None) == 'Dream.Operator':
+          for index, start in enumerate(timeStartList):
+            end = timeEndList[index]
+            if not start and not end:
+              continue
+            if not node in defaultShiftPattern:
+              defaultShiftPattern[node] = [[start, end]]
+            else:
+              defaultShiftPattern[node].append([start, end])
+      # ================================================================
       
       for node, node_data in nodes.items():
-        if node in shiftPattern:
+        modifiedDefaultDays = [] # the days of the defaultShiftPattern that have been modified according to the exceptionShiftPattern
+        if node in exceptionShiftPattern:
+          for index1, exception in enumerate(exceptionShiftPattern[node]):
+            # XXX think of the case where the exception starts one day and finishes the next
+            # calculate the time difference in hours from the end of the first day to the end of the exception
+            # check if we are still in the first day
+            if hoursToEndFirstDay.total_seconds()/3600 > exception[-1]:
+              exceptionDay = 0
+            # calculate the number of days till the end of the exception
+            else:
+              exceptionDay = math.floor((exception[-1] - hoursToEndFirstDay.total_seconds()/3600)/24) + 1
+            for index2, default in enumerate(defaultShiftPattern[node]):
+              # check if we still are in the first day
+              if hoursToEndFirstDay.total_seconds()/3600 > default[-1]:
+                defaultDay = 0
+              # calculate the number of days till the end of the default shift
+              else:
+                defaultDay = math.floor((default[-1] - hoursToEndFirstDay.total_seconds()/3600)/24) + 1
+              if exceptionDay == defaultDay:
+                # update the defaultShiftPattern of the node (operator or machine)
+                # if the exception day has not been modified then delete the previous entry and use the first exception that occurs
+                if not exceptionDay in modifiedDefaultDays:
+                  defaultShiftPattern[node][index2] = exception
+                # otherwise append it at the end 
+                else:
+                  defaultShiftPattern[node].append(exception)
+                modifiedDefaultDays.append(exceptionDay) # the day has been modified, add to the modified days
+                break
+        # update the interruptions of the nodes that have a defaultShiftPattern
+        if node in defaultShiftPattern:
+          # sort the shift pattern of every node
+          defaultShiftPattern[node].sort(key=itemgetter(0))
+          # get the interruptions of the object
           interruptions = node_data.get("interruptions", {})
           if not interruptions:
             node_data["interruptions"] = {}
-          node_data["interruptions"]["shift"] = {"shiftPattern": shiftPattern.pop(node),
+          node_data["interruptions"]["shift"] = {"shiftPattern": defaultShiftPattern.pop(node),
                                                  "endUnfinished": 0}
 	
-    return data
\ No newline at end of file
+    return data