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})
def create(self): form = BusDriverForm(self.request.POST) if self.request.method == 'POST' and form.validate(): if BusDriver.get_by_driver_id(form.data['driver_id']): self.session.add_flash(messages.BUS_DRIVER_EXISTS, level='error') return self.render_to_response('bus_driver/form.haml', {'form': form}) bus_driver = BusDriver(driver_id=form.data['driver_id'], parent=self.get_current_account()) bus_driver.name = ' '.join((form.data['first_name'], form.data['last_name'])) bus_driver.put() self.session.add_flash(messages.BUS_DRIVER_CREATE_SUCCESS, level='info') return self.redirect_to('bus_driver.list') self.session.add_flash(messages.BUS_DRIVER_CREATE_ERROR, level='error') return self.redirect_to('bus_driver.list')