@@ 221-238 (lines=18) @@ | ||
218 | ret.update(Tube().map.get(linecode=l, stationcode=self.code).getAllTrains() ) |
|
219 | return ret |
|
220 | ||
221 | ||
222 | class TubeStationManager(dict): |
|
223 | def __init__(self): |
|
224 | pass |
|
225 | def addStation(self, station): |
|
226 | if not self.has_key(station.code): |
|
227 | self[station.code] = station |
|
228 | ||
229 | class TubeLine(object): |
|
230 | def __init__(self, code, name): |
|
231 | self.code = code |
|
232 | self.name = name |
|
233 | self._stations = TubeStationManager() |
|
234 | ||
235 | def __repr__(self): |
|
236 | return "<Tube.Line: %s>" % self.name |
|
237 | ||
238 | def getStations(self): |
|
239 | return self._stations |
|
240 | ||
241 | def getAllTrains(self): |
|
@@ 195-211 (lines=17) @@ | ||
192 | newP.track_code = p.platform_number |
|
193 | newP.next_train = p.track_code |
|
194 | self.platforms[newP.name] = newP |
|
195 | ||
196 | def getAllTrains(self): |
|
197 | ret = {} |
|
198 | for plat in self.platforms.keys(): |
|
199 | ret.update(self.platforms[plat].trains) |
|
200 | return ret |
|
201 | ||
202 | ||
203 | class TubeStation(object): |
|
204 | def __init__(self, code, name): |
|
205 | self.code = code |
|
206 | self.name = name |
|
207 | self._lines = TubeLineManager() |
|
208 | ||
209 | def __repr__(self): |
|
210 | return "<Tube.Station: %s>" % self.name |
|
211 | ||
212 | def getLines(self): |
|
213 | return self._lines |
|
214 |