def delete(self, id):
bus_driver = BusDriver.get_by_id(int(id), parent=self.get_current_account())
if not bus_driver:
self.session.add_flash(messages.BUS_DRIVER_NOT_FOUND, level='error')
return self.redirect_to('bus_driver.list')
bus_driver.delete()
self.session.add_flash(messages.BUS_DRIVER_DELETE_SUCCESS)
return self.redirect_to('bus_driver.list')
def update(self, id):
bus_driver = BusDriver.get_by_id(int(id), parent=self.get_current_account())
if not bus_driver:
return self.redirect_to('bus_driver.list', messages.BUS_DRIVER_NOT_FOUND)
form = BusDriverForm(self.request.POST, obj=bus_driver)
if self.request.method == 'POST' and form.validate():
form.populate_obj(bus_driver)
bus_driver.name = ' '.join((form.data['first_name'],
form.data['last_name']))
bus_driver.put()
self.session.add_flash(messages.BUS_DRIVER_UPDATE_SUCCESS)
return self.redirect_to('bus_driver.list')
return self.render_to_response('drivers/form.haml', {'form': form})
