from item import * import bisect month_dic={"jan":1,"feb":2,"maa":3,"apr":4,"mei":5,"jun":6,"jul":7,"aug":8,"sep":9,"okt":10,"nov":11,"dec":12} def dateToNumber(item): words = item.date.split() d = int(words[0]) m = month_dic[words[1]] y = int(words[2]) return((y*12+m)*31+d) file = "data" out = "data" def addStation(name,lat,long,openings,closings): items = parseItemFile(file) for opening in openings: newStation = Station(name,lat,long,opening) bisect.insort(items,newStation,key=dateToNumber) for closing in closings : stationClosure = RemoveStation(name,closing) bisect.insort(items,stationClosure,key=dateToNumber) f = open(out,"w") for item in items: string = item.toString() lines = string.split('\n') for line in lines: f.write(line+"\n") #f.write('\n') f.close()