deliver.py

import sys
from deliveryManager import DeliveryManager

filelines = ''
with open(sys.argv[1]) as f:
    filelines = f.readlines()

bits = filelines[0].split(' ')

rows = int(bits[0])
cols = int(bits[1])
dronecount = int(bits[2])
deadline = int(bits[3])
maxload = int(bits[4])

productTypeCount = int(filelines[1])
productTypeWeights = [int(x) for x in filelines[2].split(' ')]

numberOfWarehouses = int(filelines[3])

warehouses = []

start = 4;

for i in range(0,numberOfWarehouses):
    warehouse = {'id':i}
    locChunk = filelines[start + (i*2)].split(' ')
    warehouse['location'] = {'row': int(locChunk[0]), 'col':int(locChunk[1])}
    warehouse['typeCounts'] = [int(x) for x in filelines[(start + (i*2))+1].split(' ')]
    warehouses.append(warehouse)

curPos = start + numberOfWarehouses*2

customerOrderCount = int(filelines[curPos])

curPos += 1

customerOrders = []
    
for i in range(0, customerOrderCount):
    order = {'id':i}
    locChunk = filelines[curPos + (i*3)].split(' ')
    order['location'] = {'row': int(locChunk[0]), 'col':  int(locChunk[1])}
    order['itemCount'] = int(filelines[(curPos + (i*3))+1])
    order['itemTypes'] = [int(x) for x in filelines[(curPos + (i*3))+2].split(' ')]
    customerOrders.append(order)


print 'drones:', dronecount, 'maxload:', maxload, 'warehouses:', numberOfWarehouses, 'product types:', productTypeCount, 'orders:', customerOrderCount, 'max time:', deadline

dm = DeliveryManager(dronecount, maxload, warehouses, customerOrders, productTypeWeights)

dm.go()

dm.writeOutputFile(sys.argv[1])