plugin for operator gantt

dream/plugins/BatchesOperatorGantt.py

@@ -4,52 +4,65 @@ from pprint import pformat
 
 from dream.plugins import plugin
 from dream.plugins.TimeSupport import TimeSupportMixin
+import datetime
 
 class BatchesOperatorGantt(plugin.OutputPreparationPlugin, TimeSupportMixin):
 
   def postprocess(self, data):
     """Post process the data for Gantt gadget
     """
-    print 2
-#     # in this instance there is not need to define start time
-#     data['general']['dateFormat']='%Y/%m/%d'
-#     self.initializeTimeSupport(data)
-#     date_format = '%d-%m-%Y %H:%M'
-#     resultElements=data['result']['result_list'][-1]['elementList']
-#     task_dict = {}
-#     # loop in the results to find CapacityProjects
-#     for element in resultElements:
-#       if element['_class']=="Dream.CapacityProject":
-#         # add the project in the task_dict
-#         task_dict[element['id']] = dict(
-#         id=element['id'],
-#         text='Project %s' % element['id'],
-#         type='project',
-#         open=False)
-#         
-#         # loop in the project schedule to create the sub-tasks    
-#         projectSchedule=element['results'].get('schedule',{})
-#         
-#         for record in projectSchedule:
-#             task_dict[element['id']+record['stationId']] = dict(
-#                 id=element['id']+record['stationId'],
-#                 parent=element['id'],
-#                 text=record['stationId'],
-#                 start_date=self.convertToRealWorldTime(
-#                       record['entranceTime']).strftime(date_format),
-#                 stop_date=self.convertToRealWorldTime(
-#                       record['exitTime']).strftime(date_format),
-#                 open=False,
-#                 duration=int(record['exitTime'])-int(record['entranceTime']),
-#                 entranceTime=record['entranceTime']
-#             )
-#         
-#     # return the result to the gadget
-#     result = data['result']['result_list'][-1]
-#     result[self.configuration_dict['output_id']] = dict(
-#       time_unit=self.getTimeUnitText(),
-#       task_list=sorted(task_dict.values(),
-#         key=lambda task: (task.get('parent'),
-#                           task.get('type') == 'project',
-#                           task.get('entranceTime'),task.get('id'))))
+    strptime = datetime.datetime.strptime
+    # read the current date and define dateFormat from it
+    try:
+        now = strptime(data['general']['currentDate'], '%Y/%m/%d %H:%M')
+        data['general']['dateFormat']='%Y/%m/%d %H:%M'
+    except ValueError:
+        now = strptime(data['general']['currentDate'], '%Y/%m/%d')
+        data['general']['dateFormat']='%Y/%m/%d'
+    maxSimTime=data['general']['maxSimTime']
+    self.initializeTimeSupport(data)
+    date_format = '%d-%m-%Y %H:%M'
+    resultElements=data['result']['result_list'][-1]['elementList']
+    task_dict = {}
+    # loop in the results to find Operators
+    for element in resultElements:
+        if element['_class']=="Dream.Operator":
+            operatorId=element['id']
+            # add the operator in the task_dict
+            task_dict[element['id']] = dict(
+            id=operatorId,
+            text=operatorId,
+            type='operator',
+            open=False)
+        
+            k=1
+
+            for record in schedule:
+                entranceTime=record['entranceTime']
+                try:
+                    exitTime=schedule[k]['entranceTime']
+                except IndexError:
+                    exitTime=maxSimTime    
+                k+=1     
+                task_dict[operatorId+record['stationId']] = dict(
+                    id=operatorId+record['stationId'],
+                    parent=operatorId,
+                    text=record['stationId'],
+                    start_date=self.convertToRealWorldTime(
+                          entranceTime).strftime(date_format),
+                    stop_date=self.convertToRealWorldTime(
+                          exitTime).strftime(date_format),
+                    open=False,
+                    entranceTime=entranceTime,
+                    duration=exitTime-entranceTime,
+                )           
+         
+    # return the result to the gadget
+    result = data['result']['result_list'][-1]
+    result[self.configuration_dict['output_id']] = dict(
+      time_unit=self.getTimeUnitText(),
+      task_list=sorted(task_dict.values(),
+        key=lambda task: (task.get('parent'),
+                          task.get('type') == 'project',
+                          task.get('entranceTime'),task.get('id'))))
     return data