def build_existing_outage_list(self):
self.existing_outages = {}
q = Outage.all()
q.filter('end_date = ', None)
count = q.count()
outage_list = q.fetch(count)
for outage in outage_list:
self.existing_outages[outage.equipment_id] = outage
def update_data(self, data):
self.build_existing_outage_list()
dom = parseString(data)
outages = dom.getElementsByTagName('outage')
for outage in outages:
mta_id = self.get_node_value(outage, 'equipment')
if mta_id == None:
# TODO
continue
if mta_id in self.existing_outages:
# this outage is already in the database
del self.existing_outages[mta_id]
continue
station_name = self.get_node_value(outage, 'station')
trains_value = self.get_node_value(outage, 'trainno')
if trains_value == None:
# TODO: this is a problem
continue
trains = trains_value.split('/')
equipment_id = self.get_node_value(outage, 'equipment')
if equipment_id == None:
# TODO
continue
equipment_type_value = self.get_node_value(outage, 'equipmenttype')
if equipment_type_value == None:
# TODO
continue
equipment_type = EquipmentType.abbrev_to_code(equipment_type_value)
serving = self.get_node_value(outage, 'serving')
if serving == None:
# TODO
continue
is_ada_value = self.get_node_value(outage, 'ADA')
if is_ada_value == None or (is_ada_value != 'Y' and is_ada_value != 'N'):
# TODO
continue
is_ada = is_ada_value == 'Y'
start_date_value = self.get_node_value(outage, 'outagedate')
if start_date_value == None:
# TODO
continue
start_date = datetime.strptime(start_date_value, Updater.DT_FORMAT)
est_end_date_value = self.get_node_value(outage, \
'estimatedreturntoservice')
if est_end_date_value == None:
# TODO
continue
est_end_date = datetime.strptime(est_end_date_value, Updater.DT_FORMAT)
reason = self.get_node_value(outage, 'reason')
if reason == None:
# TODO
continue
station = self.get_station(station_name, trains)
if station == None:
outage = Outage(equipment_id = equipment_id, \
equipment_type = equipment_type, equipment_serving = serving, \
equipment_is_ada = is_ada, reason = reason, \
start_date = start_date, est_end_date = est_end_date)
else:
outage = Outage(station = station, equipment_id = equipment_id, \
equipment_type = equipment_type, equipment_serving = serving, \
equipment_is_ada = is_ada, reason = reason, \
start_date = start_date, est_end_date = est_end_date)
outage.put()
now = datetime.now()
for outage_id in self.existing_outages:
outage = self.existing_outages[outage_id]
outage.end_date = now
outage.put()