def __update():
try:
bus_route_list = bus_route_accessor.get_all_bus_routes()
word_pool = set()
for route in bus_route_list:
stations = route.stations
for station in stations:
word_pool.add(station)
train_route_list = train_route_accessor.get_all_train_routes()
""":type :list[TrainRoute]"""
for route in train_route_list:
for station in route.stations:
word_pool.add(station)
walk_route_list = walk_route_accessor.get_all_walk_routes()
""":type :list[WalkRoute]"""
for route in walk_route_list:
word_pool.add(route.walk_from)
word_pool.add(route.walk_to)
busway_transfer_list = busway_transfer_accessor.get_all_busway_transfers()
""":type :list[BuswayTransfer]"""
for route in busway_transfer_list:
word_pool.add(route.from_station)
word_pool.add(route.to_station)
for word in word_pool:
add_keyword(word, word)
except Exception, e:
__logger.error(e)
def __update():
try:
bus_route_list = bus_route_accessor.get_all_bus_routes()
word_pool = set()
for route in bus_route_list:
stations = route.stations
for station in stations:
word_pool.add(station)
train_route_list = train_route_accessor.get_all_train_routes()
""":type :list[TrainRoute]"""
for route in train_route_list:
for station in route.stations:
word_pool.add(station)
walk_route_list = walk_route_accessor.get_all_walk_routes()
""":type :list[WalkRoute]"""
for route in walk_route_list:
word_pool.add(route.walk_from)
word_pool.add(route.walk_to)
busway_transfer_list = busway_transfer_accessor.get_all_busway_transfers(
)
""":type :list[BuswayTransfer]"""
for route in busway_transfer_list:
word_pool.add(route.from_station)
word_pool.add(route.to_station)
for word in word_pool:
add_keyword(word, word)
except Exception, e:
__logger.error(e)
def __generate_train_mode():
"""
:rtype list[TransportationMode]
:return:
"""
mode_list = []
train_route_list = train_route_accessor.get_all_train_routes()
""":type :list[TrainRoute]"""
for train_route in train_route_list:
station_list = train_route.stations
line_name = train_route.line_name
origin_list = station_list[:-1]
destination_list = station_list[1:]
for origin, destination in zip(origin_list, destination_list):
# TODO remove this eta hack!
eta = 3
transportation_mode = TransportationMode()
transportation_mode.eta = eta
transportation_mode.name = ' '.join(
['KRL ', train_route.line_name])
transportation_mode.origin = origin
transportation_mode.destination = destination
# TODO remove this price hack
transportation_mode.price = 500
transportation_mode.heading_from = station_list[0]
transportation_mode.heading_to = station_list[-1]
mode_list.append(transportation_mode)
for destination, origin in zip(origin_list, destination_list):
# TODO remove this eta hack!
eta = 3
transportation_mode = TransportationMode()
transportation_mode.eta = eta
transportation_mode.name = ' '.join(
['KRL ', train_route.line_name])
transportation_mode.origin = origin
transportation_mode.destination = destination
# TODO remove this price hack
transportation_mode.price = 500
transportation_mode.heading_from = station_list[-1]
transportation_mode.heading_to = station_list[0]
mode_list.append(transportation_mode)
return mode_list
def __generate_train_mode():
"""
:rtype list[TransportationMode]
:return:
"""
mode_list = []
train_route_list = train_route_accessor.get_all_train_routes()
""":type :list[TrainRoute]"""
for train_route in train_route_list:
station_list = train_route.stations
line_name = train_route.line_name
origin_list = station_list[:-1]
destination_list = station_list[1:]
for origin, destination in zip(origin_list, destination_list):
# TODO remove this eta hack!
eta = 3
transportation_mode = TransportationMode()
transportation_mode.eta = eta
transportation_mode.name = ' '.join(['KRL ', train_route.line_name])
transportation_mode.origin = origin
transportation_mode.destination = destination
# TODO remove this price hack
transportation_mode.price = 500
transportation_mode.heading_from = station_list[0]
transportation_mode.heading_to = station_list[-1]
mode_list.append(transportation_mode)
for destination, origin in zip(origin_list, destination_list):
# TODO remove this eta hack!
eta = 3
transportation_mode = TransportationMode()
transportation_mode.eta = eta
transportation_mode.name = ' '.join(['KRL ', train_route.line_name])
transportation_mode.origin = origin
transportation_mode.destination = destination
# TODO remove this price hack
transportation_mode.price = 500
transportation_mode.heading_from = station_list[-1]
transportation_mode.heading_to = station_list[0]
mode_list.append(transportation_mode)
return mode_list
def test_train_fetcher(self):
train_route_accessor.reset()
train_info_fetcher.start()
routes = train_route_accessor.get_all_train_routes()
self.assertIsNotNone(routes)
self.assertTrue(len(routes) > 0)
