def parse_schedules_trolley(trolley_list):
for trolley in trolley_list:
logging.info("Parsing schedule for trolley with id: %s", trolley["id"])
print "Current trolley_id is:", trolley["id"]
html = parser_utils.download_page(
TransportParser.TRANSPORT_PAGE_URL,
trolley["post_data"]
)
schedule = ScheduleParser.parse_html(html)
json_trolley_object = {
'type': 'trolley',
'id': trolley["id"],
'name': trolley["name"],
'workdays': False,
'weekend': False,
'everyday': False
}
# in trolley case ['workdays'] always True
if trolley['workdays'] is True:
# case, when trolley works only on workdays
if schedule['workdays'] is True:
print "Workdays only"
json_trolley_object['workdays'] = True
json_trolley_object['schedule_workdays'] = (
schedule['schedule_table']
)
json_trolley_object['stations_workdays'] = (
schedule['stations_list']
)
# regular case, when trolley works both on workdays and weekend
else:
json_trolley_object['workdays'] = True
json_trolley_object['schedule_workdays'] = (
schedule['schedule_table']
)
json_trolley_object['stations_workdays'] = (
schedule['stations_list']
)
if schedule['weekend'] is True:
json_trolley_object['weekend'] = True
html = parser_utils.download_page(
schedule['weekend_link']
)
schedule_weekend = ScheduleParser.parse_html(html)
json_trolley_object['schedule_weekend'] = (
schedule_weekend['schedule_table']
)
json_trolley_object['stations_weekend'] = (
schedule_weekend['stations_list']
)
parser_utils.save_json_file(
(parser_configs.directories["TROLLEYS_DIR"] +
trolley["id"] + ".json"),
json_trolley_object
)
def parse_schedules_suburban():
html = parser_utils.download_page(
SuburbanParser.SUBURBAN_TRANSPORT_PAGE_URL
)
suburban_buses = SuburbanParser.parse_html(html)
for bus in suburban_buses:
from_city = in_dict(bus, 'from_city')
if from_city:
# print bus['number']
from_city = no_whitespaces(from_city)
schedule_from_city = SuburbanScheduleParser.parse_schedule(
from_city
)
bus['schedule_from_city'] = schedule_from_city
if bus['number'] == '159':
print schedule_from_city
to_city = in_dict(bus, 'to_city')
if to_city:
# print bus['number']
to_city = no_whitespaces(to_city)
schedule_to_city = SuburbanScheduleParser.parse_schedule(
to_city
)
bus['schedule_to_city'] = schedule_to_city
if bus['number'] == '159':
print schedule_to_city
bus_id = in_dict(bus, 'number')
bus_id = no_whitespaces(bus_id)
bus_id = bus_id.replace(',', '_')
postfix = unidecode(bus['station']).lower().replace('/', '_')
postfix = postfix.replace('.', '')
postfix = postfix.replace('\'', '')
postfix = postfix.replace(' ', '_')
bus_id = unidecode(bus_id) + '_' + postfix
bus_id = "sub_" + bus_id
# print bus_id
bus['id'] = bus_id
parser_utils.save_json_file(
(parser_configs.directories["JSON_DIR"] +
"suburban_transport.json"),
suburban_buses
)
for dir_alias, dir_name in directories.iteritems():
if not os.path.exists(dir_name):
os.makedirs(dir_name)
logging.basicConfig(
filename=directories["LOGS_DIR"] + 'example.log',
level=logging.DEBUG,
format='%(levelname)s:%(asctime)s:%(message)s',
datefmt='%d-%m-%Y %H:%M:%S'
)
logging.info("Parsing transport id's started.")
transport_parser = TransportParser()
html = parser_utils.download_page(
TransportParser.TRANSPORT_PAGE_URL
)
transport = transport_parser.parse_html(html)
logging.info("Parsing transport id's finished.")
logging.info("Parsing buses schedules started.")
parse_schedules_bus(transport["bus_list"])
logging.info("Parsing trolleys schedules started.")
parse_schedules_trolley(transport["trolley_list"])
parse_schedules_suburban()
buses_trolleys = transport["trolley_list"] + transport["bus_list"]
def parse_schedules_bus(bus_list):
for bus in bus_list:
logging.info("Parsing schedule for bus with id: %s", bus["id"])
print "Current bus_id is:", bus["id"]
html = parser_utils.download_page(
TransportParser.TRANSPORT_PAGE_URL,
bus["post_data"]
)
schedule = ScheduleParser.parse_html(html)
json_bus_object = {
'type': 'bus',
'id': bus["id"],
'name': bus["name"],
'everyday': False
}
if bus['id'] in MIX_LIST:
type_ = 'mixed'
else:
type_ = 'bus'
json_bus_object['type'] = type_
# Case when bus have only weekend schedule
if bus['weekend'] is True:
json_bus_object['schedule_weekend'] = schedule['schedule_table']
json_bus_object['stations_weekend'] = schedule['stations_list']
# Case when bus have workdays, and maybe weekend schedule
else:
# Case of '7a' bus (everyday)
# @TODO: could be others buses like this
if schedule['everyday'] is True:
json_bus_object['schedule_everyday'] = (
schedule['schedule_table']
)
json_bus_object['stations'] = schedule['stations_list']
else:
json_bus_object['schedule_workdays'] = (
schedule['schedule_table']
)
json_bus_object['stations_workdays'] = schedule['stations_list']
if schedule['weekend'] is True:
html = parser_utils.download_page(
schedule['weekend_link']
)
schedule_weekend = ScheduleParser.parse_html(html)
json_bus_object['schedule_weekend'] = (
schedule_weekend['schedule_table']
)
json_bus_object['stations_weekend'] = (
schedule_weekend['stations_list']
)
if 'stations_weekend' in json_bus_object and 'stations_workdays' in json_bus_object:
if json_bus_object['stations_weekend'] == json_bus_object['stations_workdays']:
json_bus_object['stations'] = json_bus_object['stations_weekend']
json_bus_object.pop('stations_workdays')
json_bus_object.pop('stations_weekend')
parser_utils.save_json_file(
parser_configs.directories["BUSES_DIR"] + bus["id"] + ".json",
json_bus_object
)
def handle_endtag(self, tag):
if tag == 'table':
if self._schedule_table_started is True:
self._schedule_table_started = False
def handle_data(self, data):
if self._schedule_table_started:
if self._table_header_started:
d = no_whitespaces(unicode(data)).lower()
self._table_header.append(d)
if d in SuburbanParser.COLUMNS_HEADERS:
self._table_header_dict[
SuburbanParser.COLUMNS_HEADERS[d]
] = (
self._column_counter - 1
)
elif self._current_bus_counter < self._column_counter:
self._current_bus.append(unicode(data))
self._current_bus_counter += 1
if self._current_bus_counter == self._column_counter:
self._result.append(self._current_bus)
self._current_bus_counter = 0
self._current_bus = []
if __name__ == "__main__":
from parser_utils import download_page
html = download_page(SuburbanParser.SUBURBAN_TRANSPORT_PAGE_URL)
SuburbanParser.parse_html(html)
