def update(self, id): bus = Bus.get_by_id(int(id), parent=self.get_current_account()) if not bus: return self.redirect_to('bus.list', messages.BUS_NOT_FOUND) form = BusForm(self.request.POST, obj=bus) if self.request.method == 'POST' and form.validate(): form.populate_obj(bus) bus.put() self.session.add_flash(messages.BUS_UPDATE_SUCCESS) return self.redirect_to('bus.list') return self.render_to_response('bus/form.haml', {'form': form})
def create(self): form = BusForm(self.request.POST) if self.request.method == 'POST' and form.validate(): if Bus.get_by_bus_id(form.data['bus_id']): self.session.add_flash(messages.BUS_EXISTS, level='error') return self.render_to_response('bus/form.haml', {'form': form}) bus = Bus(bus_id=form.data['bus_id'], is_premium=form.data['is_premium'], is_operational=form.data['is_operational'], parent=self.get_current_account()) bus.put() self.session.add_flash(messages.BUS_CREATE_SUCCESS, level='info') return self.redirect_to('bus.list') self.session.add_flash(messages.BUS_CREATE_ERROR, level='error') return self.redirect_to('bus.list')
def list(self): # We pass form so we can generate it with the modal using macros. return self.render_to_response('bus/list.haml', {'form': BusForm()})