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')