50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
class Route(object):
|
|
"""docstring for Route."""
|
|
def __init__(self, route_id, load,src,dst):
|
|
super(Route, self).__init__()
|
|
self.route_id = route_id
|
|
self.load = load
|
|
self.src = src
|
|
self.dst = dst
|
|
|
|
def __repr__(self):
|
|
return str(self.route_id)+":"+self.src+" - "+self.dst+" "+self.load
|
|
|
|
|
|
class Contract(object):
|
|
"""docstring for Contract."""
|
|
def __init__(self, contract_id, routes):
|
|
super(Contract, self).__init__()
|
|
self.contract_id = contract_id
|
|
self.routes = routes
|
|
|
|
def __repr__(self):
|
|
return "\tContract : "+str(self.contract_id)+"\n\tRoutes\n\t\t"+"\n\t\t".join(map(repr,self.routes))
|
|
|
|
class Transporter(object):
|
|
"""docstring for Transporter."""
|
|
def __init__(self, contracts):
|
|
super(Transporter, self).__init__()
|
|
self.contracts = contracts
|
|
|
|
@staticmethod
|
|
def read_contracts(c_file):
|
|
c_lines = open(c_file,'r').readlines()
|
|
c_dict = {}
|
|
for l in c_lines:
|
|
(c_id,src,dst,load) = l.split(' ')
|
|
if c_dict.has_key(c_id):
|
|
route_id = len(c_dict[c_id])
|
|
c_dict[c_id].append(Route(route_id,load,src,dst))
|
|
else:
|
|
c_dict[c_id] = [Route(0,load,src,dst)]
|
|
|
|
contrants = []
|
|
for c_id in c_dict:
|
|
contrants.append(Contract(int(c_id),c_dict[c_id]))
|
|
|
|
return contrants
|
|
|
|
|
|
print Transporter.read_contracts('./contracts.txt')
|