CapacityStations.outputPreparation set up as method

dream/KnowledgeExtraction/PilotCases/CapacityStations/OutputPreparation.py

@@ -26,33 +26,27 @@ def getBuffer(data,stationId):
         if edge['destination']==stationId:
             return edge['source']
 
-# configuration=open("PilotCases\CapacityStations\sampleConfiguration.json", "r")
-configuration=open("sampleConfiguration.json", "r")
-configurationData=configuration.read()
-configurationJSON=json.loads(configurationData)
+def OutputPreparation(data,extractedData):
+    configurationJSON=data
+    dbJSON=extractedData
+    operations=dbJSON.get('operations',{})
+    stations=configurationJSON['graph']['node']
+    currentDate=configurationJSON['general']['currentDate']
+    currentDate=datetime.strptime(currentDate, '%Y/%m/%d')
     
-# db=open("PilotCases\CapacityStations\sampleDBExtraction.json", "r")
-db=open("sampleDBExtraction.json", "r")
-dbData=db.read()
-dbJSON=json.loads(dbData)
-operations=dbJSON.get('operations',{})
-stations=configurationJSON['graph']['node']
-currentDate=configurationJSON['general']['currentDate']
-currentDate=datetime.strptime(currentDate, '%Y/%m/%d')
-
-# set the interval capacity of the stations
-for operationId, operation in operations.iteritems():
+    # set the interval capacity of the stations
+    for operationId, operation in operations.iteritems():
         intervalCapacity=operation.get('intervalCapacity',[])
         for stationId, station in stations.iteritems():
             if stationId==operationId:
                 station['intervalCapacity']=intervalCapacity
     
-orders=dbJSON.get('orders',{})
-configurationJSON['input']['BOM']['productionOrders']=[]
+    orders=dbJSON.get('orders',{})
+    configurationJSON['input']['BOM']['productionOrders']=[]
     
-capacityRequirementDict={}
-earliestStartDict={}
-for order in orders:
+    capacityRequirementDict={}
+    earliestStartDict={}
+    for order in orders:
         for operation in order['sequence']: 
             operationId=operation.keys()[0]
             capacityRequirementDict[operationId]=operation[operationId]['requiredCapacity']
@@ -80,8 +74,8 @@ for order in orders:
                 "orderDate":orderDate
                 })
         
-WIP=dbJSON.get('WIP',{})
-if WIP:
+    WIP=dbJSON.get('WIP',{})
+    if WIP:
         node=configurationJSON['graph']['node']
         #create an empty wip list in all CapacityStationBuffers
         for (node_id,node_data) in node.iteritems():
@@ -101,11 +95,11 @@ if WIP:
                                 "name": orderId+'_'+oper+'_'+str(requiredCapacity)                                                          
                             })
                         
-updatedModelJSONString=json.dumps(configurationJSON, indent=5)
-updatedModel=open('UpdatedModel.json', mode='w')
-updatedModel.write(updatedModelJSONString)
-
+#     updatedModelJSONString=json.dumps(configurationJSON, indent=5)
+#     updatedModel=open('UpdatedModel.json', mode='w')
+#     updatedModel.write(updatedModelJSONString)
 
+    return configurationJSON